Abstract:With the advent of the big data era, the amounts of sampling data and the dimensions of data features are rapidly growing. It is highly desired to enable fast and efficient clustering of unlabeled samples based on feature similarities. As a fundamental primitive for data clustering, the k-means operation is receiving increasingly more attentions today. To achieve high performance k-means computations on modern multi-core/many-core systems, we propose a matrix-based fused framework that can achieve high performance by conducting computations on a distance matrix and at the same time can improve the memory reuse through the fusion of the distance-matrix computation and the nearest centroids reduction. We implement and optimize the parallel k-means algorithm on the SW26010 many-core processor, which is the major horsepower of Sunway TaihuLight. In particular, we design a task mapping strategy for load-balanced task distribution, a data sharing scheme to reduce the memory footprint and a register blocking strategy to increase the data locality. Optimization techniques such as instruction reordering and double buffering are further applied to improve the sustained performance. Discussions on block-size tuning and performance modeling are also presented. We show by experiments on both randomly generated and real-world datasets that our parallel implementation of k-means on SW26010 can sustain a double-precision performance of over 348.1 Gflops, which is 46.9% of the peak performance and 84% of the theoretical performance upper bound on a single core group, and can achieve a nearly ideal scalability to the whole SW26010 processor of four core groups. Performance comparisons with the previous state-of-the-art on both CPU and GPU are also provided to show the superiority of our optimized k-means kernel.

FT K-means: A High-Performance K-means on GPU with Fault Tolerance

Dvt-Pkm: An Improved Gpu Based Parallel K-Means Algorithm

K-Means on Commodity GPUs with CUDA

TurboFFT: Co-Designed High-Performance and Fault-Tolerant Fast Fourier Transform on GPUs

Anatomy of High-Performance GEMM with Online Fault Tolerance on GPUs

Cluster Center Initialization Parallel Algorithm for K-Means Algorithm

TurboFFT: A High-Performance Fast Fourier Transform with Fault Tolerance on GPU

Large-scale FFT on GPU clusters

FT-GEMM: A Fault Tolerant High Performance GEMM Implementation on x86 CPUs

A Fast Parallel Clustering Algorithm For Molecular Simulation Trajectories

An Efficient Parallelization Approach for Large-Scale Sparse Non-Negative Matrix Factorization Using Kullback-Leibler Divergence on Multi-GPU.

TiAcc: Triangle-inequality Based Hardware Accelerator for K-means on FPGAs

A Hybrid Gpu/Cpu Fft Library For Large Fft Problems

An Adaptive Algorithm of K-Means on Hsa Platform

Implementation of K-nearest Neighbor Algorithm Based on GPU

Research on K-means Clustering Parallel Algorithm Based on Sliding Gate Center Point Calculation

Parallelization of K-Means Clustering Algorithm for Data Mining

Enabling Highly Efficient K -Means Computations on the SW26010 Many-Core Processor of Sunway TaihuLight

Improved Parallel K-Means Clustering Based on CUDA

cuFINUFFT: a load-balanced GPU library for general-purpose nonuniform FFTs

High-performance K-means Implementation based on a Simplified Map-Reduce Architecture