Harmonic cut: An efficient and directly solved balanced graph clustering

Yu Duan, Feiping Nie, Rong Wang, Xuelong Li

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

In recent years, balanced clustering has garnered significant attention within its research domain. Previous studies have often employed various regularization terms to achieve balanced clustering. However, they inevitably bring extra hyper-parameters and suffer from high computational complexity. In this paper, we propose an efficient balanced graph cut model called Harmonic Cut. Harmonic cut is inspired by two key factors: harmonic mean and anchors’ labels are instances’ labels. The former elegantly derives a balance constraint. And the latter efficiently provides an idea for obtaining clustering labels. Unlike conventional spectral clustering methods, Harmonic cut directly acquires all cluster assignments, eliminating the need for additional k-means steps and significantly enhancing efficiency. We tackle the optimization problem associated with the Harmonic cut model using an efficient and heuristic algorithm, whose computational complexity scales linearly with the data size, denoted as n. Finally, extensive experiments conducted on various real-world datasets demonstrate that Harmonic cut not only achieves comparable or even superior results, but also ensures both balance and efficiency. The code is available at https://github.com/DuannYu/HarmonicCut.

Original languageEnglish
Article number127381
JournalNeurocomputing
Volume578
DOIs
StatePublished - 14 Apr 2024

Keywords

  • Balanced clustering
  • Clustering
  • Graph cut
  • Unsupervised learning

Fingerprint

Dive into the research topics of 'Harmonic cut: An efficient and directly solved balanced graph clustering'. Together they form a unique fingerprint.

Cite this