Parameter-Insensitive Min Cut Clustering With Flexible Size Constrains

Feiping Nie, Fangyuan Xie, Weizhong Yu, Xuelong Li

Research output: Contribution to journalArticlepeer-review

5 Scopus citations

Abstract

Clustering is a fundamental topic in machine learning and various methods are proposed, in which K-Means (KM) and min cut clustering are typical ones. However, they may produce empty or skewed clustering results, which are not as expected. In KM, the constrained clustering methods have been fully studied while in min cut clustering, it still needs to be developed. In this paper, we propose a parameter-insensitive min cut clustering with flexible size constraints. Specifically, we add lower limitations on the number of samples for each cluster, which can perfectly avoid the trivial solution in min cut clustering. As far as we are concerned, this is the first attempt of directly incorporating size constraints into min cut. However, it is a NP-hard problem and difficult to solve. Thus, the upper limits is also added in but it is still difficult to solve. Therefore, an additional variable that is equivalent to label matrix is introduced in and the augmented Lagrangian multiplier (ALM) is used to decouple the constraints. In the experiments, we find that the our algorithm is less sensitive to lower bound and is practical in image segmentation. A large number of experiments demonstrate the effectiveness of our proposed algorithm.

Original languageEnglish
Pages (from-to)5479-5492
Number of pages14
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume46
Issue number8
DOIs
StatePublished - 2024

Keywords

  • Augmented Lagrangian multiplier (ALM)
  • clustering
  • empty cluster
  • flexible size constrains
  • min cut

Fingerprint

Dive into the research topics of 'Parameter-Insensitive Min Cut Clustering With Flexible Size Constrains'. Together they form a unique fingerprint.

Cite this