DARTS: Differentiable Architecture Search
Introduces DARTS, a differentiable neural architecture search that relaxes the search space to be continuous, enabling efficient gradient-based search.
DARTS tackles the scalability of neural architecture search by casting the task in a differentiable form. Instead of evolution or reinforcement learning over a discrete, non-differentiable space, it uses a continuous relaxation of the architecture representation so search runs efficiently via gradient descent. Experiments on CIFAR-10, ImageNet, Penn Treebank, and WikiText-2 show it finds high-performance convolutional architectures for image classification and recurrent ones for language modeling, orders of magnitude faster than non-differentiable methods.
Based on: DARTS: Differentiable Architecture Search · International Conference on Learning Representations
Curated by Aramai Editorial
Read summary →