MEGNN: Meta-path extracted graph neural network for heterogeneous

被引:60
作者
Chang, Yaomin [1 ,2 ]
Chen, Chuan [1 ,2 ]
Hu, Weibo [1 ,2 ]
Zheng, Zibin [1 ,2 ]
Zhou, Xiaocong [1 ]
Chen, Shouzhi [3 ]
机构
[1] Sun Yat Sen Univ, Sch Comp Sci & Engn, Guangzhou, Peoples R China
[2] Sun Yat Sen Univ, Natl Engn Res Ctr Digital Life, Guangzhou, Peoples R China
[3] Tencent Inc, Shenzhen, Peoples R China
基金
中国国家自然科学基金;
关键词
Heterogeneous graph; Graph neural networks; Representation learning; Meta-paths;
D O I
10.1016/j.knosys.2021.107611
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Heterogeneous graphs with multiple types of nodes and edges are ubiquitous in the real world and possess immense value in many graph-based downstream applications. However, the heterogeneity within nodes and edges in heterogeneous graphs has brought pressing challenges for practical node representation learning. Existing works manually define multiple meta-paths to model the semantic relationships in heterogeneous graphs. Such strategies heavily rely on the quality of domain knowledge and require extensive hand-crafted works. In this paper, we propose a novel Meta-path Extracted heterogeneous Graph Neural Network (MEGNN) that is capable of extracting meaningful meta-paths in heterogeneous graphs, providing insights about data and explainable conclusions to the model's effectiveness. Concretely, MEGNN leverages heterogeneous convolution to combine different bipartite sub-graphs corresponding to edge types into a new trainable graph structure. By adopting the message passing paradigm of GNNs through trainable convolved graphs, MEGNN can optimize and extract effective meta-paths for heterogeneous graph representation learning. To enhance the robustness of MEGNN, we leverage multiple channels to yield various graph structures and devise a channel consistency regularizer to enforce the node embeddings learned from different channels to be similar. Extensive experimental results on three datasets not only show the effectiveness of MEGNN compared with the state-of-the-art methods, but also demonstrate the favorable interpretability of the extracted meta-paths. (C) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页数:11
相关论文
共 37 条
[1]  
Abu-El-Haija S, 2018, ADV NEUR IN, V31
[2]  
Cao Shaosheng, 2015, P 24 ACM INT C INFOR, P891
[3]   Representation Learning for Attributed Multiplex Heterogeneous Network [J].
Cen, Yukuo ;
Zou, Xu ;
Zhang, Jianwei ;
Yang, Hongxia ;
Zhou, Jingren ;
Tang, Jie .
KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, :1358-1368
[4]   Heterogeneous Network Embedding via Deep Architectures [J].
Chang, Shiyu ;
Han, Wei ;
Tang, Jiliang ;
Qi, Guo-Jun ;
Aggarwal, Charu C. ;
Huang, Thomas S. .
KDD'15: PROCEEDINGS OF THE 21ST ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2015, :119-128
[5]  
Chen ZT, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P2144
[6]   SINE: Side Information Network Embedding [J].
Chen, Zitai ;
Cai, Tongzhao ;
Chen, Chuan ;
Zheng, Zibin ;
Ling, Guohui .
DATABASE SYSTEMS FOR ADVANCED APPLICATIONS (DASFAA 2019), PT I, 2019, 11446 :692-708
[7]   How Do the Open Source Communities Address Usability and UX Issues? An Exploratory Study [J].
Cheng, Jinghui ;
Guo, Jin L. C. .
CHI 2018: EXTENDED ABSTRACTS OF THE 2018 CHI CONFERENCE ON HUMAN FACTORS IN COMPUTING SYSTEMS, 2018,
[8]  
Dahm Nicholas, 2013, Graph-Based Representations in Pattern Recognition. 9th IAPR-TC-15 International Workshop, GbRPR 2013. Proceedings, P11, DOI 10.1007/978-3-642-38221-5_2
[9]   metapath2vec: Scalable Representation Learning for Heterogeneous Networks [J].
Dong, Yuxiao ;
Chawla, Nitesh V. ;
Swami, Ananthram .
KDD'17: PROCEEDINGS OF THE 23RD ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2017, :135-144
[10]   MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding [J].
Fu, Xinyu ;
Zhang, Jiani ;
Men, Ziqiao ;
King, Irwin .
WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, :2331-2341