Highlight

DARTS: Differentiable Architecture Search

Introduces DARTS, a differentiable neural architecture search that relaxes the search space to be continuous, enabling efficient gradient-based search.

Based on

DARTS: Differentiable Architecture Search

By Hanxiao Liu, K. Simonyan, Yiming YangInternational Conference on Learning Representations
Read original article →

DARTS addresses the scalability challenge of neural architecture search by formulating the search itself in a differentiable manner. Rather than applying evolutionary algorithms or reinforcement learning over a discrete and non-differentiable search space, the method relaxes the architecture representation into a continuous form, which allows the architecture to be optimized efficiently using gradient descent.

Across extensive experiments on CIFAR-10, ImageNet, Penn Treebank, and WikiText-2, DARTS discovers high-performance convolutional architectures for image classification and recurrent architectures for language modeling. It does so while being orders of magnitude faster than state-of-the-art non-differentiable techniques, and the authors release their implementation to support further research on efficient architecture search.

Abstract

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.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

neural architecture searchdifferentiable optimizationgradient descentimage classificationlanguage modeling
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.