Residential College | false |
Status | 已發表Published |
Overlapping Communication with Computation in Parameter Server for Scalable DL Training | |
Wang,Shaoqi1; Pi,Aidi1; Zhou,Xiaobo1; Wang,Jun2; Xu,Cheng Zhong3 | |
2021-09-01 | |
Source Publication | IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS |
ISSN | 1045-9219 |
Volume | 32Issue:9Pages:2144-2159 |
Abstract | Scalability of distributed deep learning (DL) training with parameter server (PS) architecture is often communication constrained in large clusters. There are recent efforts that use a layer by layer strategy to overlap gradient communication with backward computation so as to reduce the impact of communication constraint on the scalability. However, the approaches could bring significant overhead in gradient communication. Meanwhile, they cannot be effectively applied to the overlap between parameter communication and forward computation. In this article, we propose and develop iPart, a novel approach that partitions communication and computation in various partition sizes to overlap gradient communication with backward computation and parameter communication with forward computation. iPart formulates the partitioning decision as an optimization problem and solves it based on a greedy algorithm to derive communication and computation partitions. We implement iPart in the open-source DL framework BigDL and perform evaluations with various DL workloads. Experimental results show that iPart improves the scalability of a cluster of 72 nodes by up to 94 percent over the default PS and 52 percent over the layer by layer strategy. |
Keyword | Backward Computation Forward Computation Gradient Communication Parameter Communication Parameter Server |
DOI | 10.1109/TPDS.2021.3062721 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering |
WOS Subject | Computer Science, Theory & Methods ; Engineering, Electrical & Electronic |
WOS ID | WOS:000631197900001 |
Publisher | IEEE COMPUTER SOC10662 LOS VAQUEROS CIRCLE, PO BOX 3014, LOS ALAMITOS, CA 90720-1314 |
Scopus ID | 2-s2.0-85102275748 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Zhou,Xiaobo |
Affiliation | 1.Department of Computer Science,University of Colorado,Colorado Springs,United States 2.Department of Electrical and Computer Engineering,University of Central Florida,Orlando,United States 3.Faculty of Science and Technology,University of Macau,Taipa,Macao |
Recommended Citation GB/T 7714 | Wang,Shaoqi,Pi,Aidi,Zhou,Xiaobo,et al. Overlapping Communication with Computation in Parameter Server for Scalable DL Training[J]. IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 2021, 32(9), 2144-2159. |
APA | Wang,Shaoqi., Pi,Aidi., Zhou,Xiaobo., Wang,Jun., & Xu,Cheng Zhong (2021). Overlapping Communication with Computation in Parameter Server for Scalable DL Training. IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 32(9), 2144-2159. |
MLA | Wang,Shaoqi,et al."Overlapping Communication with Computation in Parameter Server for Scalable DL Training".IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS 32.9(2021):2144-2159. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment