Abstract
This paper presents “Proactive Congestion Notification” (PCN), a congestion-avoidance technique for distributed deep learning (DDL). DDL is widely used to scale out and accelerate deep neural network training. In DDL, each worker trains a copy of the deep learning model with different training inputs and synchronizes the model gradients at the end of each iteration. However, it is well known that the network communication for synchronizing model parameters is the main bottleneck in DDL. Our key observation is that the DDL architecture makes each worker generate burst traffic every iteration, which causes network congestion and in turn degrades the throughput of DDL traffic. Based on this observation, the key idea behind PCN is to prevent potential congestion by proactively regulating the switch queue length before DDL burst traffic arrives at the switch, which prepares the switches for handling incoming DDL bursts. In our evaluation, PCN improves the throughput of DDL traffic by 72% on average.
Original language | English |
---|---|
Article number | 174 |
Pages (from-to) | 1-18 |
Number of pages | 18 |
Journal | Sensors (Switzerland) |
Volume | 21 |
Issue number | 1 |
DOIs | |
Publication status | Published - 2021 Jan 1 |
Keywords
- Congestion avoidance
- Deep learning
- Distributed deep learning
- Network congestion
- P4
- Proactive congestion notification
ASJC Scopus subject areas
- Analytical Chemistry
- Information Systems
- Atomic and Molecular Physics, and Optics
- Biochemistry
- Instrumentation
- Electrical and Electronic Engineering