A Stochastic Derivative-Free Optimization Method with Importance Sampling

Abstract

​​We consider the problem of unconstrained minimization of a smooth objective function in a setting where only function evaluations are possible. While importance sampling is one of the most popular techniques used by machine learning practitioners to accelerate the convergence of their models when applicable, there is not much existing theory for this acceleration in the derivative-free setting. In this paper, we propose an importance sampling version of the stochastic three points (STP) method proposed by Bergou et al. (2018) and derive new improved complexity results on non-convex, convex and strongly convex functions. We conduct extensive experiments on various synthetic and real LIBSVM datasets (Chang & Lin, 2011) confirming our theoretical results. We further test our method on a collection of continuous control tasks on several MuJoCo (Todorov et al., 2012) environments with varying difficulty. Our results suggest that STP is practical for high dimensional continuous control problems. Moreover, the proposed importance sampling version results in a significant sample complexity improvement.

Publication
arXiv
Date
Links