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 条
  • [21] Kaufman L., 2009, Finding Groups in Data: An Introduction to Cluster Analysis
  • [22] Kipf T. N., 2016, NIPS WORKSH BAYES DE
  • [23] ImageNet Classification with Deep Convolutional Neural Networks
    Krizhevsky, Alex
    Sutskever, Ilya
    Hinton, Geoffrey E.
    [J]. COMMUNICATIONS OF THE ACM, 2017, 60 (06) : 84 - 90
  • [24] Overlapping Attributed Graph Clustering using Mixed strategy games
    Kumar, Mayank
    Gupta, Ruchir
    [J]. APPLIED INTELLIGENCE, 2021, 51 (08) : 5299 - 5313
  • [25] A Community Detection Method for Social Network Based on Community Embedding
    Li, Meizi
    Lu, Shuyi
    Zhang, Lele
    Zhang, Yuping
    Zhang, Bo
    [J]. IEEE TRANSACTIONS ON COMPUTATIONAL SOCIAL SYSTEMS, 2021, 8 (02) : 308 - 318
  • [26] Local gap density for clustering high-dimensional data with varying densities
    Li, Ruijia
    Yang, Xiaofei
    Qin, Xiaolong
    Zhu, William
    [J]. KNOWLEDGE-BASED SYSTEMS, 2019, 184
  • [27] Amazon.com recommendation - Item-to-item collaborative filtering
    Linden, G
    Smith, B
    York, J
    [J]. IEEE INTERNET COMPUTING, 2003, 7 (01) : 76 - 80
  • [28] INFORMATION PERCOLATION AND CUTOFF FOR THE STOCHASTIC ISING MODEL
    Lubetzky, Eyal
    Sly, Allan
    [J]. JOURNAL OF THE AMERICAN MATHEMATICAL SOCIETY, 2016, 29 (03) : 729 - 774
  • [29] MacQueen J., 1967, P 5 BERK S MATH STAT, V1, P281, DOI DOI 10.1007/S11665-016-2173-6
  • [30] Finding community structure in networks using the eigenvectors of matrices
    Newman, M. E. J.
    [J]. PHYSICAL REVIEW E, 2006, 74 (03)