Deep graph clustering with enhanced feature representations for community detection

被引:15
作者
Hao, Jie [1 ]
Zhu, William [1 ]
机构
[1] Univ Elect Sci & Technol China, Inst Fundamental & Frontier Sci, 4,Sec 2,Jianshe North Rd, Chengdu, Peoples R China
基金
中国国家自然科学基金;
关键词
Community detection; Graph clustering; Autoencoder; Enhanced feature representation; MODULARITY; NETWORKS;
D O I
10.1007/s10489-022-03381-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Community detection is to partition community nodes into groups with similar attributes and topologies. In recent years, community detection becomes a research hotspot due to its great value and broad applications to social sciences. Thus, many clustering methods are developed for community detection. In particular, deep graph clustering has become a mainstream community detection approach because of its powerful abilities of feature representation and relationship extraction. Deep graph clustering uses graph neural networks (e.g., graph autoencoders) to learn the node feature representations with abundant topological relationships. Though deep graph clustering succeeds in dealing with topological relationships of nodes, there are insufficient attribute information in its learned feature representations, which hurt detection performance. To solve this problem, we propose an enhanced feature representation approach for deep graph clustering in community detection. Firstly, we construct a basic autoencoder to learn hierarchical attribute information and then deliver it into neural layers of a graph autoencoder. The graph autoencoder organically combines the received hierarchical attribute information with its extracted topological relationships to generate enhanced feature representations for clustering. Secondly, we design a self-supervised mechanism to optimize our deep graph clustering model. This mechanism uses the reconstruction losses of two autoencoders and clustering loss as self-supervised information to efficiently guide model updates. In this way, our approach overcomes the insufficient attribute information in generated feature representations, thus is more conducive to community detection. Extensive experiments demonstrate that the deep graph clustering with enhanced feature representations improves the performance of community detection compared to the other popular deep graph clustering approaches.
引用
收藏
页码:1336 / 1349
页数:14
相关论文
共 49 条
  • [1] Arthur D, 2007, PROCEEDINGS OF THE EIGHTEENTH ANNUAL ACM-SIAM SYMPOSIUM ON DISCRETE ALGORITHMS, P1027
  • [2] Structural Deep Clustering Network
    Bo, Deyu
    Wang, Xiao
    Shi, Chuan
    Zhu, Meiqi
    Lu, Emiao
    Cui, Peng
    [J]. WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, : 1400 - 1410
  • [3] A fast parallel modularity optimization algorithm (FPMQA) for community detection in online social network
    Bu, Zhan
    Zhang, Chengcui
    Xia, Zhengyou
    Wang, Jiandong
    [J]. KNOWLEDGE-BASED SYSTEMS, 2013, 50 : 246 - 259
  • [4] A new similarity combining reconstruction coefficient with pairwise distance for agglomerative clustering
    Cai, Zhiling
    Yang, Xiaofei
    Huang, Tianyi
    Zhu, William
    [J]. INFORMATION SCIENCES, 2020, 508 : 173 - 182
  • [5] The Relationship Between Age and Mental Health Among Adults in Iran During the COVID-19 Pandemic
    Chen, Jiyao
    Zhang, Stephen X.
    Wang, Yifei
    Afshar Jahanshahi, Asghar
    Mokhtari Dinani, Maryam
    Nazarian Madavani, Abbas
    Nawaser, Khaled
    [J]. INTERNATIONAL JOURNAL OF MENTAL HEALTH AND ADDICTION, 2022, 20 (05) : 3162 - 3177
  • [6] Ertöz L, 2003, SIAM PROC S, P47
  • [7] Ester M., 1996, P 2 INT C KNOWL DISC, P226, DOI DOI 10.5555/3001460.3001507
  • [8] Community structure in social and biological networks
    Girvan, M
    Newman, MEJ
    [J]. PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2002, 99 (12) : 7821 - 7826
  • [9] Guha S., 1998, SIGMOD Record, V27, P73, DOI 10.1145/276305.276312
  • [10] Guo XF, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1753