Multi-Level Composite Stochastic Optimization via Nested Variance Reduction
Not on the map. Why?
Abstract (arXiv)
We consider multi-level composite optimization problems where each mapping in the composition is the expectation over a family of random smooth mappings or the sum of some finite number of smooth mappings. We present a normalized proximal approximate gradient (NPAG) method where the approximate gradients are obtained via nested stochastic variance reduction. In order to find an approximate stationary point where the expected norm of its gradient mapping is less than ε, the total sample complexity of our method is O(ε-3) in the expectation case, and O(N+√Nε-2) in the finite-sum case where N is the total number of functions across all composition levels. In addition, the dependence of our total sample complexity on the number of composition levels is polynomial, rather than exponential as in previous work.
Key facts. Cited by 28 papers in the map (data through 14 Oct 2026). First posted on arXiv 29 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 9; 2023: 6; 2024: 4; 2025: 1; 2026: 3.
28 papers.
Sort:
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
- 2021Only Train Once: A One-Shot Neural Network Training And Pruning Frameworkcited by 70
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applicationscited by 27
- 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
- 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
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimizationcited by 8
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 8
- 2021Communication-Efficient Sampling for Distributed Training of Graph Convolutional Networkscited by 7
16 more
- 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
- 2021On the Importance of Sampling in Training GCNs: Tighter Analysis and Variance Reductioncited by 4
- 2022Stochastic Compositional Optimization with Compositional Constraintscited by 4
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Datacited by 4
- 2022Finite-Sum Optimization: A New Perspective for Convergence to a Global Solutioncited by 2
- 2022Faster Adaptive Momentum-Based Federated Methods for Distributed Composition Optimizationcited by 2
- 2023Decentralized Multi-Level Compositional Optimization Algorithms with Level-Independent Convergence Ratecited by 2
- 2023Faster Stochastic Variance Reduction Methods for Compositional MiniMax Optimizationcited by 2
- 2024Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimizationcited by 2
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learningcited by 1
- 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
- 2026Federated Compositional Muon Optimizer for Matrix-Wise Modelscited by 0
- 2026Optimal Momentum Methods for Stochastic Multilevel Compositional Optimizationcited by 0
- 2026Optimal Momentum Methods for Stochastic Multilevel Compositional Optimization1 Oct 2026
- 2026Federated Compositional Muon Optimizer for Matrix-Wise Models13 Aug 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
- 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
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data29 May 2024
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization5 Oct 2023
- 2023Faster Stochastic Variance Reduction Methods for Compositional MiniMax Optimization18 Aug 2023
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learning11 Jul 2023
- 2023Learning Unnormalized Statistical Models via Compositional Optimization13 Jun 2023
16 more
- 2023Decentralized Multi-Level Compositional Optimization Algorithms with Level-Independent Convergence Rate6 Jun 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
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimization18 Jul 2022
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequences21 Jun 2022
- 2022Algorithmic Foundations of Empirical X-risk Minimization1 Jun 2022
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applications24 Feb 2022
- 2022Optimal Algorithms for Stochastic Multi-Level Compositional Optimization15 Feb 2022
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimization9 Feb 2022
- 2022Finite-Sum Optimization: A New Perspective for Convergence to a Global Solution7 Feb 2022
- 2021Only Train Once: A One-Shot Neural Network Training And Pruning Framework15 Jul 2021
- 2021Compositional federated learning: Applications in distributionally robust averaging and meta learning21 Jun 2021
- 2021On the Importance of Sampling in Training GCNs: Tighter Analysis and Variance Reduction3 Mar 2021
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Method17 Feb 2021
- 2021Communication-Efficient Sampling for Distributed Training of Graph Convolutional Networks19 Jan 2021
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
- 2021Only Train Once: A One-Shot Neural Network Training And Pruning Frameworkcited by 70
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applicationscited by 27
- 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
- 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
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimizationcited by 8
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 8
- 2021Communication-Efficient Sampling for Distributed Training of Graph Convolutional Networkscited by 7
12 most cited of 28 All 28 (JSON)
- 2026Optimal Momentum Methods for Stochastic Multilevel Compositional Optimization1 Oct 2026
- 2026Federated Compositional Muon Optimizer for Matrix-Wise Models13 Aug 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
- 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
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data29 May 2024
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization5 Oct 2023
- 2023Faster Stochastic Variance Reduction Methods for Compositional MiniMax Optimization18 Aug 2023
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learning11 Jul 2023
- 2023Learning Unnormalized Statistical Models via Compositional Optimization13 Jun 2023
12 newest of 28 All 28 (JSON)