Home

Awesome

Paper

Made with Python GitHub stars GitHub forks visitors

Source code for the paper "Deep Fusion Clustering Network"<br>

W. Tu, S. Zhou, X. Liu, X. Guo, Z. Cai, E. Zhu, and J. Cheng.<br>

Accepted by AAAI2021. [Paper] [Poster] <br>

Installation

Clone this repo.

git clone https://github.com/WxTu/DFCN.git

Preparation

We adopt six datasets in total, including three graph datasets (ACM, DBLP, and CITE) and three non-graph datasets (USPS, HHAR, and REUT). To train a model on these datasets, please download them from Baidu Cloud (access code: 4622), and the demo files for model pre-training are provided in Baidu Cloud (access code: 4622).

Code Structure & Usage

Here we provide an implementation of Deep Fusion Clustering Network (DFCN) in PyTorch, along with an execution example on the DBLP dataset (due to file size limit). The repository is organised as follows:

Finally, main.py puts all of the above together and may be used to execute a full training run on DBLP.

<span id="jump2"></span>

Visualization

<div align=center><img width="800" height="330" src="./figure/2.jpg"/></div>

Contact

wenxuantu@163.com

Any discussions or concerns are welcomed!

Citation & License

If you use this code for your research, please cite our paper.

@inproceedings{DFCN2021,
  title={Deep Fusion Clustering Network},
  author={Wenxuan Tu and Sihang Zhou and Xinwang Liu and Xifeng Guo and Zhiping Cai and En Zhu and Jieren Cheng},
  booktitle={Proceedings of The Thirty-Fifth AAAI Conference on Artificial Intelligence},
  pages={9978-9987},
  year={2021}
}

All rights reserved. Licensed under the Apache License 2.0.

The code is released for academic research use only. For commercial use, please contact [wenxuantu@163.com].

Acknowledgement

D. Bo, X. Wang, C. Shi, et al. Structural Deep Clustering Network. In WWW, 2020.<br/> --https://github.com/bdy9527/SDCN

X. Guo, L. Gao, X. Liu, et al. Improved Deep Embedded Clustering with Local Structure Preservation. In IJCAI, 2017.<br/> --https://github.com/XifengGuo/IDEC

J. Xie, R. Girshick, and A. Farhadi. Unsupervised Deep Embedding for Clustering Analysis. In ICML, 2016.<br/> --https://github.com/vlukiyanov/pt-dec