English

Deep Modularity Networks with Diversity-Preserving Regularization

Machine Learning 2025-11-04 v2

Abstract

Graph clustering plays a crucial role in graph representation learning but often faces challenges in achieving feature-space diversity. While Deep Modularity Networks (DMoN) leverage modularity maximization and collapse regularization to ensure structural separation, they lack explicit mechanisms for feature-space separation, assignment dispersion, and assignment-confidence control. We address this limitation by proposing Deep Modularity Networks with Diversity-Preserving Regularization (DMoN-DPR), which introduces three novel regularization terms: distance-based for inter-cluster separation, variance-based for per-cluster assignment dispersion, and an assignment-entropy penalty with a small positive weight, encouraging more confident assignments gradually. Our method significantly enhances label-based clustering metrics on feature-rich benchmark datasets (paired two-tailed t-test, p0.05p\leq0.05), demonstrating the effectiveness of incorporating diversity-preserving regularizations in creating meaningful and interpretable clusters.

Keywords

Cite

@article{arxiv.2501.13451,
  title  = {Deep Modularity Networks with Diversity-Preserving Regularization},
  author = {Yasmin Salehi and Dennis Giannacopoulos},
  journal= {arXiv preprint arXiv:2501.13451},
  year   = {2025}
}

Comments

Accepted at the 39th Conference on Neural Information Processing Systems (NeurIPS 2025) Workshop: New Perspectives in Graph Machine Learning (NPGML)

R2 v1 2026-06-28T21:14:30.281Z