A curated list of community detection research papers with implementations.

Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs (CIKM 2020)

A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).

Official PyTorch implementation of Superpoint Transformer introduced in [ICCV'23] "Efficient 3D Semantic Segmentation with Superpoint Transformer" and SuperCluster introduced in [3DV'24 Oral] "Scalable 3D Panoptic Segmentation As Superpoint Graph Clustering"

Code for our ECCV 2018 work.

[AAAI 2023] An official source code for paper Hard Sample Aware Network for Contrastive Deep Graph Clustering.

An implementation of "EdMot: An Edge Enhancement Approach for Motif-aware Community Detection" (KDD 2019)

A NetworkX implementation of Label Propagation from a "Near Linear Time Algorithm to Detect Community Structures in Large-Scale Networks" (Physical Review E 2008).

A Unified Library for Deep Graph Clustering

WWW2020-One2Multi Graph Autoencoder for Multi-view Graph Clustering

A NetworkX implementation of "Ego-splitting Framework: from Non-Overlapping to Overlapping Clusters" (KDD 2017).

An implementation of Chinese Whispers in Python.

Official implementation of our paper "Contrastive Deep Nonnegative Matrix Factorization for Community Detection" (ICASSP 2024)

Implementation of "Just Balance GNN" for graph classification and node clustering from the paper "Simplifying Clusterings with Graph Neural Networks".

Graph matching and clustering by comparing heat kernels via optimal transport.

Pytorch (PyG) and Tensorflow (Keras/Spektral) implementation of Total Variation Graph Neural Network (TVGNN), as presented at ICML 2023.

ACM MM 2023: Self-contrastive graph diffusion network

[KDD 2024] Revisiting Modularity Maximization for Graph Clustering: A Contrastive Learning Perspective

Community detection using attribute and structural similarities.