Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methods
Organizations: Machine Learning and Optimization Laboratory (MLO), EPFL
Abstract
We study stochastic Cubic Newton methods for solving general possibly non-convex minimization problems. We propose a new framework, which we call the helper framework, that provides a unified view of the stochastic and variance-reduced second-order algorithms equipped with global complexity guarantees. It can also be applied to learning with auxiliary information. Our helper framework offers the algorithm designer high flexibility for constructing and analyzing the stochastic Cubic Newton methods, allowing arbitrary size batches, and the use of noisy and possibly biased estimates of the gradients and Hessians, incorporating both the variance reduction and the lazy Hessian updates. We recover the best-known complexities for the stochastic and variance-reduced Cubic Newton, under weak assumptions on the noise. A direct consequence of our theory is the new lazy stochastic second-order method, which significantly improves the arithmetic complexity for large dimension problems. We also establish complexity bounds for the classes of gradient-dominated objectives, that include convex and strongly convex problems. For Auxiliary Learning, we show that using a helper (auxiliary function) can outperform training alone if a given similarity measure is small.
Figures & tables
| Non-convex , -local solution | Convex ( ), -global solution | Ref. | |
| Gradient Descent (GD) | Nesterov (2018) | ||
| Stochastic Gradient Descent (SGD) | Ghadimi and Lan (2013) | ||
| Stochastic Variance Reduced Grad. (SVRG) | Johnson and Zhang (2013) | ||
| Cubic Newton (CN) | Nesterov and Polyak (2006) | ||
| Stochastic Cubic Newton (SCN) | Kohler and Lucchi (2017) Nilesh et al. (2018) | ||
| Variance Reduced Stoch. Cubic Newton (VRCN) | (new) | Zhou et al. (2019) Wang et al. (2019) |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Second-Order Optimization
Second-order optimization uses curvature information, such as Hessians or their approximations, to choose efficient steps toward better solutions.
See all 90 papers