arrow
返回

Mini-batch algorithms with Barzilai-Borwein update step

delete2018-11-01
delete22
PRE
AI
杨壮 封面图
杨壮 (Zhuang Yang)
王成 封面图
王成 (Cheng Wang) *
Y
Yu Zang
J
Jonathan Li
DOI:10.1016/j.neucom.2018.06.002delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
As a way to accelerate stochastic schemes, mini-batch optimization has been a popular choice for large scale learning due to its good general performance and ease of parallel computing. However, the performance of mini-batch algorithms can vary significantly based on the choice of the step size sequence, and, in general, there is a paucity of guidance for making good choices. In this paper, we propose to use the Barzilai-Borwein (BB) update step to automatically compute step sizes for the state of the art mini-batch method (mini-batch semi-stochastic gradient descent (mS2GD) method), thereby obtaining a new optimization method: mS2GD-BB. We prove that mS2GD-BB converges linearly in expectation for non-smooth strongly convex objective functions. We analyze the complexity of mS2GD-BB and show that it achieves as fast a rate as modern stochastic gradient methods. Numerical experiments on standard data sets indicate that the performance of mS2GD-BB is superior to some state of the art methods. (C) 2018 Elsevier B.V. All rights reserved.
Keyword:
Stochastic optimization
Mini batches
Barzilai-Borwein method
Variance reduction
Convex optimization

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

X
xiamen university
学者数:
5.9W
论文数: 3.8W
被引数: 67
引用论文

引用论文

Neutron diffraction investigation of the evolution of the crystal structure of oxygen-conducting solid solutions (Yb1 − x Ca x )2Ti2O7 (x = 0, 0.05, 0.10)
err2009-02-08
err0
PREAI
errA. V. Shlyakhtina; A. E. Sokolov; V. A. Ul’yanov; V. A. Trunov; M. V. Boguslavskiĭ; A. V. Levchenko; L. G. Shcherbakova
err分享
err收藏
Reweighted stochastic learning
err2016-07-01
err1
PREAI
errJumutc, Vilen; Suykens, Johan A. K.
err分享
err收藏
Explicit and implicit reinforcement learning across the psychosis spectrum.
err2017-07-01
err0
errOAAI
errDeanna M. Barch; Cameron S. Carter; James M. Gold; Sheri L. Johnson; Ann M. Kring; Angus W. MacDonald; Diego A. Pizzagalli; J. Daniel Ragland; Steven M. Silverstein; Milton E. Strauss
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容