arrow
返回

SIM: Accelerating Distributed DNN Training by Exploring Gradient Similarity

delete2025-11-01
delete0
PRE
AI
Y
Ye Jin
Y
Yijun Li
W
Wenliang Li
X
Xiaojuan Lu
Q
Qichen Su
黄
黄家玮 (Jiawei Huang) *
王
王健鑫 (Jianxin Wang)
DOI:10.1109/TON.2025.3633713delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Synchronous stochastic gradient descent (SSGD) has been widely used in distributed deep learning. However, since the local gradients need to be shared among workers at every iteration, SSGD performance is significantly influenced by network bottlenecks caused by either heterogeneous environment or bandwidth contention. To solve this problem, asynchronous parallel (ASP) strategy allows each worker to update parameters independently without synchronization, while suffering from accuracy loss and convergence inefficiency. In this paper, we propose a novel similarity-based synchronization scheme called SIM, which mitigates the impact of network bottlenecks and ensures convergence efficiency. Specifically, SIM reduces the number of aggregation workers based on the gradient similarity between global and local gradients, therefore shrinking the waiting time for the stragglers. We provide a theoretical analysis of convergence efficiency and conduct large-scale testbed experiments on CIFAR-10 and SQUAD dataset. The experimental results show that SIM reduces the convergence time of four classical deep learning models by up to 40%.
Keyword:
Training
Convergence
Synchronization
Computational modeling
Deep learning
Accuracy
Artificial neural networks
Artificial intelligence
Stochastic processes
Servers
Distributed deep learning
straggler
synchronization scheme
gradient similarity

期刊

I
IEEE Transactions on Networking
IF:
0
论文数:
551
被引数:
0

机构

G
Guangxi University
学者数:
4.1K
论文数: 1.3K
被引数: 3.2W
C
central south university
学者数:
2.2W
论文数: 6.3K
被引数: 3
引用论文

引用论文

S-Caffe
err2017-01-26
err0
PREAI
errAmmar Ahmad Awan; Khaled Hamidouche; Jahanzeb Maqbool Hashmi; Dhabaleswar K. Panda
err分享
err收藏
DeepCaps: Going Deeper With Capsule Networks
err2019-06-01
err0
errOAAI
errJathushan Rajasegaran; Vinoj Jayasundara; Sandaru Jayasekara; Hirunima Jayasekara; Suranga Seneviratne; Ranga Rodrigo
err分享
err收藏
err分享
err收藏
err分享
err收藏
Exploring the Limits of Weakly Supervised Pretraining
err2018-10-09
err0
PREAI
errDhruv Mahajan; Ross Girshick; Vignesh Ramanathan; Kaiming He; Manohar Paluri; Yixuan Li; Ashwin Bharambe; Laurens van der Maaten
err分享
err收藏
Learned Gradient Compression for Distributed Deep Learning
err2022-12-01
err32
errOAAI
errAbrahamyan, Lusine; Chen, Yiming; Bekoulis, Giannis; Deligiannis, Nikos
err分享
err收藏
PyTorch distributed
err2020-09-14
err0
PREAI
errShen Li; Yanli Zhao; Rohan Varma; Omkar Salpekar; Pieter Noordhuis; Teng Li; Adam Paszke; Jeff Smith; Brian Vaughan; Pritam Damania; Soumith Chintala
err分享
err收藏
err分享
err收藏
学者 查看更多内容