arXiv · 2510.11312
Nonlinearly Preconditioned Gradient Methods: Momentum and Stochastic Analysis
Abstract
We study nonlinearly preconditioned gradient methods for smooth nonconvex optimization problems, focusing on sigmoid preconditioners that inherently perform a form of gradient clipping akin to the widely used gradient clipping technique. Building upon this idea, we introduce a novel heavy ball-type algorithm and provide convergence guarantees under a generalized smoothness condition that is less restrictive than traditional Lipschitz smoothness, thus covering a broader class of functions. Additionally, we develop a stochastic variant of the base method and study its convergence properties under different noise assumptions. We compare the proposed algorithms with baseline methods on diverse tasks from machine learning including neural network training.
Explore related subjects
Keep this discovery
Konstantinos Oikonomidis, Jan Quan, Panagiotis Patrinos. 2025-10-13. Nonlinearly Preconditioned Gradient Methods: Momentum and Stochastic Analysis. https://arxiv.org/abs/2510.11312
Cite the original work for its findings. Save a collection to share your selection of sources.