计算机与现代化

• 数据挖掘 • 上一篇    

基于Spark的分布式大数据机器学习算法

  

  1. (1.郑州大学信息工程学院,河南郑州450001;2.中国科学院计算技术研究所先进计算机系统研究中心,北京100190)
  • 收稿日期:2018-05-03 出版日期:2018-11-22 发布日期:2018-11-23
  • 作者简介:王芮(1996-),女,辽宁盘锦人,郑州大学信息工程学院本科生,研究方向:机器学习; 韩锐(1985-),男,湖北武汉人,中国科学院计算技术研究所先进计算机系统研究中心助理研究员,博士,研究方向:云计算与数据中心软件系统。

#br# Distributed Big Data Machine Learning Algorithms Based on Spark

  1. (1. School of Information Engineering, Zhengzhou University, Zhengzhou 450001, China;
    2. Institute of Computing Technology, Chinese Academy of Sciences, Beijing 100190, China)
  • Received:2018-05-03 Online:2018-11-22 Published:2018-11-23

摘要: 对于大数据而言,机器学习技术是不可或缺的;对于机器学习而言,大规模的数据可以提升模型的精准度。然而复杂的机器学习算法从时间和性能上都急需分布式内存计算这种关键技术。Spark分布式内存计算可以实现算法的并行操作,有利于机器学习算法处理大数据集。因此本文提出在Spark分布式内存环境下实现非线性机器学习算法,其中包括多层可变神经网络、BPPGD SVM、K-means,并在实现的基础上进行数据压缩、数据偏向抽样或者数据加载等方面的优化。为了实现充分配置资源批量运行脚本,本文也实现SparkML调度框架来调度以上优化算法。实验结果表明,优化后的3种算法平均误差降低了40%,平均时间缩短了90%。

关键词: 数据压缩, 偏向抽样, 随机梯度下降, 神经网络, 支持向量机

Abstract:  For big data, machine learning technology is a tool of analysis which is indispensable. For machine learning, more and more data may improve the accuracy of the model, however complex machine learning algorithms urgently require such key technologies as distributed memory computing in terms of time and performance. Spark distributed memory computing can implement the parallel operation of the algorithm, which is beneficial for machine learning algorithms to process large data sets. Therefore, this paper presents nonlinear machine learning algorithms in Spark distributed memory environment, including multi-layer variable neural network, BPPGD SVM, K-means. And we make optimizations about data compression, data bias sampling, or data loading based on the above implementation. At the same time, the SparkML scheduling framework is implemented to dispatch the above optimization algorithms. The experimental results show that the average error of the three optimized algorithms is reduced 40% and the average time is reduced 90%.

Key words: data compression, biased sampling, random gradient descent, neural network, support vector machine

中图分类号: