Neighborhood Overlap-aware Heterogeneous Hypergraph Neural Network for Link Prediction

Yifan Lu,Mengzhou Gao,Huan Liu,Zehao Liu,Wei Yu,Xiaoming Li,Pengfei Jiao
DOI: https://doi.org/10.1016/j.patcog.2023.109818
IF: 8
2023-07-25
Pattern Recognition
Abstract:In real world, a large number of networks are heterogeneous, containing different types of semantics and connections. Existing studies typically only consider lower-order pairwise relations rather than higher-order group interactions. Furthermore, they tend to focus more on node attributes rather than graph structural information. This results models failing to maintain graph topology effectively, which reduces the effectiveness on link prediction. To address these limitations, we propose N eighborhood O verlap-aware H eterogeneous hypergraph neural network (NOH) that learns useful structural information from the heterogeneous graph and estimates overlapped neighborhood for link prediction. Our model fuses the heterogeneity of graphs with structural information so that the model maintains both lower-order pairwise relations and higher-order complex semantics. Our extensive experiments on four real-world datasets show that NOH consistently achieves state-of-the-art performance on link prediction.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?