cs.LGOct 20, 2025

Stochastic Difference-of-Convex Optimization with Momentum

Authors: El Mahdi Chayti, Martin Jaggi

Organizations: Machine Learning and Optimization Laboratory (MLO), EPFL

Abstract

We study the online stochastic difference-of-convex (DC) program min⁡x∈RnF(x)=G(x)−H(x)+r1(x)−r2(x)\min_{x \in R^n} F(x) = G(x) - H(x) + r_1(x) - r_2(x), which covers sparse learning with nonconvex penalties, robust regression with smooth robust losses, and non-negative positive--unlabeled learning. We propose momentum-based DCA (MDCA), which combines a damped DCA step with a momentum estimator for ∇H\nabla H and uses a constant per-iteration batch size. With exact solutions of the convex subproblems, MDCA with Polyak momentum uses O(ε−4)\mathcal{O}( ε^{-4}) stochastic gradients of HH to reach εε-stationarity, and momentum-based variance reduction (MVR) uses O(ε−3)\mathcal{O}(ε^{-3}) under averaged smoothness, with a single O(ε−1)\mathcal O(ε^{-1}) warm-up batch in place of the periodic large batches of DCA-PAGE, whose εε-dependence it matches. We also introduce a single-loop variant (S-MDCA) that replaces the subproblem solve by one stochastic prox-gradient step; for every fixed smoothing parameter γγ for which the separately smoothed surrogate FγF_γ is bounded below, it reaches εε-stationarity of FγF_γ in O(ε−4)\mathcal{O}(ε^{-4}) iterations, which does not by itself imply stationarity of FF. A variance--drift trade-off lemma describes how the momentum parameter trades estimator variance against iterate-drift bias in steady-state upper bounds. Guarantees are given for two criteria: the critical-distance measure, and a DCA fixed-point residual that requires no smoothness of r2r_2 and vanishes exactly at DC-critical points; both give the same εε-dependence. Experiments on two LIBSVM classification datasets illustrate the methods' behaviour.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

May 28, 2026

MoSSP: A Momentum-Based Single-Loop Stochastic Penalty Method for Nonconvex Constrained DC-Regularized Optimization

In this paper, we study a structured class of nonconvex constrained stochastic problems with difference-of-convex (DC) regularization, where the feasible set is possibly nonconvex and the concave part of the DC regularizer is allowed to be nonsmooth. The fundamental challenge lies in maintaining feasibility for nonconvex constraints while achieving favorable oracle complexity. Although single-loop algorithms efficiently solve unconstrained DC optimization problems, their potential for constrained optimization with DC structure remains largely unexplored. To address this gap, we develop MoSSP, a Momentum-based Single-loop Stochastic Penalty method for such problems with provable complexity guarantees. The key idea is to apply a single stochastic proximal-gradient step to the Moreau envelope of the penalty plus the convex DC part, with the concave part's proximal mapping computed in parallel. We derive two algorithm variants: a Polyak-momentum version with O(ε−4)O(\varepsilon^{-4}) oracle complexity for finding stochastic ε\varepsilon-KKT points, and an improved O(ε−3)O(\varepsilon^{-3}) version incorporating recursive momentum. Experimental results demonstrate the effectiveness of the proposed algorithms.
Jun 18, 2024

Accelerated Stochastic Min-Max Optimization Based on Bias-corrected Momentum

Lower-bound analyses for nonconvex strongly-concave minimax optimization problems have shown that stochastic first-order algorithms require at least O(ε−4)\mathcal{O}(\varepsilon^{-4}) sample complexity to find an ε\varepsilon-stationary point. Some works indicate that this complexity can be improved to O(ε−3)\mathcal{O}(\varepsilon^{-3}) when the stochastic loss gradient is Lipschitz continuous. The question of achieving enhanced convergence rates under distinct conditions, remains open. In this work, we address this question for optimization problems that are nonconvex in the minimization variable and strongly concave or Polyak-Lojasiewicz (PL) in the maximization variable. We introduce novel bias-corrected momentum algorithms utilizing efficient Hessian-vector products. We establish convergence conditions and demonstrate a lower iteration complexity of O(ε−3)\mathcal{O}(\varepsilon^{-3}) for the proposed algorithms. The effectiveness of the proposed method is validated through applications to robust logistic regression and robust adaptive cruise control.
May 22, 2026

RA-DCA: A Randomized Active-Set DCA for Directional Stationarity in Max-Structured DC Programs

We study nonsmooth difference-of-convex programs whose subtracted convex term is a finite maximum of smooth convex functions. In this setting, standard DCA iterations may converge to critical points that are not directionally stationary, whereas exact active-vertex screening can be expensive when active sets are large or combinatorial. We propose RA-DCA, a vertex-first randomized active-set DCA that projects active gradients onto sampled directions, checks a sampled vertex residual, and uses a small linear program only as a low-residual convex-combination fallback. The method preserves the descent structure of DCA and reduces the randomized screening layer to matrix multiplications. Under the stated regularity, numerical active-set consistency, and random-embedding assumptions, every accumulation point generated by the safeguarded method is directionally stationary with probability one. MATLAB experiments first test the theorem on degenerate max-affine, max-quadratic, and sparse support-function models, where the safeguard avoids nonstationary critical points and closely tracks a full active-vertex scan. Block top-k tests then show that the same screening idea remains useful when exact aggregate enumeration is combinatorial. Trimmed-regression, complementarity, and QUBO diagnostics separate cases where active-set selection helps from cases dominated by multistart search, the DC split, or other problem-specific features.