mlsanityData to 14 Oct 2026

Theoretical Foundations of Communication-Efficient, Robust, and Practical Distributed and Federated Optimization

arXiv 2608.06563 ↗First posted 6 Aug 2026cs.LG math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Machine learning and optimization have advanced together, with practical demands motivating new theory and theoretical breakthroughs enabling new applications. Modern large-scale training relies on classical optimization principles, but the constraints of distributed systems require these foundations to be reconsidered. This thesis addresses seven challenges at the intersection of theory and practice, focusing on key bottlenecks in federated learning and distributed optimization. First, we introduce ProxSkip and prove that local gradient steps can accelerate communication, providing a theoretical foundation for this widely used heuristic. Second, we develop Variance Reduced ProxSkip, which eliminates the neighborhood error of stochastic local updates while balancing communication and local computation. Third, we show that local steps retain their communication acceleration under partial client participation. Fourth, we prove that server-side stepsizes and sampling without replacement improve convergence in heterogeneous settings. Fifth, for Random Reshuffling, we demonstrate that compressing gradient differences rather than gradients yields better theoretical and practical performance. Sixth, we establish that Byzantine robustness and partial participation can be achieved simultaneously using gradient-difference clipping. Finally, we develop the first theoretical framework for low-rank adaptation based on randomized asymmetric chains, providing new insights into fine-tuning large models. Across these contributions, we introduce novel algorithmic frameworks, establish sharp guarantees under realistic assumptions, and support the theory with numerical experiments.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 83 of its 301 references that are in the map. First posted on arXiv 6 Aug 2026.

Builds on

83 of 301 references.

References · 83

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  2. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  3. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  4. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  5. 2017Decoupled Weight Decay Regularizationcited by 1,831
  6. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  7. 2016An overview of gradient descent optimization algorithmscited by 1,033
  8. 2021Towards a Unified View of Parameter-Efficient Transfer Learningcited by 872
  9. 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
  10. 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 674
  11. 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
  12. 2024Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Surveycited by 656
38 more
  1. 2016Federated Optimization: Distributed Machine Learning for On-Device Intelligencecited by 564
  2. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
  3. 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 341
  4. 2024LoRA Learns Less and Forgets Lesscited by 328
  5. 2021A Field Guide to Federated Optimizationcited by 315
  6. 2023Parameter-Efficient Fine-Tuning Methods for Pretrained Language Models: A Critical Review and Assessmentcited by 230
  7. 2020Federated Learning of a Mixture of Global and Local Modelscited by 210
  8. 2020Client Selection in Federated Learning: Convergence Analysis and Power-of-Choice Selection Strategiescited by 198
  9. 2015Federated Optimization:Distributed Optimization Beyond the Datacentercited by 195
  10. 2020Privacy and Robustness in Federated Learning: Attacks and Defensescited by 166
  11. 2024Improving LoRA in Privacy-preserving Federated Learningcited by 161
  12. 2019On the Convergence of Local Descent Methods in Federated Learningcited by 153
  13. 2019Distributed Learning with Compressed Gradient Differencescited by 112
  14. 2019Natural Compression for Distributed Deep Learningcited by 90
  15. 2024Asymmetry in Low-Rank Adapters of Foundation Modelscited by 86
  16. 2020On the Convergence of SGD with Biased Gradientscited by 83
  17. 2019Unified Optimal Analysis of the (Stochastic) Gradient Methodcited by 82
  18. 2024Chain of LoRA: Efficient Fine-tuning of Language Models via Residual Learningcited by 82
  19. 2019First Analysis of Local GD on Heterogeneous Datacited by 81
  20. 2021On Large-Cohort Training for Federated Learningcited by 74
  21. 2019Stochastic Distributed Learning with Gradient Quantization and Variance Reductioncited by 65
  22. 2019SGD without Replacement: Sharper Rates for General Smooth Convex Functionscited by 61
  23. 2023High-Probability Bounds for Stochastic Optimization and Variational Inequalities: the Case of Unbounded Variancecited by 61
  24. 2018Distributed learning with compressed gradientscited by 55
  25. 2021FedNL: Making Newton-Type Methods Applicable to Federated Learningcited by 52
  26. 2020Bidirectional compression in heterogeneous settings for distributed or federated learning with partial participation: tight convergence guaranteescited by 46
  27. 2018On the Fenchel Duality between Strong Convexity and Lipschitz Continuous Gradientcited by 43
  28. 2021Federated Learning Meets Natural Language Processing: A Surveycited by 43
  29. 2020A Reputation Mechanism Is All You Need: Collaborative Fairness and Adversarial Robustness in Federated Learningcited by 41
  30. 2020On the Outsized Importance of Learning Rates in Local Update Methodscited by 36
  31. 2021Minibatch vs Local SGD with Shuffling: Tight Convergence Bounds and Beyondcited by 36
  32. 2016Parallel SGD: When does averaging help?cited by 32
  33. 2021The Min-Max Complexity of Distributed Stochastic Convex Optimization with Intermittent Communicationcited by 31
  34. 2022Byzantine-Robust Decentralized Learning via ClippedGossipcited by 31
  35. 2021Removing Data Heterogeneity Influence Enhances Network Topology Dependence of Decentralized SGDcited by 29
  36. 2022RandProx: Primal-Dual Optimization Algorithms with Randomized Proximal Updatescited by 28
  37. 2020Communication-Efficient Distributed Deep Learning: A Comprehensive Surveycited by 26
  38. 2019Communication-Efficient Local Decentralized SGD Methodscited by 23

50 most cited of 83 All 83 (JSON)

Built on by

Not cited yet.