Adaptive Gradient Quantization for Data-Parallel SGD
Not on the map. Why?
Abstract (arXiv)
Many communication-efficient variants of SGD use gradient quantization schemes. These schemes are often heuristic and fixed over the course of training. We empirically observe that the statistics of gradients of deep models change during the training. Motivated by this observation, we introduce two adaptive quantization schemes, ALQ and AMQ. In both schemes, processors update their compression schemes in parallel by efficiently computing sufficient statistics of a parametric distribution. We improve the validation accuracy by almost 2% on CIFAR-10 and 1% on ImageNet in challenging low-cost communication setups. Our adaptive methods are also significantly more robust to the choice of hyperparameters.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). First posted on arXiv 23 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 7; 2023: 11; 2024: 5; 2025: 5; 2026: 3.
43 papers.
Sort:
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021NeRV: Neural Representations for Videoscited by 255
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2021Adaptive Quantization of Model Updates for Communication-Efficient Federated Learningcited by 60
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2021Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devicescited by 36
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guaranteescited by 29
- 2021Lossless Compression of Efficient Private Local Randomizerscited by 27
- 2024Differentiable Weightless Neural Networkscited by 27
- 2022lo-fi: distributed fine-tuning without communicationcited by 19
- 2022Adaptive Compression for Communication-Efficient Distributed Trainingcited by 14
31 more
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 14
- 2023Fast Optimal Locally Private Mean Estimation via Random Projectionscited by 14
- 2023Quantized Distributed Training of Large Models with Convergence Guaranteescited by 13
- 2024SDP4Bit: Toward 4-bit Communication Quantization in Sharded Data Parallelism for LLM Trainingcited by 13
- 2023Communication Compression for Byzantine Robust Learning: New Efficient Algorithms and Improved Ratescited by 12
- 2025Low-Precision Training of Large Language Models: Methods, Challenges, and Opportunitiescited by 11
- 2021Theoretically Better and Numerically Faster Distributed Optimization with Smoothness-Aware Quantization Techniquescited by 9
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Surveycited by 9
- 2023Federated Learning under Covariate Shifts with Generalization Guaranteescited by 6
- 2021Distributed Online Learning for Joint Regret with Communication Constraintscited by 5
- 2022MixTailor: Mixed Gradient Aggregation for Robust Learning Against Tailored Attackscited by 5
- 2021CGX: Adaptive System Support for Communication-Efficient Deep Learningcited by 4
- 2023FedREP: A Byzantine-Robust, Communication-Efficient and Privacy-Preserving Framework for Federated Learningcited by 3
- 2021DeepReduce: A Sparse-tensor Communication Framework for Distributed Deep Learningcited by 2
- 2021Fundamental limits of over-the-air optimization: Are analog schemes optimal?cited by 2
- 2023Inshrinkerator: Compressing Deep Learning Training Checkpoints via Dynamic Quantizationcited by 2
- 2023Self-Distilled Quantization: Achieving High Compression Rates in Transformer-Based Language Modelscited by 2
- 2023Distributed Extra-gradient with Optimal Complexity and Communication Guaranteescited by 2
- 2024Optimal and Near-Optimal Adaptive Vector Quantizationcited by 2
- 2025On the Gradient Complexity of Private Optimization with Private Oraclescited by 2
- 2022Empirical Analysis on Top-k Gradient Sparsification for Distributed Deep Learning in a Supercomputing Environmentcited by 1
- 2022L-GreCo: Layerwise-Adaptive Gradient Compression for Efficient and Accurate Deep Learningcited by 1
- 2023CORE: Common Random Reconstruction for Distributed Optimization with Provable Low Communication Complexitycited by 1
- 2025Addressing Label Shift in Distributed Learning via Entropy Regularizationcited by 1
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averagingcited by 1
- 2026LoKA: Low-precision Kernel Applications for Recommendation Models At Scalecited by 1
- 2026Entropy-Constrained Adaptive Stochastic Quantizationcited by 1
- 2022Variance Reduction is an Antidote to Byzantines: Better Rates, Weaker Assumptions and Communication Compression as a Cherry on the Topcited by 0
- 2023Contractive error feedback for gradient compressioncited by 0
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMscited by 0
- 2026Inner Product Aware Quantization: Provably Fast, Accurate, and Adaptive Algorithmscited by 0
- 2026Entropy-Constrained Adaptive Stochastic Quantization10 Aug 2026
- 2026Inner Product Aware Quantization: Provably Fast, Accurate, and Adaptive Algorithms29 May 2026
- 2026LoKA: Low-precision Kernel Applications for Recommendation Models At Scale11 May 2026
- 2025On the Gradient Complexity of Private Optimization with Private Oracles17 Nov 2025
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMs31 May 2025
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averaging20 May 2025
- 2025Low-Precision Training of Large Language Models: Methods, Challenges, and Opportunities2 May 2025
- 2025Addressing Label Shift in Distributed Learning via Entropy Regularization4 Feb 2025
- 2024SDP4Bit: Toward 4-bit Communication Quantization in Sharded Data Parallelism for LLM Training20 Oct 2024
- 2024Differentiable Weightless Neural Networks14 Oct 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Survey8 Apr 2024
31 more
- 2024Optimal and Near-Optimal Adaptive Vector Quantization5 Feb 2024
- 2023Contractive error feedback for gradient compression13 Dec 2023
- 2023Communication Compression for Byzantine Robust Learning: New Efficient Algorithms and Improved Rates15 Oct 2023
- 2023CORE: Common Random Reconstruction for Distributed Optimization with Provable Low Communication Complexity23 Sep 2023
- 2023Distributed Extra-gradient with Optimal Complexity and Communication Guarantees17 Aug 2023
- 2023Self-Distilled Quantization: Achieving High Compression Rates in Transformer-Based Language Models12 Jul 2023
- 2023Inshrinkerator: Compressing Deep Learning Training Checkpoints via Dynamic Quantization20 Jun 2023
- 2023Federated Learning under Covariate Shifts with Generalization Guarantees8 Jun 2023
- 2023Fast Optimal Locally Private Mean Estimation via Random Projections7 Jun 2023
- 2023FedREP: A Byzantine-Robust, Communication-Efficient and Privacy-Preserving Framework for Federated Learning9 Mar 2023
- 2023Quantized Distributed Training of Large Models with Convergence Guarantees5 Feb 2023
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression24 Jan 2023
- 2022Adaptive Compression for Communication-Efficient Distributed Training31 Oct 2022
- 2022L-GreCo: Layerwise-Adaptive Gradient Compression for Efficient and Accurate Deep Learning31 Oct 2022
- 2022lo-fi: distributed fine-tuning without communication19 Oct 2022
- 2022Empirical Analysis on Top-k Gradient Sparsification for Distributed Deep Learning in a Supercomputing Environment18 Sep 2022
- 2022MixTailor: Mixed Gradient Aggregation for Robust Learning Against Tailored Attacks16 Jul 2022
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guarantees2 Jun 2022
- 2022Variance Reduction is an Antidote to Byzantines: Better Rates, Weaker Assumptions and Communication Compression as a Cherry on the Top1 Jun 2022
- 2021CGX: Adaptive System Support for Communication-Efficient Deep Learning16 Nov 2021
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AI11 Nov 2021
- 2021What Do We Mean by Generalization in Federated Learning?27 Oct 2021
- 2021NeRV: Neural Representations for Videos26 Oct 2021
- 2021Fundamental limits of over-the-air optimization: Are analog schemes optimal?11 Sep 2021
- 2021Theoretically Better and Numerically Faster Distributed Optimization with Smoothness-Aware Quantization Techniques7 Jun 2021
- 2021A Survey of Quantization Methods for Efficient Neural Network Inference25 Mar 2021
- 2021Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices4 Mar 2021
- 2021Lossless Compression of Efficient Private Local Randomizers24 Feb 2021
- 2021Distributed Online Learning for Joint Regret with Communication Constraints15 Feb 2021
- 2021Adaptive Quantization of Model Updates for Communication-Efficient Federated Learning8 Feb 2021
- 2021DeepReduce: A Sparse-tensor Communication Framework for Distributed Deep Learning5 Feb 2021
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021NeRV: Neural Representations for Videoscited by 255
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2021Adaptive Quantization of Model Updates for Communication-Efficient Federated Learningcited by 60
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2021Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devicescited by 36
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guaranteescited by 29
- 2021Lossless Compression of Efficient Private Local Randomizerscited by 27
- 2024Differentiable Weightless Neural Networkscited by 27
- 2022lo-fi: distributed fine-tuning without communicationcited by 19
- 2022Adaptive Compression for Communication-Efficient Distributed Trainingcited by 14
12 most cited of 43 All 43 (JSON)
- 2026Entropy-Constrained Adaptive Stochastic Quantization10 Aug 2026
- 2026Inner Product Aware Quantization: Provably Fast, Accurate, and Adaptive Algorithms29 May 2026
- 2026LoKA: Low-precision Kernel Applications for Recommendation Models At Scale11 May 2026
- 2025On the Gradient Complexity of Private Optimization with Private Oracles17 Nov 2025
- 2025It Takes a Good Model to Train a Good Model: Generalized Gaussian Priors for Optimized LLMs31 May 2025
- 2025Layer-wise Quantization for Quantized Optimistic Dual Averaging20 May 2025
- 2025Low-Precision Training of Large Language Models: Methods, Challenges, and Opportunities2 May 2025
- 2025Addressing Label Shift in Distributed Learning via Entropy Regularization4 Feb 2025
- 2024SDP4Bit: Toward 4-bit Communication Quantization in Sharded Data Parallelism for LLM Training20 Oct 2024
- 2024Differentiable Weightless Neural Networks14 Oct 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Survey8 Apr 2024
12 newest of 43 All 43 (JSON)