Accelerating Stochastic Composition Optimization
Not on the map. Why?
Abstract (arXiv)
Consider the stochastic composition optimization problem where the objective is a composition of two expected-value functions. We propose a new stochastic first-order method, namely the accelerated stochastic compositional proximal gradient (ASC-PG) method, which updates based on queries to the sampling oracle using two different timescales. The ASC-PG is the first proximal gradient method for the stochastic composition problem that can deal with nonsmooth regularization penalty. We show that the ASC-PG exhibits faster convergence than the best known algorithms, and that it achieves the optimal sample-error complexity in several important special cases. We further demonstrate the application of ASC-PG to reinforcement learning and conduct numerical experiments.
Key facts. Cited by 46 papers in the map (data through 14 Oct 2026). First posted on arXiv 25 Jul 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 12; 2023: 14; 2024: 4; 2025: 2; 2026: 2.
46 papers.
Sort:
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
- 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2021A Novel Convergence Analysis for Algorithms of the Adam Familycited by 50
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 43
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 35
- 2022Decentralized Gossip-Based Stochastic Bilevel Optimization over Communication Networkscited by 32
- 2021Tighter Analysis of Alternating Stochastic Gradient Method for Stochastic Nested Problemscited by 22
- 2022Optimal Algorithms for Stochastic Multi-Level Compositional Optimizationcited by 17
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequencescited by 17
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimizationcited by 16
- 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
34 more
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximizationcited by 14
- 2024Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reductioncited by 13
- 2021Compositional federated learning: Applications in distributionally robust averaging and meta learningcited by 12
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimizationcited by 10
- 2021Efficient Semi-Implicit Variational Inferencecited by 9
- 2023Post-processing Private Synthetic Data for Improving Utility on Selected Measurescited by 9
- 2021Projection-Free Algorithm for Stochastic Bi-level Optimizationcited by 8
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimizationcited by 8
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 8
- 2021Rethinking Adam: A Twofold Exponential Moving Average Approachcited by 6
- 2023Debiasing Conditional Stochastic Optimizationcited by 6
- 2023Learning Unnormalized Statistical Models via Compositional Optimizationcited by 6
- 2024Adaptive Variance Reduction for Stochastic Optimization under Weaker Assumptionscited by 6
- 2022Learning Distributionally Robust Models at Scale via Composite Optimizationcited by 5
- 2022DR-DSGD: A Distributionally Robust Decentralized Learning Algorithm over Graphscited by 5
- 2022Stochastic Compositional Optimization with Compositional Constraintscited by 4
- 2023Federated Conditional Stochastic Optimizationcited by 4
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimizationcited by 4
- 2023Stability and Generalization of Stochastic Compositional Gradient Descent Algorithmscited by 3
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimizationcited by 3
- 2022Faster Adaptive Momentum-Based Federated Methods for Distributed Composition Optimizationcited by 2
- 2023A new Gradient TD Algorithm with only One Step-size: Convergence Rate Analysis using L-λ Smoothnesscited by 2
- 2023Doubly Robust Instance-Reweighted Adversarial Trainingcited by 2
- 2024Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimizationcited by 2
- 2022Riemannian Stochastic Gradient Method for Nested Composition Optimizationcited by 1
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learningcited by 1
- 2023FedDRO: Federated Compositional Optimization for Distributionally Robust Learningcited by 1
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximizationcited by 1
- 2021PASTO: Strategic Parameter Optimization in Recommendation Systems -- Probabilistic is Better than Deterministiccited by 0
- 2023Federated Multi-Sequence Stochastic Approximation with Local Hypergradient Estimationcited by 0
- 2023Federated Multi-Level Optimization over Decentralized Networkscited by 0
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimizationcited by 0
- 2026On the Convergence of Stochastic Gradient Descent with Perturbed Forward-Backward Passescited by 0
- 2026Equilibrium bias and convergence in augmented primal--dual dynamics with sampled constraintscited by 0
- 2026Equilibrium bias and convergence in augmented primal--dual dynamics with sampled constraints12 Sep 2026
- 2026On the Convergence of Stochastic Gradient Descent with Perturbed Forward-Backward Passes24 Feb 2026
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimization3 Jun 2025
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximization28 May 2025
- 2024Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimization6 Jun 2024
- 2024Adaptive Variance Reduction for Stochastic Optimization under Weaker Assumptions4 Jun 2024
- 2024Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reduction1 Jun 2024
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimization17 May 2024
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimization4 Dec 2023
- 2023FedDRO: Federated Compositional Optimization for Distributionally Robust Learning21 Nov 2023
- 2023Federated Multi-Level Optimization over Decentralized Networks10 Oct 2023
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization5 Oct 2023
34 more
- 2023Federated Conditional Stochastic Optimization4 Oct 2023
- 2023Doubly Robust Instance-Reweighted Adversarial Training1 Aug 2023
- 2023A new Gradient TD Algorithm with only One Step-size: Convergence Rate Analysis using L-λ Smoothness29 Jul 2023
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learning11 Jul 2023
- 2023Stability and Generalization of Stochastic Compositional Gradient Descent Algorithms7 Jul 2023
- 2023Learning Unnormalized Statistical Models via Compositional Optimization13 Jun 2023
- 2023LibAUC: A Deep Learning Library for X-Risk Optimization5 Jun 2023
- 2023Federated Multi-Sequence Stochastic Approximation with Local Hypergradient Estimation2 Jun 2023
- 2023Post-processing Private Synthetic Data for Improving Utility on Selected Measures24 May 2023
- 2023Debiasing Conditional Stochastic Optimization20 Apr 2023
- 2022Faster Adaptive Momentum-Based Federated Methods for Distributed Composition Optimization3 Nov 2022
- 2022Stochastic Compositional Optimization with Compositional Constraints9 Sep 2022
- 2022DR-DSGD: A Distributionally Robust Decentralized Learning Algorithm over Graphs29 Aug 2022
- 2022Riemannian Stochastic Gradient Method for Nested Composition Optimization19 Jul 2022
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimization18 Jul 2022
- 2022Decentralized Gossip-Based Stochastic Bilevel Optimization over Communication Networks22 Jun 2022
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequences21 Jun 2022
- 2022Algorithmic Foundations of Empirical X-risk Minimization1 Jun 2022
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimization4 May 2022
- 2022Learning Distributionally Robust Models at Scale via Composite Optimization17 Mar 2022
- 2022Optimal Algorithms for Stochastic Multi-Level Compositional Optimization15 Feb 2022
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimization9 Feb 2022
- 2021A Novel Convergence Analysis for Algorithms of the Adam Family7 Dec 2021
- 2021Projection-Free Algorithm for Stochastic Bi-level Optimization22 Oct 2021
- 2021On Tilted Losses in Machine Learning: Theory and Applications13 Sep 2021
- 2021PASTO: Strategic Parameter Optimization in Recommendation Systems -- Probabilistic is Better than Deterministic20 Aug 2021
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximization2 Jul 2021
- 2021Tighter Analysis of Alternating Stochastic Gradient Method for Stochastic Nested Problems25 Jun 2021
- 2021Rethinking Adam: A Twofold Exponential Moving Average Approach22 Jun 2021
- 2021Compositional federated learning: Applications in distributionally robust averaging and meta learning21 Jun 2021
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergence18 Apr 2021
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Method17 Feb 2021
- 2021A Single-Timescale Method for Stochastic Bilevel Optimization9 Feb 2021
- 2021Efficient Semi-Implicit Variational Inference15 Jan 2021
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
- 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2021A Novel Convergence Analysis for Algorithms of the Adam Familycited by 50
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 43
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 35
- 2022Decentralized Gossip-Based Stochastic Bilevel Optimization over Communication Networkscited by 32
- 2021Tighter Analysis of Alternating Stochastic Gradient Method for Stochastic Nested Problemscited by 22
- 2022Optimal Algorithms for Stochastic Multi-Level Compositional Optimizationcited by 17
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequencescited by 17
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimizationcited by 16
- 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
12 most cited of 46 All 46 (JSON)
- 2026Equilibrium bias and convergence in augmented primal--dual dynamics with sampled constraints12 Sep 2026
- 2026On the Convergence of Stochastic Gradient Descent with Perturbed Forward-Backward Passes24 Feb 2026
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimization3 Jun 2025
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximization28 May 2025
- 2024Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimization6 Jun 2024
- 2024Adaptive Variance Reduction for Stochastic Optimization under Weaker Assumptions4 Jun 2024
- 2024Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reduction1 Jun 2024
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimization17 May 2024
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimization4 Dec 2023
- 2023FedDRO: Federated Compositional Optimization for Distributionally Robust Learning21 Nov 2023
- 2023Federated Multi-Level Optimization over Decentralized Networks10 Oct 2023
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization5 Oct 2023
12 newest of 46 All 46 (JSON)