Skip to content

The code of AGCN (Attention-driven Graph Clustering Network), which is accepted by ACM MM 2021.

Notifications You must be signed in to change notification settings

ZhihaoPENG-CityU/MM21---AGCN

Repository files navigation

Attention-driven Graph Clustering Network

Made with Python GitHub stars GitHub forks visitors

DOI: 10.1145/3474085.3475276

URL: https://dl.acm.org/doi/abs/10.1145/3474085.3475276

VIDEO: https://dl.acm.org/doi/abs/10.1145/3474085.3475276

We have added comments in the code, and the specific details can correspond to the explanation in the paper. Please get in touch with me ([email protected]) if you have any issues.

We appreciate it if you use this code and cite our related papers, which can be cited as follows,

@article{peng2023egrc,
title={EGRC-Net: Embedding-Induced Graph Refinement Clustering Network},
author={Peng, Zhihao and Liu, Hui and Jia, Yuheng and Hou, Junhui},
journal={IEEE Transactions on Image Processing},
volume={32},
pages={6457--6468},
year={2023},
publisher={IEEE} }

@article{peng2022deep,
title={Deep Attention-guided Graph Clustering with Dual Self-supervision},
author={Peng, Zhihao and Liu, Hui and Jia, Yuheng and Hou, Junhui},
journal={IEEE Transactions on Circuits and Systems for Video Technology},
year={2022},
publisher={IEEE} }

@inproceedings{peng2021attention,
title={Attention-driven graph clustering network},
author={Peng, Zhihao and Liu, Hui and Jia, Yuheng and Hou, Junhui},
booktitle={Proceedings of the 29th ACM International Conference on Multimedia},
pages={935--943},
year={2021} }

Environment

  • Python[3.6.12]
  • Pytorch[1.7.1]
  • GPU (GeForce RTX 2080 Ti) & (NVIDIA GeForce RTX 3090) & (Quadro RTX 8000)

Hyperparameters

  • The learning rates of USPS, HHAR, ACM, and DBLP datasets are set to 0.001, and the learning rates of Reuters and CiteSeer datasets are set to 0.0001. lambda1 and lambda2 are set to {1000, 1000} for USPS, {1, 0.1} for HHAR, {10, 10} for Reuters, and {0.1, 0.01} for graph datasets.

To run code

  • Step 1: set the hyperparameters for the specific dataset;
  • Step 2: python AGCN.py --name [data_name]
  • For examle, if u would like to run AGCN on the usps dataset, u need to
  • first set {1000, 1000} for USPS;
  • then run the command "python AGCN.py --name usps"

Evaluation

  • evaluation.py 👉 The commonly used clustering metrics, such as acc, nmi, ari, and f1, etc.
  • get_net_par_num.py 👉 Get the network parameters by `print(num_net_parameter(model))', where model is the designed network.

Data

Due to the limitation of GitHub, we share the data in [here].

Star History

Star History Chart

About

The code of AGCN (Attention-driven Graph Clustering Network), which is accepted by ACM MM 2021.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages