arrow
返回

XAgg: Accelerating Heterogeneous Distributed Training Through XDP-Based Gradient Aggregation

delete2024-06-01
delete1
PRE
AI
Q
Qianyu Zhang
G
Gongming Zhao
徐
徐宏力 (Hongli Xu) *
P
Peng Yang
DOI:10.1109/TNET.2023.3339524delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
With the growth of model/dataset/system size for distributed model training in datacenters, the widely used Parameter Server (PS) architecture suffers from communication bottleneck of gradient transmission. Recent works attempt to utilize programmable switches to implement in-network gradient aggregation and alleviate communication bottlenecks on PSs. Due to the limited on-chip memory of programmable switches, gradient transmission requires strict synchronization to achieve ideal aggregation performance. However, the distributed training system is usually heterogeneous in datacenters (e.g., computation and bandwidth heterogeneity), and the gradient will reach the aggregation nodes asynchronously, thereby seriously affecting the aggregation performance. To solve the above issue, we propose XAgg, which accelerates heterogeneous gradient aggregation by deploying the eXpress Data Path (XDP) based aggregator on servers. Specifically, the abundant idle memory on servers can cache the entire gradient, so as to effectively deal with asynchronous gradient transmission in heterogeneous scenarios. Moreover, XDP can provide high-performance and low-latency gradient aggregation. We conduct microbenchmark and testbed with real-world DNN models and datasets. Experimental results show that XAgg improves the gradient aggregation throughput by 3.3x compared with TCP-based aggregation, reaching 100 Gbps with 10 CPU cores. In addition, XAgg reduces communication time by 49%-82% compared with state-of-the-art solutions.
Keyword:
In-network gradient aggregation
distributed model training
eXpress data path (XDP)

期刊

I
IEEE-ACM Transactions on Networking
IF:
3.6
论文数:
4.4K
被引数:
9.5K

机构

C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
引用论文

引用论文

Differences in Lung Cancer Mortality Trends From 1986–2012 By Radon Risk Areas in British Columbia, Canada
err2014-05-01
err0
PREAI
errSarah B. Henderson; Stephen A. Rauch; Perry Hystad; Tom Kosatsky
err分享
err收藏
err分享
err收藏
Characteristics of Co-Allocated Online Services and Batch Jobs in Internet Data Centers: A Case Study From Alibaba Cloud
err2019-01-01
err59
errOAAI
errJiang, Congfeng; Han, Guangjie; Lin, Jiangbin; Jia, Gangyong; Shi, Weisong; Wan, Jian
err分享
err收藏
The GPU enters computing's mainstream
err2003-10-01
err0
PREAI
errM. Macedonia
err分享
err收藏
学者 查看更多内容