arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1804.10587cs.LGcs.AIstat.ML

An improvement of the convergence proof of the ADAM-Optimizer

  • Ostbayerische Technische Hochschule (OTH) Regensburg(雷根斯堡东巴伐利亚技术大学)

机构由 AI 辅助整理,请以论文原文为准。

Sebastian Bock, Josef Goppold, Martin Weiß

更新

英文摘要:

A common way to train neural networks is the Backpropagation. This algorithm includes a gradient descent method, which needs an adaptive step size. In the area of neural networks, the ADAM-Optimizer is one of the most popular adaptive step size methods. It was invented in \cite{Kingma.2015} by Kingma and Ba. The $5865$ citations in only three years shows additionally the importance of the given paper. We discovered that the given convergence proof of the optimizer contains some mistakes, so that the proof will be wrong. In this paper we give an improvement to the convergence proof of the ADAM-Optimizer.

↑