Wang Junbo, Chen Jianrui, Wang Zhihui, Gong Maoguo
School of Computer Science, Shaanxi Normal University, Xi'an, China.
School of Computer Science, Shaanxi Normal University, Xi'an, China; Key Laboratory of Modern Teaching Technology, Ministry of Education, Xi'an, China.
Neural Netw. 2025 Jan;181:106807. doi: 10.1016/j.neunet.2024.106807. Epub 2024 Oct 19.
Hyperedge prediction aims to predict common relations among multiple nodes that will occur in the future or remain undiscovered in the current hypergraph. It is traditionally modeled as a classification task, which performs hypergraph feature learning and classifies the target samples as either present or absent. However, these approaches involve two issues: (i) in hyperedge feature learning, they fail to measure the influence of nodes on the hyperedges that include them and the neighboring hyperedges, and (ii) in the binary classification task, the quality of the generated negative samples directly impacts the prediction results. To this end, we propose a Hypergraph Contrastive Attention Network (HCAN) model for hyperedge prediction. Inspired by the brain organization, HCAN considers the influence of hyperedges with different orders through the order propagation attention mechanism. It also utilizes the contrastive mechanism to measure the reliability of attention effectively. Furthermore, we design a negative sample generator to produce three different types of negative samples. We evaluate the impact of various negative samples on the model and analyze the problems of binary classification modeling. The effectiveness of HCAN in hyperedge prediction is validated by experimentally comparing 12 baselines on 9 datasets. Our implementations will be publicly available at https://github.com/jianruichen/HCAN.
超边预测旨在预测未来可能出现或当前超图中尚未发现的多个节点之间的共同关系。传统上,它被建模为一个分类任务,该任务进行超图特征学习,并将目标样本分类为存在或不存在。然而,这些方法存在两个问题:(i)在超边特征学习中,它们无法衡量节点对包含它们的超边和相邻超边的影响;(ii)在二元分类任务中,生成的负样本的质量直接影响预测结果。为此,我们提出了一种用于超边预测的超图对比注意力网络(HCAN)模型。受大脑组织的启发,HCAN通过阶次传播注意力机制考虑不同阶次超边的影响。它还利用对比机制有效地衡量注意力的可靠性。此外,我们设计了一个负样本生成器来生成三种不同类型的负样本。我们评估了各种负样本对模型的影响,并分析了二元分类建模的问题。通过在9个数据集上对12个基线进行实验比较,验证了HCAN在超边预测中的有效性。我们的实现将在https://github.com/jianruichen/HCAN上公开提供。