Abstract:Deep neural networks (DNNs) have gained tremendous attractions as compelling solutions for applications such as image classification, object detection, speech recognition, and so forth. Its great success comes with excessive trainings to make sure the model accuracy is good enough for those applications. Nowadays, it becomes challenging to train a DNN model because of 1) the model size and data size keep increasing, which usually needs more iterations to train; 2) DNN algorithms evolve rapidly, which requires the training phase to be short for a quick deployment. To address those challenges, distributed training platforms have been proposed to leverage massive server nodes for training, with the hope of significant training time reduction. Therefore, scalability is a critical performance metric to evaluate a distributed training platform. Nevertheless, our analysis reveals that traditional server clusters have poor scalability for training due to the traffic congestions within the server and beyond. The intra-server traffic on the I/O fabric can result in severe congestions and skewed quality of service as high performance devices are competing with each other. Moreover, the traffic congestions on the Ethernet for inter-server communication could also incur significant performance degradation. In this work, we devise a novel distributed training platform, EFLOPS, that adopts an algorithm and system co-design methodology to achieve good scalability. A new server architecture is proposed to alleviate the intra-server congestions. Moreover, a new network topology, BiGraph, is proposed to divide the network into two separate parts, so that there is always a direct connection between any nodes from different parts. Finally, accompany with BiGraph, a topology-aware allreduce algorithm is proposed to eliminate the traffic congestion on the direct connection. The experimental results show that eliminating the congestions on network interface can gain up to 11.3xcommunication speedup. The proposed algorithm and topology can provide further improvement up to 6.08x. The overall performance of ResNet-50 training achieves near-linear scalability, and is competitive to the top-rankings of MLPerf results.

PerfTop: Towards Performance Prediction of Distributed Learning over General Topology

A Topology-Aware Performance Prediction Model for Distributed Deep Learning on GPU Clusters

Runtime Performance Prediction for Deep Learning Models with Graph Neural Network.

Proteus: Simulating the Performance of Distributed DNN Training

Towards Universal Performance Modeling for Machine Learning Training on Multi-GPU Platforms

A Survey on Performance Modeling and Prediction for Distributed DNN Training

TopoOpt: Co-optimizing Network Topology and Parallelization Strategy for Distributed Training Jobs

A Generic Performance Model for Deep Learning in a Distributed Environment

Performance Modeling and Evaluation of Distributed Deep Learning Frameworks on GPUs

Prediction of GPU Failures Under Deep Learning Workloads

Topology-aware GPU scheduling for learning workloads in cloud environments

Optimizing Network Performance for Distributed DNN Training on GPU Clusters: ImageNet/AlexNet Training in 1.5 Minutes

Forecasting GPU Performance for Deep Learning Training and Inference

EFLOPS: Algorithm and System Co-Design for a High Performance Distributed Training Platform

Near-Optimal Topology-adaptive Parameter Synchronization in Distributed DNN Training

dPRO: A Generic Profiling and Optimization System for Expediting Distributed DNN Training

HPH: Hybrid Parallelism on Heterogeneous Clusters for Accelerating Large-scale DNNs Training.

From promise to practice: realizing high-performance decentralized training

Optimizing execution for pipelined‐based distributed deep learning in a heterogeneously networked GPU cluster

Poseidon: An Efficient Communication Architecture for Distributed Deep Learning on GPU Clusters