A practical randomized trust-region method to escape saddle points in high dimension
This addresses a fundamental bottleneck in non-convex optimization for machine learning and scientific computing, offering a practical improvement over existing methods like perturbed gradient descent.
The paper tackles the problem of escaping saddle points in high-dimensional optimization by introducing a randomized trust-region method, which reduces the information needed from polynomial to polylogarithmic dependence on dimension and achieves convergence without precise tuning.
Without randomization, escaping the saddle points of $f \colon \mathbb{R}^d \to \mathbb{R}$ requires at least $Ω(d)$ pieces of information about $f$ (values, gradients, Hessian-vector products). With randomization, this can be reduced to a polylogarithmic dependence in $d$. The prototypical algorithm to that effect is perturbed gradient descent (PGD): through sustained jitter, it reliably escapes strict saddle points. However, it also never settles: there is no convergence. What is more, PGD requires precise tuning based on Lipschitz constants and a preset target accuracy. To improve on this, we modify the time-tested trust-region method with truncated conjugate gradients (TR-tCG). Specifically, we randomize the initialization of tCG (the subproblem solver), and we prove that tCG automatically amplifies the randomization near saddles (to escape) and absorbs it near local minimizers (to converge). Saddle escape happens over several iterations. Accordingly, our analysis is multi-step, with several novelties. The proposed algorithm is practical: it essentially tracks the good behavior of TR-tCG, with three minute modifications and a single new hyperparameter (the noise scale $Ï$). We provide an implementation and numerical experiments.