Heterogeneous graph neural network with graph-data augmentation and adaptive denoising

Lou, Xiaojun,Li, Jian
DOI: https://doi.org/10.1007/s10489-024-05363-8
IF: 5.3
2024-03-27
Applied Intelligence
Abstract:Heterogeneous graphs are especially important in our daily life, which describe objects and their connections through nodes and edges. For this complex network structure, many heterogeneous graph neural networks have been designed, but the traditional heterogeneous graph neural network has several obvious shortcomings: (1) Models using meta-paths require selection of meta-paths, failing to learn all meta-paths in the dataset, and cannot capture structural information beyond meta-paths. (2) Models that do not apply meta-paths will be limited by the relationships of the original graph. Due to the absence or omission of some very important but costly relationships in the dataset, the ability of the model to learn node features is limited. (3) The existence of noise will affect the optimization of the parameters of the heterogeneous graph neural network models. In response to these problems, we propose our model: Heterogeneous Graph Neural Network with Graph-data Augmentation and Adaptive Denoising (GAAD). We hope that our model is not affected by differences in graph structure and noise nodes in the graph structure. In summary, we hope that our model can adapt to most graph structures. So we propose a method based on graph neural network to mine the hidden relationship between heterogeneous nodes, calculate the possibility of the existence of edges between heterogeneous nodes, strengthen the graph structure by adding highly possible edges, enrich the information transmission between nodes. Then we apply an adaptive noise reduction algorithm to prevent the possible noise diffusion caused by data enhancement in the graph structure. Extensive experiments on three real-world network datasets demonstrate the superiority of GAAD over the state-of-the-art methods. The code is available on https://github.com/xiaojunlou0831/GAAD.
computer science, artificial intelligence
What problem does this paper attempt to address?