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

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

扫码查看
图神经网络(GNN)为链接预测提供了多样化的解决方案,但由于普通图的结构限制,目前的相关模型在充分利用顶点间的高阶及不对称信息方面存在明显的不足.针对以上问题,提出一种基于有向超图自适应卷积的链接预测模型.首先,使用有向超图结构更充分地表示顶点间的高阶和方向信息,兼具超图和有向图的优势;其次,有向超图自适应卷积采用自适应信息传播方式替代传统有向超图中的定向信息传播方式,从而解决了有向超边尾部顶点不能有效更新嵌入的问题,同时解决多层卷积导致的顶点过度平滑问题.在Citeseer数据集上基于显式顶点特征的实验结果显示,在链接预测任务上,相较于有向超图神经网络(DHNN)模型,所提模型的ROC(Receiver Operating Characteristic)曲线下面积(AUC)指标提升了2.23个百分点,平均精度(AP)提升了1.31个百分点.因此,所提模型可以充分表达顶点间的关系,并有效提高链接预测任务的性能.
Link prediction model based on directed hypergraph adaptive convolution
Although diverse solutions for link prediction have been provided by Graph Neural Networks(GNN),the recent models have significant shortcomings in fully utilizing high-order and asymmetric information between vertices caused by 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 represent high-order and directional information between vertices more sufficiently,possessing advantages of both hypergraphs and directed graphs.Secondly,an adaptive information propagation method was adopted by directed hypergraph adaptive convolution to replace the directional information propagation method in traditional directed hypergraphs,thereby solving the problem of ineffective updating of embeddings for tail vertices of directed hyperedges,and solving the problem of excessive smoothing of vertices caused by multi-layer convolution.Experimental results based on explicit vertex features on Citeseer dataset show that the proposed model achieves a 2.23 percentage points increase in the Area Under the ROC(Receiver Operating Characteristic)Curve(AUC)and a 1.31 percentage points increase in Average Precision(AP)compared to the Directed Hypergraph Neural Network(DHNN)model in link prediction task.Therefore,it can be concluded that this model expresses the relationships between vertices adequately and improves the accuracy of link prediction task effectively.

Graph Neural Network(GNN)directed hypergraphlink predictionhypergraph convolutionrepresentation learningadaptive convolution

赵文博、马紫彤、杨哲

展开 >

苏州大学 计算机科学与技术学院,江苏 苏州 215008

江苏省计算机信息处理技术重点实验室(苏州大学),江苏 苏州 215006

图神经网络 有向超图 链接预测 超图卷积 表示学习 自适应卷积

2025

计算机应用
中国科学院成都计算机应用研究所

计算机应用

北大核心
影响因子:0.892
ISSN:1001-9081
年,卷(期):2025.45(1)