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 条
  • [1] Friends and neighbors on the Web
    Adamic, LA
    Adar, E
    [J]. SOCIAL NETWORKS, 2003, 25 (03) : 211 - 230
  • [2] Arya D, 2020, Arxiv, DOI [arXiv:2010.04558, DOI 10.48550/ARXIV.2010.04558]
  • [3] Networks beyond pairwise interactions: Structure and dynamics
    Battiston, Federico
    Cencetti, Giulia
    Iacopini, Iacopo
    Latora, Vito
    Lucas, Maxime
    Patania, Alice
    Young, Jean-Gabriel
    Petri, Giovanni
    [J]. PHYSICS REPORTS-REVIEW SECTION OF PHYSICS LETTERS, 2020, 874 : 1 - 92
  • [4] Simplicial closure and higher-order link prediction
    Benson, Austin R.
    Abebe, Rediet
    Schaub, Michael T.
    Jadbabaie, Ali
    Kleinberg, Jon
    [J]. PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2018, 115 (48) : E11221 - E11230
  • [5] The structure and dynamics of networks with higher order interactions
    Boccaletti, S.
    De Lellis, P.
    del Genio, C. I.
    Alfaro-Bittner, K.
    Criado, R.
    Jalan, S.
    Romance, M.
    [J]. PHYSICS REPORTS-REVIEW SECTION OF PHYSICS LETTERS, 2023, 1018 : 1 - 64
  • [6] A Survey on Hyperlink Prediction
    Chen, Can
    Liu, Yang-Yu
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (11) : 15034 - 15050
  • [7] Teasing out missing reactions in genome-scale metabolic networks through hypergraph learning
    Chen, Can
    Liao, Chen
    Liu, Yang-Yu
    [J]. NATURE COMMUNICATIONS, 2023, 14 (01)
  • [8] Chen L, 2020, AAAI CONF ARTIF INTE, V34, P27
  • [9] RSGNN: residual structure graph neural network
    Chen, Shuang
    Zhang, Changlun
    Gu, Fan
    Wang, Haochen
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2024, 15 (09) : 4079 - 4092
  • [10] Dai Q., 2023, HYPERGRAPH COMPUTATI, DOI [10.1007/978-981-99-0185-2, DOI 10.1007/978-981-99-0185-2]