关键词: 分布式机器学习平台, 参数服务器, 大数据处理系统, 分布式机器学习系统

Abstract:

[Objective] Real-world data becomes much more complex, sparse and high-dimensional for the big data shock in this era. According to this, modern ML models are designed in a deep, complicated way, which arises challenges when designing a distributed machine learning (ML) system. Though researchers have developed many efficient centralized ML systems like TensorFlow, PyTorch and XGBoost, these systems suffer from the following two problems: (1) They cannot integrate well with existing big data systems, (2) they are not general enough and are usually designed for specific ML models. [Methods] To tackle these challenges, we introduce Angel ⁺, a large-scale ML platform based on parameter servers. [Results] With the power of parameter servers, Angel ⁺can efficiently support existing big data systems and ML systems without neither breaking the core of big data systems, Apache Spark for instance, nor degrades the computation efficiency of current ML frameworks like PyTorch. Furthermore, Angel ⁺ provides algorithms like model averaging, gradient compression and heterogeneous-aware stochastic gradient descent, to deal with the huge communication cost and the straggler problem in distributed training process. [Conclusions] We also enhance the usability of Angel ⁺by providing efficient implementation for many ML models. We conduct extensive experiments to demonstrate the superiority of Angel ⁺.

Key words: machine learning platform, parameter servers, big data systems, distributed machine learning systems