arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.12243cs.LGmath.OCstat.ML

Adaptive Importance Sampling for Finite-Sum Optimization and Sampling with Decreasing Step-Sizes

McGill University(麦吉尔大学)

Ayoub El Hanchi, David A. Stephens

英文摘要:

Reducing the variance of the gradient estimator is known to improve the convergence rate of stochastic gradient-based optimization and sampling algorithms. One way of achieving variance reduction is to design importance sampling strategies. Recently, the problem of designing such schemes was formulated as an online learning problem with bandit feedback, and algorithms with sub-linear static regret were designed. In this work, we build on this framework and propose Avare, a simple and efficient algorithm for adaptive importance sampling for finite-sum optimization and sampling with decreasing step-sizes. Under standard technical conditions, we show that Avare achieves $\mathcal{O}(T^{2/3})$ and $\mathcal{O}(T^{5/6})$ dynamic regret for SGD and SGLD respectively when run with $\mathcal{O}(1/t)$ step sizes. We achieve this dynamic regret bound by leveraging our knowledge of the dynamics defined by the algorithm, and combining ideas from online learning and variance-reduced stochastic optimization. We validate empirically the performance of our algorithm and identify settings in which it leads to significant improvements.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑