acceptodds
Under review as a conference paper at ICLR 2027

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.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.