Trainless Model Performance Estimation for Neural Architecture Search

TL;DR

This paper proposes a new method for neural architecture search that estimates architecture performance based on the stability of untrained accuracy across initializations, reducing the need for training architectures.

Contribution

It introduces the coefficient of variation of untrained accuracy ($CV_{U}$) as a novel scoring metric for architecture selection, demonstrating its effectiveness across multiple datasets.

Findings

Lower $CV_{U}$ correlates with higher trained accuracy.

The method significantly reduces search time compared to traditional training-based approaches.

Architectures with minimal $CV_{U}$ achieve above-baseline performance on CIFAR-10, CIFAR-100, and ImageNet.

Abstract

Neural architecture search has become an indispensable part of the deep learning field. Modern methods allow to find one of the best performing architectures, or to build one from scratch, but they typically make decisions based on the trained accuracy information. In the present article we explore instead how the architectural component of a neural network affects its prediction power. We focus on relationships between the trained accuracy of an architecture and its accuracy prior to training, by considering statistics over multiple initialisations. We observe that minimising the coefficient of variation of the untrained accuracy, $CV_{U}$, consistently leads to better performing architectures. We test the $CV_{U}$ as a neural architecture search scoring metric using the NAS-Bench-201 database of trained neural architectures. The architectures with the lowest $CV_{U}$ value have on average an accuracy of $91.90 \pm 2.27$, $64.08 \pm 5.63$ and $38.76 \pm 6.62$ for CIFAR-10, CIFAR-100 and a downscaled version of ImageNet, respectively. Since these values are statistically above the random baseline, we make a conclusion that a good architecture should be stable against weights initialisations. It takes about $190$ s for CIFAR-10 and CIFAR-100 and $133.9$ s for ImageNet16-120 to process $100$ architectures, on a batch of $256$ images, with $100$ initialisations.