NHP: Neural Hypergraph Link Prediction

被引:65
作者
Yadati, Naganand [1 ]
Nitin, Vikram [2 ]
Nimishakavi, Madhav [3 ]
Yadav, Prateek [4 ]
Louis, Anand [1 ]
Talukdar, Partha [1 ]
机构
[1] Indian Inst Sci, Bangalore, Karnataka, India
[2] Columbia Univ, New York, NY USA
[3] Facebook AI, London, England
[4] LinkedIn, Bangalore, Karnataka, India
来源
CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT | 2020年
关键词
Link Prediction; Directed Hypergraph; Graph Neural Network; Knowledge Graph Canonicalisation;
D O I
10.1145/3340531.3411870
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Link prediction in simple graphs is a fundamental problem in which new links between vertices are predicted based on the observed structure of the graph. However, in many real-world applications, there is need to model relationships among vertices which go beyond pairwise associations. For example, in a chemical reaction, relationship among the reactants and products is inherently higher-order. Additionally, there is need to represent the direction from reactants to products. Hypergraphs provide a natural way to represent such complex higher-order relationships. Graph Convolutional Networks (GCN) have recently emerged as a powerful deep learning-based approach for link prediction over simple graphs. However, their suitability for link prediction in hypergraphs is underexplored - we fill this gap in this paper and propose Neural Hyperlink Predictor (NHP). NHP adapts GCNs for link prediction in hypergraphs. We propose two variants of NHP - NHP-U and NHP-D - for link prediction over undirected and directed hypergraphs, respectively. To the best of our knowledge, NHP-D is the first ever method for link prediction over directed hypergraphs. An important feature of NHP is that it can also be used for hyperlinks in which dissimilar vertices interact (e.g. acids reacting with bases). Another attractive feature of NHP is that it can be used to predict unseen hyperlinks at test time (inductive hyperlink prediction). Through extensive experiments on multiple real-world datasets, we show NHP's effectiveness.
引用
收藏
页码:1705 / 1714
页数:10
相关论文
共 44 条
[1]  
[Anonymous], 2018, PR MACH LEARN RES
[2]  
Arya Devanshu, 2018, ICMR
[3]   Neural Relational Topic Models for Scientific Article Analysis [J].
Bai, Haoli ;
Chen, Zhuangbin ;
Lyu, Michael R. ;
King, Irwin ;
Xu, Zenglin .
CIKM'18: PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2018, :27-36
[4]  
Battaglia P.W., 2018, COMPUTING RES REPOSI
[5]   Heterogeneous Hyper-Network Embedding [J].
Baytas, Inci M. ;
Xiao, Cao ;
Wang, Fei ;
Jain, Anil K. ;
Zhou, Jiayu .
2018 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2018, :875-880
[6]   Geometric Deep Learning Going beyond Euclidean data [J].
Bronstein, Michael M. ;
Bruna, Joan ;
LeCun, Yann ;
Szlam, Arthur ;
Vandergheynst, Pierre .
IEEE SIGNAL PROCESSING MAGAZINE, 2017, 34 (04) :18-42
[7]   Spectral Properties of Hypergraph Laplacian and Approximation Algorithms [J].
Chan, T. -H. Hubert ;
Louis, Anand ;
Tang, Zhihao Gavin ;
Zhang, Chenzi .
JOURNAL OF THE ACM, 2018, 65 (03)
[8]   Diffusion operator and spectral analysis for directed hypergraph Laplacian [J].
Chan, T-H Hubert ;
Tang, Zhihao Gavin ;
Wu, Xiaowei ;
Zhang, Chenzi .
THEORETICAL COMPUTER SCIENCE, 2019, 784 :46-64
[9]  
Fatemi Bahare, 2020, P 29 INT JOINT C ART
[10]  
Feng YF, 2019, AAAI CONF ARTIF INTE, P3558