English

Community detection using fast low-cardinality semidefinite programming

Machine Learning 2020-12-07 v1 Optimization and Control

Abstract

Modularity maximization has been a fundamental tool for understanding the community structure of a network, but the underlying optimization problem is nonconvex and NP-hard to solve. State-of-the-art algorithms like the Louvain or Leiden methods focus on different heuristics to help escape local optima, but they still depend on a greedy step that moves node assignment locally and is prone to getting trapped. In this paper, we propose a new class of low-cardinality algorithm that generalizes the local update to maximize a semidefinite relaxation derived from max-k-cut. This proposed algorithm is scalable, empirically achieves the global semidefinite optimality for small cases, and outperforms the state-of-the-art algorithms in real-world datasets with little additional time cost. From the algorithmic perspective, it also opens a new avenue for scaling-up semidefinite programming when the solutions are sparse instead of low-rank.

Keywords

Cite

@article{arxiv.2012.02676,
  title  = {Community detection using fast low-cardinality semidefinite programming},
  author = {Po-Wei Wang and J. Zico Kolter},
  journal= {arXiv preprint arXiv:2012.02676},
  year   = {2020}
}

Comments

Accepted at NeurIPS'20. The code can be found at https://github.com/locuslab/sdp_clustering