Enhancing k-Means Algorithm with Tensor Processing Unit

Enhancing k-Means Algorithm with Tensor Processing Unit

Abstract

Clustering in Data Mining is the process of discovering groups of similar objects in data. The k-Means clustering algorithm, is designed to partition data into k distinct groups or clusters. With recent growth in data production, the need to scale up existing algorithms and computational ability has increased. Google introduced the Tensor Processing Unit (TPU), a powerful hardware, to meet the growing computational needs of modern technologies. In this paper, we aim to enhance the k-Means algorithm with the use of the Google TPU in terms of runtime while preserving the quality of the clustering results. We developed two versions that distribute training on the TPU in two different ways. The clustering results of the versions have advantages that complement each other in terms of runtime and accuracy.

Grafik Top
Authors
  • Mummoju, Pranava
  • Wolff, Anna
  • Perdacher, Martin
  • Plant, Claudia
  • Böhm, Christian
Grafik Top
Shortfacts
Category
Paper in Conference Proceedings or in Workshop Proceedings (Paper)
Event Title
2022 IEEE International Conference on Big Data (IEEE BigData 2022)
Divisions
Data Mining and Machine Learning
Subjects
Parallele Datenverarbeitung
Event Location
Osaka, Japan
Event Type
Conference
Event Dates
17-20 Dec 2022
Date
17 December 2022
Export
Grafik Top