arrow
返回

Staleness-Reduction Mini-Batch K-Means

delete2024-10-01
delete4
PRE
AI
X
Xueying Zhu
J
Jie Sun
Z
Zhenhao He
J
Jiantong Jiang
Z
Zeke Wang *
DOI:10.1109/TNNLS.2023.3279122delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
K-means (km) is a clustering algorithm that has been widely adopted due to its simple implementation and high clustering quality. However, the standard km suffers from high computational complexity and is therefore time-consuming. Accordingly, the mini-batch (mbatch) km is proposed to significantly reduce computational costs in a manner that updates centroids after performing distance computations on just a mbatch, rather than a full batch, of samples. Even though the mbatch km converges faster, it leads to a decrease in convergence quality because it introduces staleness during iterations. To this end, in this article, we propose the staleness-reduction mbatch (srmbatch) km, which achieves the best of two worlds: low computational costs like the mbatch km and high clustering quality like the standard km. Moreover, srmbatch still exposes massive parallelism to be efficiently implemented on multicore CPUs and many-core GPUs. The experimental results show that srmbatch can converge up to 40x-130x faster than mbatch when reaching the same target loss, and srmbatch is able to reach 0.2%-1.7% lower final loss than that of mbatch.
Keyword:
Clustering
K-means (km)
machine learning
staleness-reduction

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

E
ETH Zurich
学者数:
3.0W
论文数: 2.4W
被引数: 8.4W
S
swiss federal institutes of technology domain
学者数:
9.0W
论文数: 8.0W
被引数: 163
Z
zhejiang university
学者数:
17.7W
论文数: 12.1W
被引数: 152
学者 查看更多机构
引用论文

引用论文

err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
err分享
err收藏
Toward Sustainability in Concrete Industry by Using Of Solid Wastes from Palm Oil Industry
err2016-07-13
err0
errOAAI
errAlireza Javadi Pordesari; Salmaliza Salleh; Payam Shafigh; Hilmi Bin Mahmuda
err分享
err收藏
Cerebral morphology and dopamine D2/D3 receptor distribution in humans: A combined [18F]fallypride and voxel-based morphometry study
err2009-05-15
err0
errOAAI
errNeil D. Woodward; David H. Zald; Zhaohua Ding; Patrizia Riccardi; M. Sib Ansari; Ronald M. Baldwin; Ronald L. Cowan; Rui Li; Robert M. Kessler
err分享
err收藏
Spectral Analysis of the EEG
err2008-02-20
err0
PREAI
errG. Dumermuth; L. Molinari
err分享
err收藏
学者 查看更多内容