自编码
链接(几何体)
对抗制
计算机科学
图形
人工智能
图论
理论计算机科学
模式识别(心理学)
数学
人工神经网络
组合数学
计算机网络
作者
Zhihong Pan,Zhijie Zhong,Lingling Wei,Yunxuan Lin,Weisheng Li,Ronghua Lin,Yong Tang
标识
DOI:10.1109/tcss.2025.3569106
摘要
Link prediction is an important task that has numerous applications, including in recommender systems and social network analysis. Autoencoder is an effective method for solving the link prediction task. However, most existing autoencoder-based methods neither fully utilize the attribute information of the nodes nor fully take into account the potential data distribution in the graph. In this article, we propose a novel method named attribute-augmented adversarial variational graph autoencoder (A${}^{3}$VGAE), which can effectively solve the above two problems. The method first constructs the attribute structure graph based on the attribute information. Then, it inputs the topological structure graph, the attribute structure graph, and the attribute information into the shared encoder to obtain two latent representations. Besides, the topology structure graph, attribute structure graph, and attribute information are reconstructed by the dual decoder. The adversarial mechanism is introduced to ensure that the two latent representations match specific prior distributions. Extensive experiments conducted on four real-world graph datasets demonstrate the superiority of our proposed A${}^{3}$VGAE in link prediction tasks.
科研通智能强力驱动
Strongly Powered by AbleSci AI