arrow
返回

PS plus : A Simple yet Effective Framework for Fast Training on Parameter Server

delete2022-12-01
delete3
PRE
AI
A
A-Long Jin
W
Wenchao Xu
Song Guo 封面图
Song Guo (Song Guo)
B
Bing Hu *
K
Kwan L. Yeung
DOI:10.1109/TPDS.2022.3200518delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
In distributed training, workers collaboratively refine the global model parameters by pushing their updates to the Parameter Server and pulling fresher parameters for the next iteration. This introduces high communication costs for training at scale, and incurs unproductive waiting time for workers. To minimize the waiting time, existing approaches overlap communication and computation for deep neural networks. Yet, these techniques not only require the layer-by-layer model structures, but also need significant efforts in runtime profiling and hyperparameter tuning. To make the overlapping optimization simple and generic, in this article, we propose a new Parameter Server framework. Our solution decouples the dependency between push and pull operations, and allows workers to eagerly pull the global parameters. This way, both push and pull operations can be easily overlapped with computations. Besides, the overlapping manner offers a different way to address the straggler problem, where the stale updates greatly retard the training process. In the new framework, with adequate information available to workers, they can explicitly modulate the learning rates for their updates. Thus, the global parameters can be less compromised by stale updates. We implement a prototype system in PyTorch and demonstrate its effectiveness on both CPU/GPU clusters. Experimental results show that our prototype saves up to 54% less time for each iteration and up to 37% fewer iterations for model convergence, achieving up to 2.86x speedup over widely-used synchronization schemes.
Keyword:
Training
Servers
Synchronization
Computational modeling
Hardware
Data models
Computational efficiency
Machine learning
distributed training
parameter server

期刊

IEEE Transactions on Parallel and Distributed Systems 封面图
IEEE Transactions on Parallel and Distributed Systems
IF:
6
论文数:
5.2K
被引数:
1.1W

机构

U
University of Hong Kong
学者数:
4.1W
论文数: 3.9W
被引数: 10.1W
H
hong kong polytechnic university
学者数:
3.0W
论文数: 4.1W
被引数: 921
Z
zhejiang university
学者数:
17.7W
论文数: 12.1W
被引数: 152
学者 查看更多机构
引用论文

引用论文

Rat plasma 25-hydroxyvitamin D3 binding protein: An inhibitor of the 25-hydroxyvitamin D3-1 α-hydroxylase
err1978-07-01
err0
PREAI
errJacob G. Ghazarian; Barbara Kream; Kathleen M. Botham; Michael W. Nickells; Hector F. DeLuca
err分享
err收藏
Acid catalysis in hydrolytic reactions水解反应中的酸催化
err1929-01-01
err0
PREAI
errJ. N. Brönsted; W. F. K. Wynne-Jones
err分享
err收藏
A mechanism of Pc 1 pearl formation based on the Alfvén sweep maser
err2000-03-01
err0
PREAI
errV.Y. Trakhtengerts; A.G. Demekhov; S.V. Polyakov; P.P. Belyaev; V.O. Rapoport
err分享
err收藏
Scheduling Policies for Federated Learning in Wireless Networks
err2020-01-01
err445
errOAAI
errYang, Howard H.; Liu, Zuozhu; Quek, Tony Q. S.; Poor, H. Vincent
err分享
err收藏
Floquet-Bloch operator for the Bose-Hubbard model with static field
err2003-11-25
err0
PREAI
errAndrey R. Kolovsky; Andreas Buchleitner
err分享
err收藏
学者 查看更多内容