ИАПУ ДВО РАН

Hierarchical Graph Neural Network Based-on Semi-implicit Variational Inference


2023

IEEE Transactions on Cognitive and Developmental Systems, Q2

Статьи в журналах

Su H.L., Li Z.P., Zhu X.B., et al. Hierarchical Graph Neural Network Based-on Semi-implicit Variational Inference // IEEE Transactions on Cognitive and Developmental Systems. 2023. Vol. 15, no. 2. Pp. 887-895. DOI: 10.1109/TCDS.2022.3193398.

Graph neural network(GNN) has obtained outstanding achievements in relational data. However, these data have uncertain properties, for example, spurious edges may be included. Recently, Variational graph autoencoder(VGAE) has been proposed to solve this problem. However, the distributional assumptions in the variational family restrict the variational inference (VI) flexibility and they define variational families using mean-field, which can not capture complex posterior distributional. To solve the above question, in this paper, we proposed a novel GNN model based on semi-implicit variational inference (SIVI), which can embed the node to the latent space to improve VI flexibility and enhance VI expressiveness with mixing distribution. Specifically, to approximate the true posterior, a variational posterior was given utilizing a semi-implicit hierarchical variational framework, which can model complex posterior. Moreover, an iterative decoder is used to better capture graph properties. Besides, due to the hierarchical structure in our model, it can incorporation neighbour information between nodes. Experiments on multiple data sets, our method has achieved state-of-the-art results compared to other similar methods. Particularly, on the citation dataset Citeseer without features, our method outperforms VGAE by nine percentage.

10.1109/TCDS.2022.3193398

https://ieeexplore.ieee.org/document/9839324