xinwangliu_homepage/DFCN: Deep Fusion Clustering Network - DFCN - OpenI

关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

zhaijianyang 9feb07d8be first commit		2 年前
data	first commit	2 年前

figure	first commit	2 年前

graph	first commit	2 年前

model	first commit	2 年前

AE.py	first commit	2 年前

DFCN.py	first commit	2 年前

Deep Fusion Clustering Network.pdf	first commit	2 年前

IGAE.py	first commit	2 年前

LICENSE	first commit	2 年前

README.md	first commit	2 年前

load_data.py	first commit	2 年前

main.py	first commit	2 年前

opt.py	first commit	2 年前

train.py	first commit	2 年前

utils.py	first commit	2 年前

Paper

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]

Installation

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

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) or SDCN.

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:

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.

Visualization

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{TuDeep,
  title={Deep Fusion Clustering Network},
  author={Tu, Wenxuan and Zhou, Sihang and Liu, Xinwang and Guo, Xifeng and Cai, Zhiping and zhu, En and Cheng, Jieren},
  booktitle={The Thirty-Fifth AAAI Conference on Artificial Intelligence},
  year={2021}
}

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.

--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

Deep Fusion Clustering Network

Pickle Text

1240758677@qq.com

代码中如何访问数据资源