Natasha 2: Faster Non-Convex Optimization Than SGD
Not on the map. Why?
Abstract (arXiv)
We design a stochastic algorithm to train any smooth neural network to varepsilon-approximate local minima, using O(varepsilon-3.25) backpropagations. The best result was essentially O(varepsilon-4) by SGD. More broadly, it finds varepsilon-approximate local minima of any smooth nonconvex function in rate O(varepsilon-3.25), with only oracle access to stochastic gradients.
Key facts. Cited by 49 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Aug 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 12; 2023: 8; 2024: 6; 2025: 5; 2026: 1.
49 papers.
Sort:
- 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Strongercited by 87
- 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tailscited by 80
- 2021Learning with User-Level Privacycited by 72
- 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
- 2021Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysiscited by 38
- 2021Bias-Variance Reduced Local SGD for Less Heterogeneous Federated Learningcited by 30
- 2021Proxy Convexity: A Unified Framework for the Analysis of Neural Networks Trained by Gradient Descentcited by 21
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
- 2024Revisiting Gradient Normalization and Clipping for Nonconvex SGD under Heavy-Tailed Noise: Necessity, Sufficiency, and Accelerationcited by 18
- 2023Near-Optimal Nonconvex-Strongly-Convex Bilevel Optimization with Fully First-Order Oraclescited by 16
- 2022Interference Management for Over-the-Air Federated Learning in Multi-Cell Wireless Networkscited by 15
- 2023How to escape sharp minima with random perturbationscited by 13
37 more
- 2021Escape saddle points by a simple gradient-descent based algorithmcited by 11
- 2021Wirelessly Powered Federated Edge Learning: Optimal Tradeoffs Between Convergence and Power Transfercited by 10
- 2021Differentially Private Bayesian Neural Networks on Accuracy, Privacy and Reliabilitycited by 6
- 2021A Stochastic Newton Algorithm for Distributed Convex Optimizationcited by 6
- 2022Zeroth-Order Negative Curvature Finding: Escaping Saddle Points without Gradientscited by 6
- 2022Optimization for Amortized Inverse Problemscited by 6
- 2022Simple and Optimal Stochastic Gradient Methods for Nonsmooth Nonconvex Optimizationcited by 5
- 2025Integrated Sensing, Communication, and Computation for Over-the-Air Federated Edge Learningcited by 5
- 2022Accelerated Riemannian Optimization: Handling Constraints with a Prox to Bound Geometric Penaltiescited by 4
- 2023Rethinking SIGN Training: Provable Nonconvex Acceleration without First- and Second-Order Gradient Lipschitzcited by 4
- 2025SAPPHIRE: Preconditioned Stochastic Variance Reduction for Faster Large-Scale Statistical Learningcited by 4
- 2021Second-order step-size tuning of SGD for non-convex optimizationcited by 3
- 2023Stochastic Variable Metric Proximal Gradient with variance reduction for non-convex composite optimizationcited by 3
- 2023Convergence of stochastic gradient descent under a local Lojasiewicz condition for deep neural networkscited by 3
- 2025Improving Online-to-Nonconvex Conversion for Smooth Optimization via Double Optimismcited by 3
- 2021A Variance Controlled Stochastic Method with Biased Estimation for Faster Non-convex Optimizationcited by 2
- 2022Coordinate Descent Methods for Fractional Minimizationcited by 2
- 2022Tackling benign nonconvexity with smoothing and stochastic gradientscited by 2
- 2022PINCH: An Adversarial Extraction Attack Framework for Deep Learning Modelscited by 2
- 2025Stopping Rules for Stochastic Gradient Descent via Anytime-Valid Confidence Sequencescited by 2
- 2021A Comprehensive Study on Optimization Strategies for Gradient Descent In Deep Learningcited by 1
- 2021An Efficient Algorithm for Deep Stochastic Contextual Banditscited by 1
- 2021Escaping Saddle Points with Compressed SGDcited by 1
- 2021Faster Perturbed Stochastic Gradient Methods for Finding Local Minimacited by 1
- 2023A randomized algorithm for nonconvex minimization with inexact evaluations and complexity guaranteescited by 1
- 2024A Stochastic Quasi-Newton Method for Non-convex Optimization with Non-uniform Smoothnesscited by 1
- 2024Lower Bounds and Optimal Algorithms for Non-Smooth Convex Decentralized Optimization over Time-Varying Networkscited by 1
- 2024Noise is All You Need: Private Second-Order Convergence of Noisy SGDcited by 1
- 2026AdaCubic: An Adaptive Cubic Regularization Optimizer for Deep Learningcited by 1
- 2021Boundary Conditions for Linear Exit Time Gradient Trajectories Around Saddle Points: Analysis and Algorithmcited by 0
- 2021Escaping Saddle Points with Stochastically Controlled Stochastic Gradient Methodscited by 0
- 2022Provable Constrained Stochastic Convex Optimization with XOR-Projected Gradient Descentcited by 0
- 2022Can We Do Better Than Random Start? The Power of Data Outsourcingcited by 0
- 2022HOUDINI: Escaping from Moderately Constrained Saddlescited by 0
- 2023Private Federated Learning with Dynamic Power Control via Non-Coherent Over-the-Air Computationcited by 0
- 2024Efficient Continual Finite-Sum Minimizationcited by 0
- 2025OptiVote: Non-Coherent FSO Over-the-Air Majority Vote for Communication-Efficient Distributed Federated Learning in Space Data Centerscited by 0
- 2026AdaCubic: An Adaptive Cubic Regularization Optimizer for Deep Learning10 Apr 2026
- 2025OptiVote: Non-Coherent FSO Over-the-Air Majority Vote for Communication-Efficient Distributed Federated Learning in Space Data Centers30 Dec 2025
- 2025Stopping Rules for Stochastic Gradient Descent via Anytime-Valid Confidence Sequences15 Dec 2025
- 2025Improving Online-to-Nonconvex Conversion for Smooth Optimization via Double Optimism3 Oct 2025
- 2025Integrated Sensing, Communication, and Computation for Over-the-Air Federated Edge Learning21 Aug 2025
- 2025SAPPHIRE: Preconditioned Stochastic Variance Reduction for Faster Large-Scale Statistical Learning27 Jan 2025
- 2024Revisiting Gradient Normalization and Clipping for Nonconvex SGD under Heavy-Tailed Noise: Necessity, Sufficiency, and Acceleration21 Oct 2024
- 2024Noise is All You Need: Private Second-Order Convergence of Noisy SGD9 Oct 2024
- 2024Efficient Continual Finite-Sum Minimization7 Jun 2024
- 2024Lower Bounds and Optimal Algorithms for Non-Smooth Convex Decentralized Optimization over Time-Varying Networks28 May 2024
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimization16 May 2024
- 2024A Stochastic Quasi-Newton Method for Non-convex Optimization with Non-uniform Smoothness22 Mar 2024
37 more
- 2023A randomized algorithm for nonconvex minimization with inexact evaluations and complexity guarantees28 Oct 2023
- 2023Rethinking SIGN Training: Provable Nonconvex Acceleration without First- and Second-Order Gradient Lipschitz23 Oct 2023
- 2023Private Federated Learning with Dynamic Power Control via Non-Coherent Over-the-Air Computation5 Aug 2023
- 2023Near-Optimal Nonconvex-Strongly-Convex Bilevel Optimization with Fully First-Order Oracles26 Jun 2023
- 2023How to escape sharp minima with random perturbations25 May 2023
- 2023Convergence of stochastic gradient descent under a local Lojasiewicz condition for deep neural networks18 Apr 2023
- 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversion7 Feb 2023
- 2023Stochastic Variable Metric Proximal Gradient with variance reduction for non-convex composite optimization2 Jan 2023
- 2022Accelerated Riemannian Optimization: Handling Constraints with a Prox to Bound Geometric Penalties26 Nov 2022
- 2022Optimization for Amortized Inverse Problems25 Oct 2022
- 2022Zeroth-Order Negative Curvature Finding: Escaping Saddle Points without Gradients4 Oct 2022
- 2022PINCH: An Adversarial Extraction Attack Framework for Deep Learning Models13 Sep 2022
- 2022Simple and Optimal Stochastic Gradient Methods for Nonsmooth Nonconvex Optimization22 Aug 2022
- 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Stronger14 Jun 2022
- 2022Interference Management for Over-the-Air Federated Learning in Multi-Cell Wireless Networks6 Jun 2022
- 2022HOUDINI: Escaping from Moderately Constrained Saddles27 May 2022
- 2022Can We Do Better Than Random Start? The Power of Data Outsourcing17 May 2022
- 2022Provable Constrained Stochastic Convex Optimization with XOR-Projected Gradient Descent22 Mar 2022
- 2022Tackling benign nonconvexity with smoothing and stochastic gradients18 Feb 2022
- 2022Coordinate Descent Methods for Fractional Minimization30 Jan 2022
- 2021Escape saddle points by a simple gradient-descent based algorithm28 Nov 2021
- 2021Faster Perturbed Stochastic Gradient Methods for Finding Local Minima25 Oct 2021
- 2021A Stochastic Newton Algorithm for Distributed Convex Optimization7 Oct 2021
- 2021Differentially Private Bayesian Neural Networks on Accuracy, Privacy and Reliability18 Jul 2021
- 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tails28 Jun 2021
- 2021Proxy Convexity: A Unified Framework for the Analysis of Neural Networks Trained by Gradient Descent25 Jun 2021
- 2021Escaping Saddle Points with Compressed SGD21 May 2021
- 2021Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysis4 May 2021
- 2021An Efficient Algorithm for Deep Stochastic Contextual Bandits12 Apr 2021
- 2021Escaping Saddle Points with Stochastically Controlled Stochastic Gradient Methods7 Mar 2021
- 2021Second-order step-size tuning of SGD for non-convex optimization5 Mar 2021
- 2021Wirelessly Powered Federated Edge Learning: Optimal Tradeoffs Between Convergence and Power Transfer24 Feb 2021
- 2021Learning with User-Level Privacy23 Feb 2021
- 2021A Variance Controlled Stochastic Method with Biased Estimation for Faster Non-convex Optimization19 Feb 2021
- 2021Bias-Variance Reduced Local SGD for Less Heterogeneous Federated Learning5 Feb 2021
- 2021Boundary Conditions for Linear Exit Time Gradient Trajectories Around Saddle Points: Analysis and Algorithm7 Jan 2021
- 2021A Comprehensive Study on Optimization Strategies for Gradient Descent In Deep Learning7 Jan 2021
- 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Strongercited by 87
- 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tailscited by 80
- 2021Learning with User-Level Privacycited by 72
- 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
- 2021Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysiscited by 38
- 2021Bias-Variance Reduced Local SGD for Less Heterogeneous Federated Learningcited by 30
- 2021Proxy Convexity: A Unified Framework for the Analysis of Neural Networks Trained by Gradient Descentcited by 21
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
- 2024Revisiting Gradient Normalization and Clipping for Nonconvex SGD under Heavy-Tailed Noise: Necessity, Sufficiency, and Accelerationcited by 18
- 2023Near-Optimal Nonconvex-Strongly-Convex Bilevel Optimization with Fully First-Order Oraclescited by 16
- 2022Interference Management for Over-the-Air Federated Learning in Multi-Cell Wireless Networkscited by 15
- 2023How to escape sharp minima with random perturbationscited by 13
12 most cited of 49 All 49 (JSON)
- 2026AdaCubic: An Adaptive Cubic Regularization Optimizer for Deep Learning10 Apr 2026
- 2025OptiVote: Non-Coherent FSO Over-the-Air Majority Vote for Communication-Efficient Distributed Federated Learning in Space Data Centers30 Dec 2025
- 2025Stopping Rules for Stochastic Gradient Descent via Anytime-Valid Confidence Sequences15 Dec 2025
- 2025Improving Online-to-Nonconvex Conversion for Smooth Optimization via Double Optimism3 Oct 2025
- 2025Integrated Sensing, Communication, and Computation for Over-the-Air Federated Edge Learning21 Aug 2025
- 2025SAPPHIRE: Preconditioned Stochastic Variance Reduction for Faster Large-Scale Statistical Learning27 Jan 2025
- 2024Revisiting Gradient Normalization and Clipping for Nonconvex SGD under Heavy-Tailed Noise: Necessity, Sufficiency, and Acceleration21 Oct 2024
- 2024Noise is All You Need: Private Second-Order Convergence of Noisy SGD9 Oct 2024
- 2024Efficient Continual Finite-Sum Minimization7 Jun 2024
- 2024Lower Bounds and Optimal Algorithms for Non-Smooth Convex Decentralized Optimization over Time-Varying Networks28 May 2024
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimization16 May 2024
- 2024A Stochastic Quasi-Newton Method for Non-convex Optimization with Non-uniform Smoothness22 Mar 2024
12 newest of 49 All 49 (JSON)