情報処理学会 第83回全国大会 会期:2021年3月18日~20日 会場:オンライン開催 情報処理学会 第83回全国大会 会期:2021年3月18日~20日 会場:オンライン開催

2J-03
学習初期に学習率の分散を考慮した学習率の範囲を動的に制御するAdam
○行木大輝,山口 智(千葉工大)
ニューラルネットワークの最適化アルゴリズムには「速度」と「安定性」が常に求められている。Adamはそれらを満たす手法として考えられていたが、大規模な問題ではSGDに比べて汎化性能が悪くなることや、学習初期に極端な学習率を選択してしまうなど「安定性」に課題が確認されている。近年ではこれらの問題を解決するために学習中にAdamからSGDに切り替えるAdaboundや、学習初期での学習率の分散を考慮し、小さな学習率でパラメータを更新するWarmUpを取り入れたRAdamが提案されている。本研究ではAdaboundに対して学習初期にWarmUpを適用することでより効率的な学習を行う手法を提案する。