English

$t$-$k$-means: A Robust and Stable $k$-means Variant

Machine Learning 2021-02-02 v4 Machine Learning

Abstract

kk-means algorithm is one of the most classical clustering methods, which has been widely and successfully used in signal processing. However, due to the thin-tailed property of the Gaussian distribution, kk-means algorithm suffers from relatively poor performance on the dataset containing heavy-tailed data or outliers. Besides, standard kk-means algorithm also has relatively weak stability, i.e.i.e. its results have a large variance, which reduces its credibility. In this paper, we propose a robust and stable kk-means variant, dubbed the tt-kk-means, as well as its fast version to alleviate those problems. Theoretically, we derive the tt-kk-means and analyze its robustness and stability from the aspect of the loss function and the expression of the clustering center, respectively. Extensive experiments are also conducted, which verify the effectiveness and efficiency of the proposed method. The code for reproducing main results is available at \url{https://github.com/THUYimingLi/t-k-means}.

Keywords

Cite

@article{arxiv.1907.07442,
  title  = {$t$-$k$-means: A Robust and Stable $k$-means Variant},
  author = {Yiming Li and Yang Zhang and Qingtao Tang and Weipeng Huang and Yong Jiang and Shu-Tao Xia},
  journal= {arXiv preprint arXiv:1907.07442},
  year   = {2021}
}

Comments

Accepted by the ICASSP 2021. The first two authors contributed equally to this work

R2 v1 2026-06-23T10:23:02.813Z