Stochastic extensions of mirror Polyak
Establish non-asymptotic convergence guarantees for stochastic extensions of mirror Polyak, including settings without interpolation and potentially using the primal-dual lifting technique to remove knowledge of the optimal value.
References
Extensions of these ideas to the stochastic setting remain open and of interest for applications in machine learning. The Polyak step-size has been applied successfully for finite-sum problems under interpolation, when all functions share a minimizer and an optimal value of 0, but is difficult to apply without interpolation. Existing results assuming interpolation or using the idealized variant of~\citet{gower2025analysis} would show mirror Polyak converges asymptotically as it leads to a contraction~\citep[][Prop.~4.13]{bauschke2003bregman}, but its non-asymptotic behavior is unknown. Our lifting technique might help improve existing schemes in the stochastic case~\citep{garrigos2023functionvaluelearning,jiang2023adaptive}.
We explore extensions of {PolyakEG} to stochastic settings and their limitations, and they raise concrete questions for future work. Can the deterministic algorithm's adaptivity to H"older continuity and $(L_0,L_1)$-Lipschitzness be extended to stochastic operators?
Which stability mechanisms can be incorporated into {DecPolyakSEG} to guarantee convergence for any stochastic monotone root-finding problems with adequate regularity assumptions?