Centered Permutation Prefixes for SGD with Random Reshuffling: Sharp Rates, Hölder Geometry, and Composite Proximal Extensions
Abstract: We study stochastic gradient descent with random reshuffling for finite sums [ F(x)=\frac1n\sum_{i=1}n f_i(x). ] For fresh reshuffling with a constant component stepsize, if each has an -Lipschitz gradient and the average is -strongly convex with a Lipschitz-continuous Hessian, we prove the last-epoch rate [ \mathbb E[F(y_K)-F(x_\star)] =\widetilde O!\left(T{-2}+n2T{-3}\right), \qquad T=nK, ] matching the known quadratic lower bound in its -dependence. The components may be nonconvex, and no componentwise Hessian continuity or separate bounded-iterate assumption is required. More generally, a -Hölder-continuous average Hessian adds only , so every preserves the quadratic rate. Under convex components, a decreasing-stepsize result removes the large-epoch requirement and recovers the same two-term scale once exceeds the condition-number scale. We also analyze epoch-wise ProxRR for . Writing for the composite minimizer and , we prove [ \mathbb E|y_K-x\dagger|2 =\widetilde O!\left( \frac{β\star2}{K2} +T{-2}+n2T{-3} +n{1+ν}T{-2-2ν} \right). ] For , we show that the splitting term is unavoidable and obtain a matching lower bound up to logarithms in the stated constant-stepsize regime.
Paper Prompts
Sign up for free to create and run prompts on this paper.