PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimization
Not on the map. Why?
Abstract (arXiv)
We study gradient compression methods to alleviate the communication bottleneck in data-parallel distributed optimization. Despite the significant attention received, current compression schemes either do not scale well or fail to achieve the target test accuracy. We propose a new low-rank gradient compressor based on power iteration that can i) compress gradients rapidly, ii) efficiently aggregate the compressed gradients using all-reduce, and iii) achieve test performance on par with SGD. The proposed algorithm is the only method evaluated that achieves consistent wall-clock speedups when benchmarked against regular SGD with an optimized communication backend. We demonstrate reduced training times for convolutional networks as well as LSTMs on common datasets. Our code is available at https://github.com/epfml/powersgd.
Key facts. Cited by 238 papers in the map (data through 14 Oct 2026). First posted on arXiv 31 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 40; 2022: 32; 2023: 36; 2024: 39; 2025: 39; 2026: 52.
238 papers.
Sort:
By area:
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2021A Field Guide to Federated Optimizationcited by 315
- 2021Sustainable AI: Environmental Implications, Challenges and Opportunitiescited by 242
- 2024GEAR: An Efficient KV Cache Compression Recipe for Near-Lossless Generative Inference of LLMcited by 158
- 2021FedPara: Low-Rank Hadamard Product for Communication-Efficient Federated Learningcited by 141
- 2021Large Scale Private Learning via Low-rank Reparametrizationcited by 88
- 2021Do Not Let Privacy Overbill Utility: Gradient Embedding Perturbation for Private Learningcited by 86
- 2022Sharper Convergence Guarantees for Asynchronous SGD for Distributed and Federated Learningcited by 79
- 20211-bit Adam: Communication Efficient Large-Scale Training with Adam's Convergence Speedcited by 75
- 2025Dion: Distributed Orthonormalized Updatescited by 68
- 2021Provably Secure Federated Learning against Malicious Clientscited by 63
- 2021Adaptive Quantization of Model Updates for Communication-Efficient Federated Learningcited by 63
38 more
- 2021Pufferfish: Communication-efficient Models At No Extra Costcited by 61
- 2023SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficientcited by 59
- 2021DRIVE: One-bit Distributed Mean Estimationcited by 52
- 2021FedNL: Making Newton-Type Methods Applicable to Federated Learningcited by 52
- 2022Generalizing to New Physical Systems via Context-Informed Dynamics Modelcited by 50
- 2021Rethinking gradient sparsification as total error minimizationcited by 41
- 2022A Survey on Gradient Inversion: Attacks, Defenses and Future Directionscited by 40
- 2022Compute-Efficient Deep Learning: Algorithmic Trends and Opportunitiescited by 38
- 2024LDAdam: Adaptive Optimization from Low-Dimensional Gradient Statisticscited by 38
- 2025MuonBP: Faster Muon via Block-Periodic Orthogonalizationcited by 35
- 2022Lower Bounds and Nearly Optimal Algorithms in Distributed Learning with Communication Compressioncited by 33
- 2021An Efficient Statistical-based Gradient Compression Technique for Distributed Training Systemscited by 32
- 2021Permutation Compressors for Provably Faster Distributed Nonconvex Optimizationcited by 32
- 2022Sparse Random Networks for Communication-Efficient Federated Learningcited by 32
- 2023Momentum Provably Improves Error Feedback!cited by 31
- 2023DoCoFL: Downlink Compression for Cross-Device Federated Learningcited by 28
- 2025Streaming DiLoCo with overlapping communication: Towards a Distributed Free Lunchcited by 28
- 2022projUNN: efficient method for training deep networks with unitary matricescited by 27
- 2025Communication-Efficient Language Model Training Scales Reliably and Robustly: Scaling Laws for DiLoCocited by 27
- 2021On the Utility of Gradient Compression in Distributed Training Systemscited by 26
- 20211-bit LAMB: Communication Efficient Large-Scale Large-Batch Training with LAMB's Convergence Speedcited by 24
- 2021ScaleCom: Scalable Sparsified Gradient Compression for Communication-Efficient Distributed Trainingcited by 22
- 2023Reducing Communication for Split Learning by Randomized Top-k Sparsificationcited by 22
- 2021Smoothness Matrices Beat Smoothness Constants: Better Communication Compression Techniques for Distributed Optimizationcited by 21
- 2022FedShuffle: Recipes for Better Use of Local Work in Federated Learningcited by 21
- 2021Trade-offs of Local SGD at Scale: An Empirical Studycited by 20
- 2022Alpa: Automating Inter- and Intra-Operator Parallelism for Distributed Deep Learningcited by 20
- 2022Communication Acceleration of Local Gradient Methods via an Accelerated Primal-Dual Algorithm with Inexact Proxcited by 19
- 2022GradSkip: Communication-Accelerated Local Gradient Methods with Better Computational Complexitycited by 18
- 2024Grass: Compute Efficient Low-Memory LLM Training with Structured Sparse Gradientscited by 18
- 2025MuLoCo: Muon is a practical inner optimizer for DiLoCocited by 18
- 2023z-SignFedAvg: A Unified Stochastic Sign-based Compression for Federated Learningcited by 17
- 2025Generalized Gradient Norm Clipping & Non-Euclidean (L0,L1)-Smoothnesscited by 17
- 2022Recycling Model Updates in Federated Learning: Are Gradient Subspaces Low-Rank?cited by 16
- 2021Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guaranteescited by 15
- 2022FedLite: A Scalable Approach for Federated Learning on Resource-constrained Clientscited by 15
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 15
- 2023EControl: Fast Distributed Optimization with Compression and Error Controlcited by 15
50 most cited of 238 All 238 (JSON)
- 2026Echo: Error-Compensated 4-Bit Tensor-Parallel Communication for LLM Training and Inference9 Oct 2026
- 2026AutoLoCo: Communication Efficient Distributed LLM Training via Adaptive Synchronization29 Sep 2026
- 2026Stiefel-AdamW: Geometry-Aware AdamW for Linear Factorization Blocks17 Sep 2026
- 2026Compressed Active Subspaces for Scalable Bayesian Inference17 Sep 2026
- 2026Communication-Efficient LLM Adaptation over Decentralized GPU Meshes13 Sep 2026
- 2026PACE: Propagation-Aware Collaborative Correction for One-Shot Personalized Federated Graph Learning4 Sep 2026
- 2026Distributed Training using an Intelligent Network26 Aug 2026
- 2026Measuring Structured Predictability in Neural Training Dynamics: A Cross-Regime Study16 Aug 2026
- 2026Capacity Confounds and Coverage Guarantees in Adaptive Sub-model Federated Learning7 Aug 2026
- 2026Incremental Aggregation on the Grassmannian for Asynchronous Eigenspace Computation5 Aug 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026GeoFP8: Geometry-Aware FP8 Gradient Compression for Distributed LLM Training8 Jul 2026
38 more
- 2026Entropy-Guided Tensor Compression for Multimodal Federated Learning on Edge Devices7 Jul 2026
- 2026SCAPE: Accurate and Efficient LLM Training with Extreme Sparse Communication2 Jul 2026
- 2026NEURON-Fabric: Architecture-Runtime Co-Design for Controlled Low-Bit Gradient Communication24 Jun 2026
- 2026Decentralised AI Training and Inference with BlockTrain23 Jun 2026
- 2026When to use what Schatten-p norm in deep learning?13 Jun 2026
- 2026Learned Subspace Compression for Communication-Efficient Pipeline Parallelism3 Jun 2026
- 2026Don't Let a Few Network Failures Slow the Entire AllReduce1 Jun 2026
- 2026Decentralized Parameter-Free Online Learning with Compressed Gossip27 May 2026
- 2026Unextractable Protocol Models: Collaborative Training and Inference without Weight Materialization22 May 2026
- 2026Factor Augmented High-Dimensional SGD19 May 2026
- 2026Can Visual Mamba Improve AI-Generated Image Detection? An In-Depth Investigation14 May 2026
- 2026Accelerating Zeroth-Order Spectral Optimization with Partial Orthogonalization from Power Iteration9 May 2026
- 2026SparseRL-Sync: Lossless Weight Synchronization with ~100x Less Communication8 May 2026
- 2026Orth-Dion: Eliminating Geometric Mismatch in Distributed Low-Rank Spectral Optimization7 May 2026
- 2026SignMuon: Communication-Efficient Distributed Muon Optimization4 May 2026
- 2026DBLP: Phase-Aware Bounded-Loss Transport for Burst-Resilient Distributed ML Training3 May 2026
- 2026FED-FSTQ: Fisher-Guided Token Quantization for Communication-Efficient Federated Fine-Tuning of LLMs on Edge Devices28 Apr 2026
- 2026TACO: Efficient Communication Compression of Intermediate Tensors for Scalable Tensor-Parallel LLM Training27 Apr 2026
- 2026FedSLoP: Memory-Efficient Federated Learning with Low-Rank Gradient Projection27 Apr 2026
- 2026Exploiting Correlations in Federated Learning: Opportunities and Practical Limitations16 Apr 2026
- 2026Towards The Implicit Bias on Multiclass Separable Data Under Norm Constraints24 Mar 2026
- 2026Scalable Cross-Facility Federated Learning for Scientific Foundation Models on Multiple Supercomputers20 Mar 2026
- 2026QuantFL: Sustainable Federated Learning for Edge IoT via Pre-Trained Model Quantisation18 Mar 2026
- 2026NuMuon: Nuclear-Norm-Constrained Muon for Compressible LLM Training4 Mar 2026
- 2026FlashOptim: Optimizers for Memory-Efficient Training26 Feb 2026
- 2026Scaling State-Space Models on Multiple GPUs with Tensor Parallelism24 Feb 2026
- 2026Exploring New Frontiers in Vertical Federated Learning: the Role of Saddle Point Reformulation17 Feb 2026
- 2026Stabilizing Native Low-Rank LLM Pretraining12 Feb 2026
- 2026DynamiQ: Accelerating Gradient Synchronization using Compressed Multi-hop All-reduce9 Feb 2026
- 2026From O(mn) to O(r2): Two-Sided Low-Rank Communication for Adam in Distributed Training with Memory Efficiency8 Feb 2026
- 2026LoRDO: Distributed Low-Rank Optimization with Infrequent Communication4 Feb 2026
- 2026Understanding and Exploiting Weight Update Sparsity for Communication-Efficient Distributed RL3 Feb 2026
- 2026Zero Sum SVD: Balancing Loss Sensitivity for Low Rank LLM Compression2 Feb 2026
- 2026NSC-SL: A Bandwidth-Aware Neural Subspace Compression for Communication-Efficient Split Learning2 Feb 2026
- 2026AsyncMesh: Fully Asynchronous Optimization for Data and Pipeline Parallelism30 Jan 2026
- 2026LoRIF: Low-Rank Influence Functions for Scalable Training Data Attribution29 Jan 2026
- 2026Convergence Analysis of Randomized Subspace Normalized SGD under Heavy-Tailed Noise28 Jan 2026
- 2026Ordered Local Momentum for Asynchronous Distributed Learning under Arbitrary Delays18 Jan 2026
50 newest of 238 All 238 (JSON)
- 2023Knowledge Distillation Performs Partial Variance Reductioncited by 11
- 2023Knowledge Distillation Performs Partial Variance Reduction27 May 2023
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2021A Field Guide to Federated Optimizationcited by 315
- 2021Sustainable AI: Environmental Implications, Challenges and Opportunitiescited by 242
- 2021FedPara: Low-Rank Hadamard Product for Communication-Efficient Federated Learningcited by 141
- 2021Large Scale Private Learning via Low-rank Reparametrizationcited by 88
- 2021Do Not Let Privacy Overbill Utility: Gradient Embedding Perturbation for Private Learningcited by 86
- 2022Sharper Convergence Guarantees for Asynchronous SGD for Distributed and Federated Learningcited by 79
- 20211-bit Adam: Communication Efficient Large-Scale Training with Adam's Convergence Speedcited by 75
- 2025Dion: Distributed Orthonormalized Updatescited by 68
- 2021Provably Secure Federated Learning against Malicious Clientscited by 63
- 2021Adaptive Quantization of Model Updates for Communication-Efficient Federated Learningcited by 63
- 2021Pufferfish: Communication-efficient Models At No Extra Costcited by 61
12 most cited of 229 All 229 (JSON)
- 2026Echo: Error-Compensated 4-Bit Tensor-Parallel Communication for LLM Training and Inference9 Oct 2026
- 2026AutoLoCo: Communication Efficient Distributed LLM Training via Adaptive Synchronization29 Sep 2026
- 2026Stiefel-AdamW: Geometry-Aware AdamW for Linear Factorization Blocks17 Sep 2026
- 2026Compressed Active Subspaces for Scalable Bayesian Inference17 Sep 2026
- 2026Communication-Efficient LLM Adaptation over Decentralized GPU Meshes13 Sep 2026
- 2026PACE: Propagation-Aware Collaborative Correction for One-Shot Personalized Federated Graph Learning4 Sep 2026
- 2026Distributed Training using an Intelligent Network26 Aug 2026
- 2026Measuring Structured Predictability in Neural Training Dynamics: A Cross-Regime Study16 Aug 2026
- 2026Capacity Confounds and Coverage Guarantees in Adaptive Sub-model Federated Learning7 Aug 2026
- 2026Incremental Aggregation on the Grassmannian for Asynchronous Eigenspace Computation5 Aug 2026
- 2026Federated Lightweight Fine-Tuning20 Jul 2026
- 2026GeoFP8: Geometry-Aware FP8 Gradient Compression for Distributed LLM Training8 Jul 2026
12 newest of 229 All 229 (JSON)