← Search

Conglong Li

7 accepted papers

2024

DeepSpeed Data Efficiency: Improving Deep Learning Model Quality and Training Efficiency via Efficient Data Sampling and Routing

AAAI 2024technical

Recent advances on deep learning models come at the price of formidable training cost. The increasing model size is one of the root causes, but another less-emphasized fact is that data scale is actually increasing at a similar speed as model scale, and the training cost is proportional to both of t…

2023

Maximizing Communication Efficiency for Large-scale Training via 0/1 Adam

ICLR 2023poster

1-bit gradient compression and local steps are two representative techniques that enable drastic communication reduction in distributed SGD. Their benefits, however, remain an open question on Adam-based large model pre-training (e.g. BERT and GPT). In this paper, we demonstrate the non-linearity in…

2022

DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scale

ICML 2022spotlight

As the training of giant dense models hits the boundary on the availability and capability of the hardware resources today, Mixture-of-Experts (MoE) models have become one of the most promising model architectures due to their significant training cost reduction compared to quality-equivalent dense…

2022

The Stability-Efficiency Dilemma: Investigating Sequence Length Warmup for Training GPT Models

NeurIPS 2022accept

Recent works have demonstrated great success in pre-training large-scale autoregressive language models (e.g., GPT-3) on massive GPUs. To reduce the wall-clock training time, a common practice is to increase the batch size and learning rate. However, such practice is often brittle and leads to a so-…

2022

XTC: Extreme Compression for Pre-trained Transformers Made Simple and Efficient

NeurIPS 2022accept

Extreme compression, particularly ultra-low bit precision (binary/ternary) quantization, has been proposed to fit large NLP models on resource-constraint devices. However, to preserve the accuracy for such aggressive compression schemes, cutting-edge methods usually introduce complicated compressio…

Cited by 29SourcePDFScholar
2022

ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformers

NeurIPS 2022accept

How to efficiently serve ever-larger trained natural language models in practice has become exceptionally challenging even for powerful cloud servers due to their prohibitive memory/computation requirements. In this work, we present an efficient and affordable post-training quantization approach to…

2021

1-bit Adam: Communication Efficient Large-Scale Training with Adam’s Convergence Speed

ICML 2021spotlight

Scalable training of large models (like BERT and GPT-3) requires careful optimization rooted in model design, architecture, and system capabilities. From a system standpoint, communication has become a major bottleneck, especially on commodity systems with standard TCP interconnects that offer limit…