Neural Optimizer Search using 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 specific domain 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. These optimizers can also be transferred to perform well on different neural network architectures, including Google’s neural machine translation system.
Code (0)
등록된 구현이 없습니다.
Tasks
Machine Translationreinforcement-learningReinforcement LearningReinforcement Learning (RL)TranslationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
MolOpt: Autonomous Molecular Geometry Optimization using Multi-Agent Reinforcement Learning
In this paper, we propose MolOpt, the first attempt of its kind to use Multi-Agent Reinforcement Learning (MARL) for autonomous molecular geometry optimization (MGO). Typically MGO algorithms are hand-designed, but MolOp…
3D geometryComputational chemistryMolecular geometry optimizationMulti-agent Reinforcement Learning+5Learning to Optimize for Reinforcement Learning
In recent years, by leveraging more data, computation, and diverse tasks, learned optimizers have achieved remarkable success in supervised learning, outperforming classical hand-designed optimizers. Reinforcement learni…
Inductive BiasMeta-Learningreinforcement-learningReinforcement Learning+1Neural 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 la…
Machine Translationreinforcement-learningReinforcement LearningReinforcement Learning (RL)+1In Defense of the Unitary Scalarization for Deep Multi-Task Learning
Recent multi-task learning research argues against unitary scalarization, where training simply minimizes the sum of the task losses. Several ad-hoc multi-task optimization algorithms have instead been proposed, inspired…
Multi-Task LearningReinforcement Learning (RL)AIOptimizer - Software performance optimisation prototype for cost minimisation
This study presents AIOptimizer, a prototype for a cost-reduction-based software performance optimisation tool. The study focuses on the design elements of AIOptimizer, including user-friendliness, scalability, accuracy,…
Fault DiagnosisRecommendation Systemsreinforcement-learning