基于Julia语言的K-means算法性能优化

摘要

Julia语言是一种新出现的动态高性能脚本语言,支持多处理器或多核编程,程序性能高.K-means算法是最常用和最典型的聚类算法之一,但其在大数据量情形下需要提升性能.本文实现了K-means算法的Julia基本程序;然后,根据Julia语言的类型系统和语法机制特点,提出了程序优化方案;最后,针对多核CPU,实现了K-means的并行算法.在12核CPU上的实验表明,优化后的串行程序接近C语言程序性能,并行程序性能取得4.07倍加速比.与其它动态语言相比,Julia的K-means算法性能有明显优势.