arrow
返回

SALR: Sharpness-Aware Learning Rate Scheduler for Improved Generalization

delete2024-09-01
delete3
delete
OA
AI
X
Xubo Yue
M
Maher Nouiehed
R
Raed Al Kontar *
DOI:10.1109/TNNLS.2023.3263393delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
In an effort to improve generalization in deep learning and automate the process of learning rate scheduling, we propose SALR: a sharpness-aware learning rate update technique designed to recover flat minimizers. Our method dynamically updates the learning rate of gradient-based optimizers based on the local sharpness of the loss function. This allows optimizers to automatically increase learning rates at sharp valleys to increase the chance of escaping them. We demonstrate the effectiveness of SALR when adopted by various algorithms over a broad range of networks. Our experiments indicate that SALR improves generalization, converges faster, and drives solutions to significantly flatter regions.
Keyword:
Schedules
Deep learning
Neural networks
Convergence
Bayes methods
Training
Stochastic processes
generalization
learning rate schedule
sharpness

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

U
University of Michigan
学者数:
6.4W
论文数: 5.3W
被引数: 124
U
university of michigan system
学者数:
9.1W
论文数: 8.6W
被引数: 133
引用论文

引用论文

The termination and multilayer relaxation at the Co{1010} surface
err1990-09-01
err0
PREAI
errM. Lindroos; C.J. Barnes; P. Hu; D.A. King
err分享
err收藏
When Does Model-Based Control Pay Off?
err2016-08-26
err0
errOAAI
errWouter Kool; Fiery A. Cushman; Samuel J. Gershman
err分享
err收藏
Self-organised path formation in a swarm of robots
err2011-04-21
err0
PREAI
errValerio Sperati; Vito Trianni; Stefano Nolfi
err分享
err收藏
Why ResNet Works? Residuals Generalize
err2020-12-01
err173
errOAAI
errHe, Fengxiang; Liu, Tongliang; Tao, Dacheng
err分享
err收藏
Differentiation of Type 1 ILCs from a Common Progenitor to All Helper-like Innate Lymphoid Cell Lineages
errCell
IF0
err2014-04-01
err0
errOAAI
errChristoph S.N. Klose; Melanie Flach; Luisa Möhle; Leif Rogell; Thomas Hoyler; Karolina Ebert; Carola Fabiunke; Dietmar Pfeifer; Veronika Sexl; Diogo Fonseca-Pereira; Rita G. Domingues; Henrique Veiga-Fernandes; Sebastian J. Arnold; Meinrad Busslinger; Ildiko R. Dunay; Yakup Tanriver; Andreas Diefenbach
err分享
err收藏
学者 查看更多内容