math.OCMay 14, 2026

Stochastic Compositional Optimization via Hybrid Momentum Frank--Wolfe

Authors: El Mahdi Chayti

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

Abstract

Stochastic compositional optimization minimizes objectives of the form min⁡x∈XF(f(x),x)\min_{\bm{x} \in \mathcal{X}} F(\bm{f}(\bm{x}), \bm{x}), where f\bm{f} is accessible only through noisy stochastic queries. Existing methods for this problem assume that the outer function FF is continuously differentiable, which excludes many practically important applications such as robust max-of-losses, Conditional Value-at-Risk, and norm regularizers. We propose the Hybrid Momentum Stochastic Frank--Wolfe algorithm, which drops the smoothness assumption on FF. By combining a momentum-based Jacobian tracker with a Taylor-corrected function tracker, the algorithm feeds an entire stochastic linearization -- rather than a single gradient -- into a generalized linear minimization oracle. We establish an O(K−1/4)\mathcal{O}(K^{-1/4}) convergence rate in the generalized Frank--Wolfe gap for non-convex objectives with LFL_F-Lipschitz outer functions, matching the optimal complexity for projection-free single-sample stochastic methods under expected smoothness. The analysis extends to heavy-tailed noise oracles with bounded rr-th moments for r∈(1,2]r \in (1, 2] and recovers the deterministic rates of Vladarean et al (2023) as the noise vanishes.

Explore similar work

Sep 14, 2026math.OC

Projection-Free Multi-level Algorithms for Stochastic Constrained Compositional Optimization

This paper studies projection-free algorithms for stochastic constrained multi-level compositional optimization. In this context, the objective function is a nested composition of several smooth functions, and the decision set is closed and convex. Since projection onto the constraint set can be computationally expensive, we develop projection-free methods that rely on linear minimization oracles. For non-convex objectives, we propose variance-reduced projection-free algorithms and establish complexity guarantees under both the Frank-Wolfe gap and the gradient mapping criteria. We also develop momentum-based methods that achieve convergence guarantees under weaker smoothness assumptions. Additionally, by using a stage-wise design, we derive a parameter-free variant that preserves the same complexities for the Frank-Wolfe gap. Such a design can be further used to develop algorithms for convex and strongly convex functions whose rates match those of single-level projection-free counterparts. Finally, we consider finite-sum problems and derive complexities for non-convex, convex, and strongly convex objectives. Numerical experiments across multiple tasks demonstrate the effectiveness of the proposed methods.
Wei Jiang, Sifan Yang, Wenhao Yang +4
Jul 28, 2026math.OC

Variance-Reduced Conditional Gradient Methods under Markovian Sampling for Nonconvex Composite Optimization

We study stochastic composite nonconvex optimization over a compact convex set when gradient samples arrive along a single trajectory of a fixed ergodic Markov chain. Existing single-trajectory variance-reduction theory covers smooth unconstrained objectives; we address the projection-free composite setting using the generalized Frank-Wolfe gap. We propose MC-ALFCG, which combines a momentum conditional-gradient method with coupled capped multilevel Monte Carlo estimation and per-iteration clipping. The deepest nested average uses consecutive states from the same trajectory, yielding conditional bias O(τmix/T)O(τ_{\mathrm{mix}}/T) uniformly over the starting state, while coupling controls the gradient-difference second moment through the iterate displacement. Clipping enforces the pathwise bounds needed by the adaptive analysis. We reduce the Markovian recursion to its independent-sampling counterpart under σ2↦2ΛGσ2σ^2\mapsto 2ΛG_σ^2 and L2↦2ΛL2L^2\mapsto 2ΛL^2, where Λ=O(τmixlog⁡T)Λ=O(τ_{\mathrm{mix}}\log T). For positive centered noise, the tuned method achieves expected sample complexity O~((τmix2Gσ+τmix5/2Gσ2)ε−3+τmix5ε−2)\widetilde{O}((τ_{\mathrm{mix}}^2G_σ+τ_{\mathrm{mix}}^{5/2}G_σ^2)\varepsilon^{-3}+τ_{\mathrm{mix}}^5\varepsilon^{-2}). The exactly noiseless specialization achieves O~(ε−2)\widetilde{O}(\varepsilon^{-2}) with mixing-time-free constants, while a mixing-time-oblivious variant achieves O~(τmix6ε−3+τmix3ε−2)\widetilde{O}(τ_{\mathrm{mix}}^6\varepsilon^{-3}+τ_{\mathrm{mix}}^3\varepsilon^{-2}). All guarantees are in expectation under a fixed transition kernel. Controlled numerical studies examine dependence sensitivity, a nonconvex composite instance, and clipping behavior.
Zhaojun Peng
Oct 13, 2025math.OC

Accelerated stochastic first-order method for convex optimization under heavy-tailed noise

We study convex composite optimization problems, where the objective function is given by the sum of a prox-friendly function and a convex function whose subgradients are estimated under heavy-tailed noise. Existing work often employs gradient clipping or normalization techniques in stochastic first-order methods to address heavy-tailed noise. %In this paper, we demonstrate that a vanilla stochastic algorithm---without additional modifications such as clipping or normalization---can achieve optimal complexity for these problems. In this paper, we analyze the first-order oracle complexity of vanilla stochastic algorithms---without additional modifications such as clipping or normalization---for solving these problems. In particular, we establish that an accelerated stochastic proximal subgradient method achieves a first-order oracle complexity for finding an approximate optimal solution in expectation that is universally optimal for smooth, weakly smooth, and nonsmooth convex optimization, as well as for stochastic convex optimization under heavy-tailed noise. Moreover, we derive high-probability first-order oracle complexity bounds for the accelerated stochastic proximal subgradient method under heavy-tailed and sub-Weibull noise, respectively. Numerical experiments are further provided to illustrate the numerical behavior of the methods.
Chuan He, Bowen Li, Zhaosong Lu