cs.LGJun 9, 2026

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

Authors: Shira Vansover-HagerMatan SchlisermanOfir SchlisselbergTomer Koren

Organizations: Blavatnik School of Computer Science and AI, Tel Aviv University · Google Research

Abstract

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 min{polylog1(1/ε),εeΩ(ηT)}\min\{\text{polylog}^{-1}(1/\varepsilon), \varepsilon e^{Ω(ηT)}\} after TT 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.

Explore similar work

CardsList