about
Neural Networks Designed Using Reinforcement Learning (arxiv.org)
2 points by jbpetersen on Dec 5, 2016 | hide | past | pdf | discuss on HN

In plain words: A computer agent builds image-recognizing networks by choosing layers one at a time and learning from trial and error, replacing hand-tuning by experts. Its designs beat hand-built networks using the same standard layers and other automatic design systems.

Abstract · Designing Neural Network Architectures using Reinforcement Learning

At present, designing convolutional neural network (CNN) architectures requires both human expertise and labor. New architectures are handcrafted by careful experimentation or modified from a handful of existing networks. We introduce MetaQNN, a meta-modeling algorithm based on reinforcement learning to automatically generate high-performing CNN architectures for a given learning task. The learning agent is trained to sequentially choose CNN layers using $Q$-learning with an $ε$-greedy exploration strategy and experience replay. The agent explores a large but finite space of possible architectures and iteratively discovers designs with improved performance on the learning task. On image classification benchmarks, the agent-designed networks (consisting of only standard convolution, pooling, and fully-connected layers) beat existing networks designed with the same layer types and are competitive against the state-of-the-art methods that use more complex layer types. We also outperform existing meta-modeling approaches for network design on image classification tasks.

Bowen Baker, Otkrist Gupta, Nikhil Naik, Ramesh Raskar
arXiv:1611.02167 · cs.LG · submitted Nov 7, 2016 · updated Mar 22, 2017
abstract · pdf · html

add comment on HN
Also discussed: Jan 2017 (5 points, 0 comments)