Variance-Reduced Fast Krasnoselkii-Mann Methods for Finite-Sum Root-Finding Problems
math.OC, stat.ML
Submitted: 2024-06-04
Updated: 2026-09-09
Comments: 31 pages, 2 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: We propose a new class of fast Krasnoselkii--Mann methods with variance reduction to solve a finite-sum co-coercive equation Gx = 0.
Terminology
Abstract
We propose a new class of fast Krasnoselkii--Mann methods with variance reduction to solve a finite-sum co-coercive equation Gx = 0. Our algorithm is single-loop and leverages a new family of unbiased variance-reduced estimators specifically designed for a wider class of root-finding algorithms. Our method achieves both O(1/k 2) and o(1/k 2) last-iterate convergence rates in terms of E[Gx k 2], where k is the iteration counter and E[times] is the total expectation. We also establish almost sure o(1/k 2) convergence rates and the almost sure convergence of iterates x k to a solution of Gx=0. We instantiate our framework for two prominent estimators: SVRG and SAGA. By an appropriate choice of parameters, both variants attain an oracle complexity of O(n + n 2/3ε-1) to reach an ε-solution, where n represents the number of summands in the finite-sum operator G. Furthermore, under σ-strong quasi-monotonicity, our method achieves a linear convergence rate and an oracle complexity of O(n+ n, n 2/3κ (ε)), where κ:= L/σ. We extend our approach to solve a class of finite-sum inclusions (possibly nonmonotone), demonstrating that our schemes retain the same theoretical guarantees as in the equation setting. Finally, numerical experiments validate our algorithms and demonstrate their promising performance compared to state-of-the-art methods.
Sources
- Forward-backward-forward methods with variance reduction for stochastic variational inequalities
- Fast Krasnosel'skii-Mann algorithm with a convergence rate of the fixed point iteration of $o\left(\frac{1}{k}\right)$
- SMART: The Stochastic Monotone Aggregated Root-Finding Algorithm
- An Accelerated Variance Reduced Extra-Point Approach to Finite-Sum VI and Optimization
- Halpern-Type Accelerated and Splitting Algorithms For Monotone Inclusions
Related papers
- Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noise
- Adam-HNAG: A Convergent Reformulation of Adam with Accelerated Rate
- Incremental Learning in Mirror Flows
- Online Control via Counterfactual Tracking
- Asynchronous Replanning in Two Population Linear Quadratic Mean Field Games: Information Requirements and Stability
- Petrov-Galerkin operator inference with application to stability-encouraging identification