about
Neural Optimizer Search with Reinforcement Learning [pdf] (arxiv.org)
1 point by stablemap on Sep 22, 2017 | hide | past | pdf | discuss on HN

In plain words: A program writes update rules as math formulas from basic pieces like gradients and running averages, earning a reward when models trained with them do well early. The rules it found were better than Adam, RMSProp, and SGD on image classification and helped on ImageNet and translation.

Abstract · Neural Optimizer Search with Reinforcement Learning

We present an approach to automate the process of discovering optimization methods, with a focus on deep learning architectures. We train a Recurrent Neural Network controller to generate a string in a domain specific language that describes a mathematical update equation based on a list of primitive functions, such as the gradient, running average of the gradient, etc. The controller is trained with Reinforcement Learning to maximize the performance of a model after a few epochs. On CIFAR-10, our method discovers several update rules that are better than many commonly used optimizers, such as Adam, RMSProp, or SGD with and without Momentum on a ConvNet model. We introduce two new optimizers, named PowerSign and AddSign, which we show transfer well and improve training on a variety of different tasks and architectures, including ImageNet classification and Google's neural machine translation system.

Irwan Bello, Barret Zoph, Vijay Vasudevan, Quoc V. Le
arXiv:1709.07417 · cs.AI, cs.LG, stat.ML · submitted Sep 21, 2017 · updated Sep 22, 2017
abstract · pdf · html · ICML 2017 Conference paper

add comment on HN