InstaNAS: Instance-aware Neural Architecture Search
arXiv:1811.10201
Abstract
Conventional Neural Architecture Search (NAS) aims at finding a single architecture that achieves the best performance, which usually optimizes task related learning objectives such as accuracy. However, a single architecture may not be representative enough for the whole dataset with high diversity and variety. Intuitively, electing domain-expert architectures that are proficient in domain-specific features can further benefit architecture related objectives such as latency. In this paper, we propose InstaNAS---an instance-aware NAS framework---that employs a controller trained to search for a "distribution of architectures" instead of a single architecture; This allows the model to use sophisticated architectures for the difficult samples, which usually comes with large architecture related cost, and shallow architectures for those easy samples. During the inference phase, the controller assigns each of the unseen input samples with a domain expert architecture that can achieve high accuracy with customized inference costs. Experiments within a search space inspired by MobileNetV2 show InstaNAS can achieve up to 48.8% latency reduction without compromising accuracy on a series of datasets against MobileNetV2.
References in corpus (9)
- Adam: A Method for Stochastic Optimization
- UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
- Neural Architecture Search with Reinforcement Learning
- Improved Regularization of Convolutional Neural Networks with Cutout
- Large-Scale Evolution of Image Classifiers
- MONAS: Multi-Objective Neural Architecture Search using Reinforcement Learning
- Conditional Computation in Neural Networks for faster models
- Transfer Learning with Neural AutoML
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks