Stochastic Second-Order Methods Improve Best-Known Sample Complexity of SGD for Gradient-Dominated Function
Not on the map. Why?
Abstract (arXiv)
We study the performance of Stochastic Cubic Regularized Newton (SCRN) on a class of functions satisfying gradient dominance property with 1≤α≤2 which holds in a wide range of applications in machine learning and signal processing. This condition ensures that any first-order stationary point is a global optimum. We prove that the total sample complexity of SCRN in achieving ε-global optimum is O(ε-7/(2α)+1) for 1≤α< 3/2 and mathcaltildeO(ε-2/(α)) for 3/2≤α≤ 2. SCRN improves the best-known sample complexity of stochastic gradient descent. Even under a weak version of gradient dominance property, which is applicable to policy-based reinforcement learning (RL), SCRN achieves the same improvement over stochastic policy gradient methods. Additionally, we show that the average sample complexity of SCRN can be reduced to O(ε-2) for α=1 using a variance reduction method with time-varying batch sizes. Experimental results in various RL settings showcase the remarkable performance of SCRN compared to first-order methods.
Key facts. Cited by 20 papers in the map (data through 14 Oct 2026). Builds on 28 of its 65 references that are in the map. First posted on arXiv 25 May 2022.
Builds on
28 of 65 references.
References · 28
- 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 539
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 306
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 264
- 2019Neural Policy Gradient Methods: Global Optimality and Rates of Convergencecited by 159
- 2020Better Theory for SGD in the Nonconvex Worldcited by 158
- 2017Convergence Analysis of Two-layer Neural Networks with ReLU Activationcited by 137
- 2019Global Convergence of Policy Gradient Methods to (Almost) Locally Optimal Policiescited by 124
- 2020Variational Policy Gradient Method for Reinforcement Learning with General Utilitiescited by 111
- 2016Identity Matters in Deep Learningcited by 109
- 2021A general sample complexity analysis of vanilla policy gradientcited by 76
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
- 2017Sub-sampled Cubic Regularization for Non-convex Optimizationcited by 63
16 more
- 2019An Improved Convergence Analysis of Stochastic Variance-Reduced Policy Gradientcited by 59
- 2017Stochastic Cubic Regularization for Fast Nonconvex Optimizationcited by 55
- 2020Second-Order Information in Non-Convex Stochastic Optimization: Power and Limitationscited by 40
- 2018Uniform Convergence of Gradients for Non-Convex Learning and Optimizationcited by 38
- 2017Second-Order Optimization for Non-Convex Machine Learning: An Empirical Studycited by 35
- 2020Momentum-Based Policy Gradient Methodscited by 30
- 2013An Asynchronous Parallel Stochastic Coordinate Descent Algorithmcited by 26
- 2018Tight Dimension Independent Lower Bound on the Expected Convergence Rate for Diminishing Step Sizes in SGDcited by 18
- 2020Sample Complexity of Policy Gradient Finding Second-Order Stationary Pointscited by 17
- 2018Stochastic Variance-Reduced Cubic Regularization for Nonconvex Optimizationcited by 12
- 2019Stochastic Recursive Variance-Reduced Cubic Regularization Methodscited by 12
- 2018Stochastic Variance-Reduced Cubic Regularized Newton Methodcited by 10
- 2016Finding Approximate Local Minima Faster than Gradient Descentcited by 8
- 2018Convergence of Cubic Regularization for Nonconvex Optimization under KL Propertycited by 8
- 2018Adaptive Stochastic Variance Reduction for Subsampled Newton Method with Cubic Regularizationcited by 3
- 2019Inexact Proximal Cubic Regularized Newton Methods for Convex Optimizationcited by 1
Built on by
Citing papers per year: 2023: 7; 2024: 8; 2025: 2; 2026: 3.
20 papers.
Sort:
By area:
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergencecited by 30
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processescited by 25
- 2023Reinforcement Learning with General Utilities: Simpler Variance Reduction and Large State-Action Spacecited by 18
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methodscited by 12
- 2024Learning Optimal Deterministic Policies with Stochastic Policy Gradientscited by 12
- 2024Sample-Efficient Constrained Reinforcement Learning with General Parameterizationcited by 11
- 2023A safe exploration approach to constrained Markov decision processescited by 10
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learningcited by 10
- 2024Global Convergence of Natural Policy Gradient with Hessian-aided Momentum Variance Reductioncited by 6
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversariescited by 6
- 2026Regret Analysis of Unichain Average Reward Constrained MDPs with General Parameterizationcited by 4
- 2024On The Global Convergence Of Online RLHF With Neural Parametrizationcited by 3
8 more
- 2024Last-Iterate Global Convergence of Policy Gradients for Constrained Reinforcement Learningcited by 2
- 2024Deep Learning Frameworks for Cognitive Radio Networks: Review and Open Research Challengescited by 2
- 2024Last-Iterate Convergence of General Parameterized Policies in Constrained MDPscited by 1
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrencecited by 1
- 2023Stochastic Gradient Langevin Dynamics Based on Quantization with Increasing Resolutioncited by 0
- 2023A Homogenization Approach for Gradient-Dominated Stochastic Optimizationcited by 0
- 2025A Variance-Reduced Cubic-Regularized Newton for Policy Optimizationcited by 0
- 2026Variance-Reduced Policy Gradient is Nearly Minimax-Optimalcited by 0
- 2026Variance-Reduced Policy Gradient is Nearly Minimax-Optimal9 Oct 2026
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrence22 Aug 2026
- 2026Regret Analysis of Unichain Average Reward Constrained MDPs with General Parameterization8 Feb 2026
- 2025A Variance-Reduced Cubic-Regularized Newton for Policy Optimization14 Jul 2025
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learning22 Mar 2025
- 2024Deep Learning Frameworks for Cognitive Radio Networks: Review and Open Research Challenges31 Oct 2024
- 2024On The Global Convergence Of Online RLHF With Neural Parametrization21 Oct 2024
- 2024Last-Iterate Convergence of General Parameterized Policies in Constrained MDPs21 Aug 2024
- 2024Last-Iterate Global Convergence of Policy Gradients for Constrained Reinforcement Learning15 Jul 2024
- 2024Sample-Efficient Constrained Reinforcement Learning with General Parameterization17 May 2024
- 2024Learning Optimal Deterministic Policies with Stochastic Policy Gradients3 May 2024
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversaries15 Mar 2024
8 more
- 2024Global Convergence of Natural Policy Gradient with Hessian-aided Momentum Variance Reduction2 Jan 2024
- 2023A safe exploration approach to constrained Markov decision processes1 Dec 2023
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processes18 Oct 2023
- 2023A Homogenization Approach for Gradient-Dominated Stochastic Optimization21 Aug 2023
- 2023Reinforcement Learning with General Utilities: Simpler Variance Reduction and Large State-Action Space2 Jun 2023
- 2023Stochastic Gradient Langevin Dynamics Based on Quantization with Increasing Resolution30 May 2023
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methods23 Feb 2023
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergence30 Jan 2023
- 2024Learning Optimal Deterministic Policies with Stochastic Policy Gradientscited by 12
- 2024Sample-Efficient Constrained Reinforcement Learning with General Parameterizationcited by 11
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learningcited by 10
- 2024Global Convergence of Natural Policy Gradient with Hessian-aided Momentum Variance Reductioncited by 6
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversariescited by 6
- 2026Regret Analysis of Unichain Average Reward Constrained MDPs with General Parameterizationcited by 4
- 2024Last-Iterate Global Convergence of Policy Gradients for Constrained Reinforcement Learningcited by 2
- 2024Last-Iterate Convergence of General Parameterized Policies in Constrained MDPscited by 1
- 2025A Variance-Reduced Cubic-Regularized Newton for Policy Optimizationcited by 0
- 2026Variance-Reduced Policy Gradient is Nearly Minimax-Optimalcited by 0
- 2026Variance-Reduced Policy Gradient is Nearly Minimax-Optimal9 Oct 2026
- 2026Regret Analysis of Unichain Average Reward Constrained MDPs with General Parameterization8 Feb 2026
- 2025A Variance-Reduced Cubic-Regularized Newton for Policy Optimization14 Jul 2025
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learning22 Mar 2025
- 2024Last-Iterate Convergence of General Parameterized Policies in Constrained MDPs21 Aug 2024
- 2024Last-Iterate Global Convergence of Policy Gradients for Constrained Reinforcement Learning15 Jul 2024
- 2024Sample-Efficient Constrained Reinforcement Learning with General Parameterization17 May 2024
- 2024Learning Optimal Deterministic Policies with Stochastic Policy Gradients3 May 2024
- 2024Global Convergence Guarantees for Federated Policy Gradient Methods with Adversaries15 Mar 2024
- 2024Global Convergence of Natural Policy Gradient with Hessian-aided Momentum Variance Reduction2 Jan 2024
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergencecited by 30
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processescited by 25
- 2023Reinforcement Learning with General Utilities: Simpler Variance Reduction and Large State-Action Spacecited by 18
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methodscited by 12
- 2023A safe exploration approach to constrained Markov decision processescited by 10
- 2024Deep Learning Frameworks for Cognitive Radio Networks: Review and Open Research Challengescited by 2
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrencecited by 1
- 2023Stochastic Gradient Langevin Dynamics Based on Quantization with Increasing Resolutioncited by 0
- 2023A Homogenization Approach for Gradient-Dominated Stochastic Optimizationcited by 0
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrence22 Aug 2026
- 2024Deep Learning Frameworks for Cognitive Radio Networks: Review and Open Research Challenges31 Oct 2024
- 2023A safe exploration approach to constrained Markov decision processes1 Dec 2023
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processes18 Oct 2023
- 2023A Homogenization Approach for Gradient-Dominated Stochastic Optimization21 Aug 2023
- 2023Reinforcement Learning with General Utilities: Simpler Variance Reduction and Large State-Action Space2 Jun 2023
- 2023Stochastic Gradient Langevin Dynamics Based on Quantization with Increasing Resolution30 May 2023
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methods23 Feb 2023
- 2023A Novel Framework for Policy Mirror Descent with General Parameterization and Linear Convergence30 Jan 2023