arXiv · 1708.02544
Stochastic Optimization with Bandit Sampling
Abstract
Many stochastic optimization algorithms work by estimating the gradient of the cost function on the fly by sampling datapoints uniformly at random from a training set. However, the estimator might have a large variance, which inadvertently slows down the convergence rate of the algorithms. One way to reduce this variance is to sample the datapoints from a carefully selected non-uniform distribution. In this work, we propose a novel non-uniform sampling approach that uses the multi-armed bandit framework. Theoretically, we show that our algorithm asymptotically approximates the optimal variance within a factor of 3. Empirically, we show that using this datapoint-selection technique results in a significant reduction in the convergence time and variance of several stochastic optimization algorithms such as SGD, SVRG and SAGA. This approach for sampling datapoints is general, and can be used in conjunction with any algorithm that uses an unbiased gradient estimation -- we expect it to have broad applicability beyond the specific examples explored in this work.
Explore related subjects
Keep this discovery
Farnood Salehi, L. Elisa Celis, Patrick Thiran. 2017-08-08. Stochastic Optimization with Bandit Sampling. https://arxiv.org/abs/1708.02544
Cite the original work for its findings. Save a collection to share your selection of sources.