mlsanityData to 14 Oct 2026

Mirror Descent Beyond Euclidean Stability: An Exponential Separation in Initialization Sensitivity

arXiv 2606.11431 ↗First posted 9 Jun 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Mirror Descent (MD) extends Gradient Descent (GD) beyond Euclidean geometry and has recently reappeared as a lens for KL-regularized policy optimization in reinforcement learning and LLM post-training. This raises a basic robustness question, crucial to reproducibility and reliability: how sensitive are MD dynamics to their inputs? We focus on initialization, often itself a pretrained or previously aligned model. Quadratic-regularized MD, including GD and Mahalanobis geometries, is well-known to be stable for convex smooth objectives. We show a sharp contrast: once the regularizer is non-quadratic, MD can be exponentially more sensitive to initialization than GD, even with a well-conditioned regularizer in Euclidean norm. We give a three-dimensional construction with a convex, smooth objective and a strongly convex, smooth, well-conditioned regularizer where an initial varepsilon perturbation is quickly amplified to minpolylog-1(1/varepsilon), varepsilon eOmega(η T) after T iterations of MD with step size η. For canonical KL-regularized MD on the simplex, we show that even linear objectives can amplify an initial varepsilon perturbation exponentially fast in high-dimensional or near-boundary regimes. Finally, we show that adding a Bregman regularization term toward an anchor point can stabilize the dynamics while largely preserving the optimization guarantees, and that the choice of anchor is crucial: anchoring at the initialization only partially mitigates the instability, whereas anchoring at a fixed point yields a more stable mechanism.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 30 of its 42 references that are in the map. First posted on arXiv 9 Jun 2026.

Builds on

30 of 42 references.

References · 30

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  5. 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
  6. 2019Introduction to Online Convex Optimizationcited by 803
  7. 2019Solving Rubik's Cube with a Robot Handcited by 734
  8. 2014Convex Optimization: Algorithms and Complexitycited by 575
  9. 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
  10. 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 263
  11. 2020Stability of Stochastic Gradient Descent on Nonsmooth Convex Lossescited by 173
  12. 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 119
18 more
  1. 2016Relatively-Smooth Convex Optimization by First-Order Methods, and Applicationscited by 108
  2. 2025The Art of Scaling Reinforcement Learning Compute for LLMscited by 101
  3. 2021Private Stochastic Convex Optimization: Optimal Rates in ℓ1 Geometrycited by 83
  4. 2019Stochastic Mirror Descent on Overparameterized Nonlinear Models: Convergence, Implicit Regularization, and Generalizationcited by 40
  5. 2021SGD Generalizes Better Than GD (And Regularization Doesn't Help)cited by 38
  6. 2022Stability vs Implicit Bias of Gradient Methods on Separable Data and Beyondcited by 29
  7. 2022Mirror Descent Maximizes Generalized Margin and Can Be Implemented Efficientlycited by 26
  8. 2021Stochastic Mirror Descent: Convergence Analysis and Adaptive Variants via the Mirror Stochastic Polyak Stepsizecited by 22
  9. 2022Reproducibility in Optimization: Theoretical Framework and Limitscited by 21
  10. 2021Algorithmic Instabilities of Accelerated Gradient Descentcited by 16
  11. 2023A Unified Approach to Controlling Implicit Regularization via Mirror Descentcited by 14
  12. 2022Uniform Stability for First-Order Empirical Risk Minimizationcited by 8
  13. 2024The Sample Complexity of Gradient Descent in Stochastic Convex Optimizationcited by 6
  14. 2025Flat Minima and Generalization: Insights from Stochastic Convex Optimizationcited by 4
  15. 2023Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimizationcited by 3
  16. 2024Black-Box Uniform Stability for Non-Euclidean Empirical Risk Minimizationcited by 3
  17. 2025Rapid Overfitting of Multi-Pass Stochastic Gradient Descent in Stochastic Convex Optimizationcited by 3
  18. 2025The Hidden Cost of Approximation in Online Mirror Descentcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025On the Effect of Regularization in Policy Mirror Descent
  2. 2026Approximation of Log-Partition Function in Policy Mirror Descent Induces Implicit Regularization for LLM Post-Training
  3. 2025StaQ: a Finite Memory Approach to Discrete Action Policy Mirror Descent
  4. 2025Mirror Descent Actor Critic via Bounded Advantage Learning
  5. 2022Robust Imitation via Mirror Descent Inverse Reinforcement Learning
  6. 2024Functional Acceleration for Policy Mirror Descent
  7. 2026Beyond State-Wise Mirror Descent: Offline Policy Optimization with Parametric Policies
  8. 2024Learning mirror maps in policy mirror descent