arrow
返回

Incremental PID Controller-Based Learning Rate Scheduler for Stochastic Gradient Descent

delete2024-05-01
delete14
PRE
AI
Z
Zenghui Wang
J
Jun Zhang *
DOI:10.1109/TNNLS.2022.3213677delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
As we all know, the learning rate plays a vital role in deep neural network (DNN) training. This study introduces an incremental proportional-integral-derivative (PID) controller widely used in automatic control as a learning rate scheduler for stochastic gradient descent (SGD). To automatically calculate the current learning rate, we utilize feedback control to determine the relationship between training losses and learning rates, named incremental PID learning rates, which include PID-Base and PID-Warmup. The new schedulers reduce the dependence on the initial learning rate and achieve higher accuracy. Compared with multistep learning rates (MSLR), cyclical learning rates (CLR), and SGD with warm restarts (SGDR), incremental PID learning rates based on feedback control obtain higher accuracy on CIFAR-10, CIFAR-100, and Tiny-ImageNet-200. We believe that our methods can improve the performance of SGD.
Keyword:
Training
PD control
PI control
Feedback control
Convergence
Oscillators
Optimization
incremental proportional-integral-derivative (PID) controller
learning rate scheduler
stochastic gradient descent (SGD)

期刊

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

机构

A
anhui university
学者数:
1.9W
论文数: 1.2W
被引数: 24
引用论文

引用论文

CLINICAL APPROACH TO DIAGNOSIS OF SYNCOPE
err1997-05-01
err0
PREAI
errDavid G. Benditt; Keith G. Lurie; William H. Fabian
err分享
err收藏
When Does Model-Based Control Pay Off?
err2016-08-26
err0
errOAAI
errWouter Kool; Fiery A. Cushman; Samuel J. Gershman
err分享
err收藏
Synthesis of n-type semiconducting diamond film using diphosphorus pentaoxide as the doping source以五氧化二磷为掺杂源合成n型半导体金刚石膜
err1990-10-01
err0
PREAI
errKen Okano; Hideo Kiyota; Tatsuya Iwasaki; Yoshitaka Nakamura; Yukio Akiba; Tateki Kurosu; Masamori Iida; Terutaro Nakamura
err分享
err收藏
Thermo-mechanical Characterization of Insulating Bio-plasters Containing Recycled Volcanic Pyroclasts
err2020-06-05
err0
errOAAI
errLoredana Contrafatto; Salvatore Gazzo; Antonio Purrazzo; Antonio Gagliano
err分享
err收藏
学者 查看更多内容