Source code for the paper "Deep Fusion Clustering Network"
W. Tu, S. Zhou, X. Liu, X. Guo, Z. Cai, E. Zhu, and J. Cheng.
Accepted by AAAI2021. [Paper] [Poster]
Clone this repo.
git clone https://github.com/WxTu/DFCN.git- Windows 10 or Linux 18.04
- Python 3.7.5
- Pytorch (1.2.0+)
- Numpy 1.18.0
- Sklearn 0.21.3
- Torchvision 0.3.0
- Matplotlib 3.2.1
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).
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:
load_data.py: processes the dataset before passing to the network.DFCN.py: defines the architecture of the whole network.IGAE.py: defines the improved graph autoencoder.AE.py: defines the autoencoder.opt.py: defines some hyper-parameters.utils.py: defines the lr-policy, metrics, and others.train.py: the entry point for training and testing.
Finally, main.py puts all of the above together and may be used to execute a full training run on DBLP.
Any discussions or concerns are welcomed!
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].
D. Bo, X. Wang, C. Shi, et al. Structural Deep Clustering Network. In WWW, 2020.
--https://github.com/bdy9527/SDCN
X. Guo, L. Gao, X. Liu, et al. Improved Deep Embedded Clustering with Local Structure Preservation. In IJCAI, 2017.
--https://github.com/XifengGuo/IDEC
J. Xie, R. Girshick, and A. Farhadi. Unsupervised Deep Embedding for Clustering Analysis. In ICML, 2016.
--https://github.com/vlukiyanov/pt-dec