Weakly Convex Optimization on Manifolds with Heavy-Tailed Oracles
Abstract
Weakly convex optimization on manifolds underlies nonsmooth learning problems with orthogonality constraints. Existing Riemannian stochastic subgradient guarantees rely on bounded second moments; we study unbiased oracles with only a finite -th central moment, . Our main result is high-probability convergence of Riemannian stochastic subgradient descent (RSSG) on compact embedded manifolds with globally defined smooth retractions. Heavy-tailed steps can leave the region where local retraction bounds apply. We overcome this mismatch by controlling the iterates' actual movement on the compact manifold, without clipping or assuming finite oracle variance. Its oracle complexity is , recovering the classical order at . Two complementary variants bound every tangent step. GeoClip-RSSG couples clipping and stepsize selection to attain the same oracle exponent within a prescribed retraction neighborhood. GeoNorm-RSSG normalizes mini-batch subgradients, attaining the same iteration-rate order with parameter choices independent of , at an additional sample cost. Experiments support the predicted convergence trends and demonstrate the methods' value for subspace recovery from heavy-tailed data.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.