《计算机应用》唯一官方网站

• •    下一篇

基于有向超图自适应卷积的链接预测模型

赵文博1,马紫彤1,杨哲2   

  1. 1. 苏州大学计算机科学与技术学院
    2. 苏州大学 计算机科学与技术学院, 江苏 苏州 215006
  • 收稿日期:2024-01-04 修回日期:2024-03-12 发布日期:2024-03-28 出版日期:2024-03-28
  • 通讯作者: 杨哲
  • 基金资助:
    国家自然科学基金;教育部产学合作协同育人项目

Link Prediction Based on Directed Hypergraphs Adaptive Convolution

  • Received:2024-01-04 Revised:2024-03-12 Online:2024-03-28 Published:2024-03-28
  • Contact: YANG Zhe
  • Supported by:
    the National Natural Science Foundation of China

摘要: 摘 要: 链接预测作为图学习领域中的典型问题,一直是研究的热点。尽管图神经网络为这一问题提供了多样化的解决方案,但由于普通图的结构限制,目前的模型在充分利用顶点间的高阶及不对称信息方面存在明显的不足。针对以上问题,提出了一种基于有向超图自适应卷积的链接预测模型,首先有向超图结构能够更充分地表示顶点间的高阶信息和方向信息,兼具超图和有向图的优势。其次有向超图自适应卷积采用自适应信息传播方式替代传统有向超图中的定向信息传播方式,从而解决了有向超边尾部顶点不能有效更新嵌入的问题,同时减轻多层卷积导致的顶点过度平滑问题。在Citeseer数据集上上基于显式顶点特征的实验结果显示,有向超图自适应卷积模型在链接预测任务上,相较于有向超图神经网络(DHNN)模型的ROC曲线下面积(AUC)指标提升2.23个百分点,平均精度(AP)提升1.31个百分点。因此该模型可以充分表达顶点间的关系,有效提高链接预测任务的准确度。

关键词: 有向超图, 链接预测, 超图卷积, 表示学习, 自适应卷积

Abstract: Abstract: Link prediction, as a typical problem in the field of graph learning, has always been a research hotspot. Although diverse solutions to this problem have been provided by graph neural networks, significant shortcomings in fully utilizing high-order and asymmetric information between vertices were identified due to the structural constraints of ordinary graphs. To address the above issues, a link prediction model based on directed hypergraph adaptive convolution was proposed. Firstly, the directed hypergraph structure was employed to more fully represent high-order and directional information between vertices, possessing advantages of both hypergraphs and directed graphs. Secondly, an adaptive information propagation method was adopted by directed hypergraph adaptive convolution to replace the traditional directional information propagation method in directed hypergraphs, thereby solving the problem of ineffective updating of embeddings for tail vertices of directed hyperedges and alleviating the problem of excessive smoothing of vertices caused by multi-layer convolution. Experimental results based on explicit vertex features on the Citeseer dataset show that the directed hypergraph adaptive convolution model achieves a 2.23 percentage point increase in the area under the ROC curve (AUC) and a 1.31 percentage point increase in average precision (AP) compared to the directed hypergraph neural network (DHNN) model in the link prediction task. Therefore, it can be concluded that this model adequately expresses the relationships between vertices and effectively improves the accuracy of the link prediction task.

Key words: directed hypergraph, link prediction, hypergraph convolution, representation learning, adaptive convolution

中图分类号: