arrow
返回

On the step size selection in variance-reduced algorithm for nonconvex optimization

delete2021-05-01
delete8
PRE
AI
杨壮 封面图
杨壮 (Zhuang Yang) *
DOI:10.1016/j.eswa.2020.114336delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Nonconvex problems have recently received considerable attention in statistics, machine learning and vision areas. Variance reduction techniques like the SVRG-type and SARAH-type methods, originally proposed for convex optimization, have been proved to be effective for nonconvex optimization. The performance of variance-reduced stochastic gradient methods greatly depends on how step sizes are tuned and decreased over time. However, research on specifying online step size in nonconvex optimization is quite limited. Motivated by this gap, we propose using the random stabilized Barzilai-Borwein (RSBB) method to compute step size for variance-reduced stochastic gradient methods. In particular, we incorporate RSBB into the modern variance-reduced stochastic gradient method, the SARAH-type method, for nonconvex optimization. We theoretically prove that the proposed method converges sub-linearly for general nonconvex objective functions and converges linearly for gradient dominated functions. We also show that the complexity of our proposed method is comparable to modern stochastic gradient methods. To further show the efficacy of the RSBB method, we proposed MSVRG-RSBB, by introducing it into MSVRG (a variant of the SVRG method). We provide experimental results on various datasets to show the efficacy of our proposed methods.
Keyword:
Nonconvex optimization
Variance reduction
Online step size
Stochastic optimization
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Expert Systems with Applications 封面图
Expert Systems with Applications
IF:
7.5
论文数:
3.0W
被引数:
10.2W

机构

S
Sun Yat Sen University
学者数:
9.9W
论文数: 7.2W
被引数: 95
引用论文

引用论文

err分享
err收藏
err分享
err收藏
err分享
err收藏
Accelerated stochastic gradient descent with step size selection rules
err2019-06-01
err20
PREAI
errYang, Zhuang; Wang, Cheng; Zhang, Zhemin; Li, Jonathan
err分享
err收藏
学者 查看更多内容