A proximal subgradient method for nonconvex stochastic optimization under the Kurdyka-Łojasiewicz condition

arXiv:2608.05460 2026 Optimization 1 ideas extracted · analyzed Aug 31, 2026

What the math gives to ML

The paper provides a constructive way to combine stochastic sample refinement, nonsmooth proximal updates, and Armijo-style backtracking without assuming weak convexity of the regularizer or a uniformly bounded stochastic-oracle variance. The transferable asset is the local sufficient-decrease guarantee for a proximal subgradient step: sufficiently small steps satisfy a quadratic decrease condition even when the empirical loss is nonconvex and nonsmooth. In neural-network training, this suggests an adaptive-batch proximal optimizer that increases batch size only when line-search tests are unreliable, rather than imposing a fixed batch-growth schedule. The method is especially suitable for exact proximal regularizers such as group sparsity, structured pruning, or hard parameter constraints.

Ideas from this paper

Unverified 2026

Adaptive-Batch Proximal Armijo Training

Replace a fixed-batch SGD or proximal-gradient update by a stochastic proximal-subgradient step whose step size is backtracked against an empirical sufficient-decrease condition. If the condition is too noisy or repeatedly fails, enlarge the batch and retry; otherwise retain the current batch, allowing sample size to grow only when needed.

Useful6/10
Difficulty5/10
Novelty5/10
Paper: A proximal subgradient method for nonconvex stochastic optimization under the Kurdyka-Łojasiewicz condition arXiv:2608.05460