← 返回资源分享
On the Variance of the Adaptive Learning Rate and Beyond
论文
论文
发布时间2019-08-08
发表ICLR 2020 1 · arXiv:1908.03265
作者:Xiaodong Liu,Jianfeng Gao,Liyuan Liu,Jiawei Han,Pengcheng He,Weizhu Chen,Haoming Jiang
详细介绍
The learning rate warmup heuristic achieves remarkable success in stabilizing training, accelerating convergence and improving generalization for adaptive stochastic optimization algorithms like RMSprop and Adam. Here, we study its mechanism in details. Pursuing the theory behind warmup, we identify a problem of the adaptive learning rate (i.e., it has problematically large variance in the early stage), suggest warmup works as a variance reduction technique, and provide both empirical and theoretical evidence to verify our hypothesis. We further propose RAdam, a new variant of Adam, by introducing a term to rectify the variance of the adaptive learning rate. Extensive experimental results on image classification, language modeling, and neural machine translation verify our intuition and demonstrate the effectiveness and robustness of our proposed method. All implementations are available at: https://github.com/LiyuanLucasLiu/RAdam.
代码仓库 (24)
namisan/mt-dnn官方PyTorch
LiyuanLucasLiu/RAdam官方PyTorch
bond005/yandex-shifts-weatherTensorFlow
shun601/4th-tellus-satellite-challengePyTorch
kpe/params-flowTensorFlow
SdahlSean/RangerOptimizerTensorflowTensorFlow
nachiket273/lookahead_pytorchPyTorch
allen108108/Model-Optimizer_Implementation
mnikitin/RAdamMXNet
201419/Optimizer-PyTorchPyTorch
