Multi-Scale Variational Graph Auto-Encoder Based on Subgraph Learning
Link prediction is a crucial component in the field of graph data analysis, playing a vital role in various disciplines such as social network analysis, construction of biological information networks, and research on information dissemination networks. This paper proposes a multi-scale variational graph auto-encoder model combined with subgraph partitioning, SGMS-VGAE, which aims to capture the structural features of graph data by integrating subgraph learning and a multi-scale variational graph auto-encoder framework. The model effectively extracts structural features of graph data through static subgraph partitioning and multi-scale variational inference techniques. Moreover, by employing subgraph partitioning techniques, the model can filter out noise and redundant information, focusing on local structures, thereby accelerating the training process. To validate the effectiveness of the SGMS-VGAE model, extensive experiments were conducted on real-world datasets such as Cora, Citeseer, and PubMed. The experimental results demonstrate that SGMS-VGAE outperforms existing methods in link prediction tasks.
Authors
- Xinyu Pei (ORCID: https://orcid.org/0009-0008-9105-7972)
- Yunhai Gao (ORCID: https://orcid.org/0009-0001-0966-6963)
- Shiwen Zhao (ORCID: https://orcid.org/0009-0000-8646-5228)
- Lianjun Song (ORCID: https://orcid.org/0000-0002-4463-5687)
- Wei Yuan (ORCID: https://orcid.org/0000-0002-5867-5364)
- Yuexiu Zhang
Institutions
- Tianjin Agricultural University (CN)
- Tianjin University of Finance and Economics (CN)
Publication Details
- Published
- 2024-11-20
- DOI
- https://doi.org/10.1109/dsins64146.2024.10992228
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 0.00