When Convolutional Network Meets Temporal Heterogeneous Graphs: An Effective Community Detection Method

被引:4
作者
Zheng, Yaping [1 ]
Zhang, Xiaofeng [1 ]
Chen, Shiyi [1 ]
Zhang, Xinni [1 ]
Yang, Xiaofei [2 ]
Wang, Di [3 ]
机构
[1] Harbin Inst Technol, Sch Comp Sci, Shenzhen 150001, Peoples R China
[2] Univ Macau, Zhuhai 999078, Peoples R China
[3] Nanyang Technol Univ, Singapore 639798, Singapore
基金
中国国家自然科学基金;
关键词
Feature extraction; Task analysis; Tensors; Aggregates; Image edge detection; Graph neural networks; Three-dimensional displays; Graph convolutional network; heterogeneous graph; temporal graph; community detection;
D O I
10.1109/TKDE.2021.3096122
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Community detection has long been an important yet challenging task to analyze complex networks with a focus on detecting topological structures of graph data. Essentially, real-world graph data is generally heterogeneous which dynamically varies over time, and this invalidates most existing community detection approaches. To cope with these issues, this paper proposes the temporal-heterogeneous graph convolutional networks (THGCN) to detect communities using the learnt feature representations of a set of temporal heterogeneous graphs. Particularly, we first design a heterogeneous GCN component to represent features of heterogeneous graph at each time step. Then, a residual compressed aggregation component is proposed to learn temporal feature representations extracted from two consecutive heterogeneous graphs. These temporal features are considered to contain evolutionary patterns of underlying communities. To the best of our knowledge, this is the first attempt to detect communities from temporal heterogeneous graphs. To evaluate the model performance, extensive experiments are performed on two real-world datasets, i.e., DBLP and IMDB. The promising results have demonstrated that the proposed THGCN is superior to both benchmark and the state-of-the-art approaches, e.g., GCN, GAT, GNN, LGNN, HAN and STAR, with respect to a number of evaluation criteria.
引用
收藏
页码:2173 / 2178
页数:6
相关论文
共 19 条
  • [1] [Anonymous], 2004, Rev. E, Stat. Phys. Plasmas Fluids Relat.Interdiscip. Top., V69
  • [2] Latent Dirichlet allocation
    Blei, DM
    Ng, AY
    Jordan, MI
    [J]. JOURNAL OF MACHINE LEARNING RESEARCH, 2003, 3 (4-5) : 993 - 1022
  • [3] Chen X., 2019, P INT C LEARN REPR
  • [4] Cucurull G., 2018, PROC INT C LEARN REP, V6
  • [5] Hamilton WL, 2017, ADV NEUR IN, V30
  • [6] He DX, 2020, PROCEEDINGS OF THE TWENTY-NINTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P3515
  • [7] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [8] Jin D, 2019, AAAI CONF ARTIF INTE, P152
  • [9] Kipf T. N., 2017, P 5 INT C LEARN REPR, DOI DOI 10.48550/ARXIV.1609.02907
  • [10] Heterogeneous Dynamic Graph Attention Network
    Li, Qiuyan
    Shang, Yanlei
    Qiao, Xiuquan
    Dai, Wei
    [J]. 11TH IEEE INTERNATIONAL CONFERENCE ON KNOWLEDGE GRAPH (ICKG 2020), 2020, : 404 - 411