← 返回资源分享
Efficient Neural Architecture Search via Parameter Sharing
论文
论文
发布时间2018-02-09
发表arXiv:1802.03268
作者:Hieu Pham,Melody Y. Guan,Barret Zoph,Quoc V. Le,Jeff Dean
详细介绍
We propose Efficient Neural Architecture Search (ENAS), a fast and
inexpensive approach for automatic model design. In ENAS, a controller learns
to discover neural network architectures by searching for an optimal subgraph
within a large computational graph. The controller is trained with policy
gradient to select a subgraph that maximizes the expected reward on the
validation set. Meanwhile the model corresponding to the selected subgraph is
trained to minimize a canonical cross entropy loss. Thanks to parameter sharing
between child models, ENAS is fast: it delivers strong empirical performances
using much fewer GPU-hours than all existing automatic model design approaches,
and notably, 1000x less expensive than standard Neural Architecture Search. On
the Penn Treebank dataset, ENAS discovers a novel architecture that achieves a
test perplexity of 55.8, establishing a new state-of-the-art among all methods
without post-training processing. On the CIFAR-10 dataset, ENAS designs novel
architectures that achieve a test error of 2.89%, which is on par with NASNet
(Zoph et al., 2018), whose test error is 2.65%.
代码仓库 (29)
Ezereal/enasTensorFlow
MengTianjian/enas-pytorchPyTorch
google-research/google-research/tree/master/enas_lmTensorFlow
cshannonn/blackscholes_nas
aymanshams07/enas_cifar10TensorFlow
kaileymonn/Quantized-ENAS-ConvNetsTensorFlow
CiscoAI/amlaTensorFlow
RualPerez/AutoMLPyTorch
zbyte64/pytorch-dagsearchPyTorch
yashkant/ENAS-Quantized-Neural-NetworksTensorFlow
