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 条
  • [41] Wang X., 2018, AAAI Conf. Artif. Intell, V32, P2374, DOI DOI 10.1609/AAAI.V32I1.11266
  • [42] Wu F, 2019, PR MACH LEARN RES, V97
  • [43] Graph neural networks in node classification: survey and evaluation
    Xiao, Shunxin
    Wang, Shiping
    Dai, Yuanfei
    Guo, Wenzhong
    [J]. MACHINE VISION AND APPLICATIONS, 2022, 33 (01)
  • [44] A novel message passing neural network based on neighborhood expansion
    Xue, Yanfeng
    Jin, Zhen
    Apasiba, Abeo Timothy
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2023, 14 (03) : 849 - 860
  • [45] Yadati N, 2019, ADV NEUR IN, V32
  • [46] NHP: Neural Hypergraph Link Prediction
    Yadati, Naganand
    Nitin, Vikram
    Nimishakavi, Madhav
    Yadav, Prateek
    Louis, Anand
    Talukdar, Partha
    [J]. CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, : 1705 - 1714
  • [47] Instruction-based Hypergraph Pretraining
    Yang, Mingdai
    Liu, Zhiwei
    Yang, Liangwei
    Liu, Xiaolong
    Wang, Chen
    Peng, Hao
    Yu, Philip S.
    [J]. PROCEEDINGS OF THE 47TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, SIGIR 2024, 2024, : 501 - 511
  • [48] Unified Pretraining for Recommendation via Task Hypergraphs
    Yang, Mingdai
    Liu, Zhiwei
    Yang, Liangwei
    Liu, Xiaolong
    Wang, Chen
    Peng, Hao
    Yu, Philip S.
    [J]. PROCEEDINGS OF THE 17TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, WSDM 2024, 2024, : 891 - 900
  • [49] Group Identification via Transitional Hypergraph Convolution with Cross-view Self-supervised Learning
    Yang, Mingdai
    Liu, Zhiwei
    Yang, Liangwei
    Liu, Xiaolong
    Wang, Chen
    Peng, Hao
    Yu, Philip S.
    [J]. PROCEEDINGS OF THE 32ND ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2023, 2023, : 2969 - 2979
  • [50] Framelet-based dual hypergraph neural networks for student performance prediction
    Yang, Yazhi
    Shi, Jiandong
    Li, Ming
    Fujita, Hamido
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2024, 15 (09) : 3863 - 3877