Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Curvature-Guided Sheaf Diffusion for Unsupervised Community Detection on Heterophilic Graphs

About

Detecting communities in heterophilic graphs -- where connected nodes often belong to different classes -- is hard for unsupervised methods: classical modularity and spectral methods are feature agnostic, while deep graph-clustering methods rely on contrastive or generative machinery that is opaque. We propose Curvature-Guided Sheaf Diffusion (CGSD), a fully unsupervised community-detection algorithm that uses the discrete Forman--Ricci curvature of each edge as its single topological signal, propagated through every stage of an end-to-end pipeline. CGSD makes three concrete contributions: (i)~a curvature-gated sheaf-diffusion encoder that gates edge messages by $\sigma(\kappa_e)$ and is trained from three label-free structural losses (modularity, anti-collapse, curvature-weighted reconstruction); (ii)~a curvature-aware spectral clusterer (CSpec) that re-weights the $k$-NN affinity of the embedding by $\sigma(\alpha \kappa_{e^*})$ before Ng--Jordan--Weiss; and (iii)~a unified label-free evaluation against nine truly-unsupervised baselines. On five heterophilic benchmarks (Cora, Cornell, Texas, Wisconsin, Chameleon), CGSD wins outright on Wisconsin and Chameleon and is competitive on the remaining three against nine unsupervised baselines. The gain over the strongest baseline is driven by the clusterer, not the encoder: on the same embedding, CSpec improves mean NMI from $0.091$ with $K$-Means to $0.107$ ($+15\%$, paired $t$-test $p=0.008$). The mechanism is interpretable: intra-community and inter-community curvature distributions are visibly separated. Code is open-sourced at https://github.com/woodywff/cgsd.

Feifan Wang• 2026

Related benchmarks

TaskDatasetResultRank
Graph ClusteringChameleon
NMI0.158
21
Community DetectionWisconsin
NMI18.2
11
Community DetectionCora
NMI5
11
Community DetectionCornell
NMI9.3
11
Community DetectionTexas
NMI5.4
11
Graph ClusteringRoman-Empire
NMI13.5
10
ClusteringActor
Mean NMI6.1
2
Clusteringamazon-ratings
Mean NMI9.2
2
Clusteringtolokers
Mean NMI0.018
2
Clusteringquestions
Mean NMI0.011
2
Showing 10 of 10 rows

Other info

Follow for update