We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

math.OC

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Mathematics > Optimization and Control

Title: A Stochastic Derivative Free Optimization Method with Momentum

Abstract: We consider the problem of unconstrained minimization of a smooth objective function in $\mathbb{R}^d$ in setting where only function evaluations are possible. We propose and analyze stochastic zeroth-order method with heavy ball momentum. In particular, we propose, SMTP, a momentum version of the stochastic three-point method (STP) \cite{Bergou_2018}. We show new complexity results for non-convex, convex and strongly convex functions. We test our method on a collection of learning to continuous control tasks on several MuJoCo \cite{Todorov_2012} environments with varying difficulty and compare against STP, other state-of-the-art derivative-free optimization algorithms and against policy gradient methods. SMTP significantly outperforms STP and all other methods that we considered in our numerical experiments. Our second contribution is SMTP with importance sampling which we call SMTP_IS. We provide convergence analysis of this method for non-convex, convex and strongly convex objectives.
Subjects: Optimization and Control (math.OC)
Cite as: arXiv:1905.13278 [math.OC]
  (or arXiv:1905.13278v2 [math.OC] for this version)

Submission history

From: Eduard Gorbunov [view email]
[v1] Thu, 30 May 2019 19:58:02 GMT (203kb,D)
[v2] Sun, 16 Feb 2020 10:52:13 GMT (227kb,D)

Link back to: arXiv, form interface, contact.