Adaptive multi-channel Bayesian Graph Neural Network

被引:3
作者
Yang, Dong [1 ]
Liu, Zhaowei [2 ]
Wang, Yingjie [2 ]
Xu, Jindong [2 ]
Yan, Weiqing [2 ]
Li, Ranran [2 ,3 ]
机构
[1] Georgia State Univ, Dept Comp Sci, Atlanta, GA 30303 USA
[2] Yantai Univ, Sch Comp & Control Engn, Yantai 264005, Peoples R China
[3] Inst Network Technol Yantai, Yantai 264006, Peoples R China
关键词
Graph neural networks; Graph representation learning; Graph structure learning; Bayesian framework;
D O I
10.1016/j.neucom.2024.127260
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recent years have seen a surge in interest in graph neural networks (GNNs) due to their superior performance in a range of graph and network mining applications. Graph embedding attempts to convert nodes in graph data to a low-dimensional vector representation by capturing the edges between them. However, because the bulk of GNNs currently use unstable graph structures, they perform well on graphs with a high degree of homogeneity and badly on those with a low degree of homogeneity. As a result, GNNs that rely solely on the original graph structure may give unsatisfactory results. In this research, we introduce an adaptive multi -channel Bayesian graph neural network (AMBGN1) for estimating new graph structures and adaptively fusing some depth-related information between the original topological structures. The key idea is to follow the GNN mechanism by estimating the ideal graph structure using Bayesian inference and extracting specific and common embeddings from estimated graph structures, topological structures, and their combinations. We are able to maximize both estimated graph structure learning and node embedding in an iterative framework by using the attention approach to learn the significant weights of these three node embeddings. Our extensive research on a number of benchmark datasets with varied degrees of homogeneity demonstrated that AMBGN reliably estimates the graph structure and effectively learns the most relevant node information in both graph representations, confirming the usefulness of AMBGN.
引用
收藏
页数:12
相关论文
共 47 条
[1]  
Abu-El-Haifa S, 2019, PR MACH LEARN RES, V97
[2]  
Balcilar M., 2020, arXiv
[3]  
Bo D., 2022, 11 INT C LEARNING RE
[4]   Molecular generative Graph Neural Networks for Drug Discovery [J].
Bongini, Pietro ;
Bianchini, Monica ;
Scarselli, Franco .
NEUROCOMPUTING, 2021, 450 :242-252
[5]  
Bruna J., 2014, INT C LEARN REPR ICL
[6]  
Chen DL, 2020, AAAI CONF ARTIF INTE, V34, P3438
[7]  
Franceschi L, 2019, PR MACH LEARN RES, V97
[8]   STOCHASTIC BLOCKMODELS - 1ST STEPS [J].
HOLLAND, PW ;
LASKEY, KB ;
LEINHARDT, S .
SOCIAL NETWORKS, 1983, 5 (02) :109-137
[9]  
Hu Weihua, 2020, Advances in Neural Information Processing Systems, V33
[10]   Graph Structure Learning for Robust Graph Neural Networks [J].
Jin, Wei ;
Ma, Yao ;
Liu, Xiaorui ;
Tang, Xianfeng ;
Wang, Suhang ;
Tang, Jiliang .
KDD '20: PROCEEDINGS OF THE 26TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2020, :66-74