Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Deep Fusion Clustering Network

About

Deep clustering is a fundamental yet challenging task for data analysis. Recently we witness a strong tendency of combining autoencoder and graph neural networks to exploit structure information for clustering performance enhancement. However, we observe that existing literature 1) lacks a dynamic fusion mechanism to selectively integrate and refine the information of graph structure and node attributes for consensus representation learning; 2) fails to extract information from both sides for robust target distribution (i.e., "groundtruth" soft labels) generation. To tackle the above issues, we propose a Deep Fusion Clustering Network (DFCN). Specifically, in our network, an interdependency learning-based Structure and Attribute Information Fusion (SAIF) module is proposed to explicitly merge the representations learned by an autoencoder and a graph autoencoder for consensus representation learning. Also, a reliable target distribution generation measure and a triplet self-supervision strategy, which facilitate cross-modality information exploitation, are designed for network training. Extensive experiments on six benchmark datasets have demonstrated that the proposed DFCN consistently outperforms the state-of-the-art deep clustering methods.

Wenxuan Tu, Sihang Zhou, Xinwang Liu, Xifeng Guo, Zhiping Cai, En zhu, Jieren Cheng• 2020

Related benchmarks

TaskDatasetResultRank
Node ClusteringCora
NMI19.36
168
Node ClusteringCiteseer
NMI43.9
140
Graph ClusteringAMAP
Accuracy76.88
35
Graph ClusteringWiki
ARI17.17
27
Node ClusteringPubMed (test)
Accuracy68.89
25
ClusteringACM (test)
Accuracy (ACC)90.9
15
ClusteringDBLP (test)
Accuracy76
15
ClusteringCITE (test)
Accuracy69.5
15
ClusteringAMAP (test)
Accuracy76.88
15
Attributed Graph ClusteringCornell
Accuracy39.72
12
Showing 10 of 18 rows

Other info

Follow for update