Higher-order link prediction via light hypergraph neural network and hybrid aggregator

被引:0
作者
Rui, Xiaobin [1 ,2 ]
Zhuang, Jiaxin [1 ]
Sun, Chengcheng [1 ]
Wang, Zhixiao [1 ,2 ]
机构
[1] China Univ Min & Technol, Sch Comp Sci & Technol, Xuzhou 221116, Jiangsu, Peoples R China
[2] Minist Educ, Mine Digitizat Engn Res Ctr, Xuzhou 221116, Jiangsu, Peoples R China
关键词
Higher-order link prediction; Hypergraph embedding; Light neural networks; Hyperlink aggregator;
D O I
10.1007/s13042-024-02414-x
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Link prediction, which aims to predict missing links or possible future links between two nodes, is one of the most important research in social network analysis. Higher-order link prediction, a natural extension of this problem in hypergraphs, focuses on predicting hyperlinks among multiple nodes. The extended problem has a wide range of applications, including predicting chemical reactions and forecasting social communications. Hypergraph neural networks (HGNNs), designed to address the characteristics of higher-order networks (hypergraphs), are variants of graph neural networks (GNNs). Compared with traditional methods, these end-to-end approaches based on HGNNs are effective tools for higher-order link prediction. However, these approaches still have some shortcomings. On the one hand, HGNNs, deriving from traditional GNNs, inevitably inherit unnecessary complexity and redundant computation from deep learning lineage. On the other hand, the existing aggregators do not take into account the similarity among the nodes' features, resulting in information loss in the embeddings of hyperlinks. To solve the both shortcomings, firstly, we propose a Light HyperGraph Neural Network (LHGNN) by removing some specific linear feature transformation layers and activation function layers to reduce complexity and increase reliability. Secondly, we propose a Hybrid Aggregator (HA) to obtain hyperlinks' embeddings more comprehensively by concatenating the mean and standard deviation of the nodes' embeddings. Experimental results on six datasets from four different domains show that our method outperforms state-of-the-art methods with a simpler structure.
引用
收藏
页码:2671 / 2685
页数:15
相关论文
共 53 条
  • [51] How Much and When Do We Need Higher-order Information in Hypergraphs? A Case Study on Hyperedge Prediction
    Yoon, Se-eun
    Song, Hyungseok
    Shin, Kijung
    Yi, Yung
    [J]. WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, : 2627 - 2633
  • [52] Zhang MH, 2018, AAAI CONF ARTIF INTE, P4430
  • [53] Predicting missing links via local information
    Zhou, Tao
    Lu, Linyuan
    Zhang, Yi-Cheng
    [J]. EUROPEAN PHYSICAL JOURNAL B, 2009, 71 (04) : 623 - 630