Gaussian Process with Graph Convolutional Kernel for Relational Learning

被引:10
作者
Fang, Jinyuan [1 ]
Liang, Shangsong [1 ]
Meng, Zaiqiao [2 ]
Zhang, Qiang [3 ]
机构
[1] Sun Yat Sen Univ, Guangzhou, Peoples R China
[2] Univ Cambridge, Cambridge, England
[3] UCL, London, England
来源
KDD '21: PROCEEDINGS OF THE 27TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING | 2021年
基金
中国国家自然科学基金;
关键词
Gaussian Process; Graph Convolutional Kernel; Relational Learning;
D O I
10.1145/3447548.3467327
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Gaussian Process (GP) offers a principled non-parametric framework for learning stochastic functions. The generalization capability of GPs depends heavily on the kernel function, which implicitly imposes the smoothness assumptions of the data. However, common feature-based kernel functions are inefficient to model the relational data, where the smoothness assumptions implied by the kernels are violated. To model the complex and non-differentiable functions over relational data, we propose a novel Graph Convolutional Kernel, which enables to incorporate relational structures to feature-based kernels to capture the statistical structure of data. To validate the effectiveness of proposed kernel function in modeling relational data, we introduce GP models with Graph Convolutional Kernel in two relational learning settings, i.e., unsupervised settings of link prediction and semi-supervised settings of object classification. The parameters of our GP models are optimized through the scalable variational inducing point method. However, the highly structured likelihood objective requires densely sampling from variational distributions, which is costly and makes its optimization challenging in the unsupervised settings. To tackle this challenge, we propose a Local Neighbor Sampling technique with a provable more efficient computational complexity. Experimental results on real-world datasets demonstrate that our model achieves state-of-the-art performance in two relational learning tasks.
引用
收藏
页码:353 / 363
页数:11
相关论文
共 50 条
[21]  
Liang SS, 2019, AAAI CONF ARTIF INTE, P4269
[22]   Search Result Diversification in Short Text Streams [J].
Liang, Shangsong ;
Yilmaz, Emine ;
Shen, Hong ;
De Rijke, Maarten ;
Croft, W. Bruce .
ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2017, 36 (01)
[23]   Learning Dynamic Embeddings for Temporal Knowledge Graphs [J].
Liao, Siyuan ;
Liang, Shangsong ;
Meng, Zaiqiao ;
Zhang, Qiang .
WSDM '21: PROCEEDINGS OF THE 14TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2021, :544-552
[24]  
Liu Linfeng, 2019, Proc. Mach. Learn. Res., V89, P2291
[25]   The bottlenose dolphin community of Doubtful Sound features a large proportion of long-lasting associations - Can geographic isolation explain this unique trait? [J].
Lusseau, D ;
Schneider, K ;
Boisseau, OJ ;
Haase, P ;
Slooten, E ;
Dawson, SM .
BEHAVIORAL ECOLOGY AND SOCIOBIOLOGY, 2003, 54 (04) :396-405
[26]  
MacKay D. J. C., 1998, Neural Networks and Machine Learning. Proceedings, P133
[27]  
Meng Z, 2019, Neural Inf. Process. Syst.., P6507
[28]   Jointly Learning Representations of Nodes and Attributes for Attributed Networks [J].
Meng, Zaiqiao ;
Liang, Shangsong ;
Zhang, Xiangliang ;
Mccreadie, Richard ;
Ounis, Iadh .
ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2020, 38 (02)
[29]   Co-embedding Attributed Networks [J].
Meng, Zaiqiao ;
Liang, Shangsong ;
Bao, Hongyan ;
Zhang, Xiangliang .
PROCEEDINGS OF THE TWELFTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM'19), 2019, :393-401
[30]  
Ng Y., 2018, ADV NEURAL INFORM PR, V31, P1690