Variation Matters: from Mitigating to Embracing Zero-Shot NAS Ranking Function Variation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rumiantsev, Pavel, Coates, Mark
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916633044320256
author Rumiantsev, Pavel
Coates, Mark
author_facet Rumiantsev, Pavel
Coates, Mark
contents Neural Architecture Search (NAS) is a powerful automatic alternative to manual design of a neural network. In the zero-shot version, a fast ranking function is used to compare architectures without training them. The outputs of the ranking functions often vary significantly due to different sources of randomness, including the evaluated architecture's weights' initialization or the batch of data used for calculations. A common approach to addressing the variation is to average a ranking function output over several evaluations. We propose taking into account the variation in a different manner, by viewing the ranking function output as a random variable representing a proxy performance metric. During the search process, we strive to construct a stochastic ordering of the performance metrics to determine the best architecture. Our experiments show that the proposed stochastic ordering can effectively boost performance of a search on standard benchmark search spaces.
format Preprint
id arxiv_https___arxiv_org_abs_2502_19657
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Variation Matters: from Mitigating to Embracing Zero-Shot NAS Ranking Function Variation
Rumiantsev, Pavel
Coates, Mark
Machine Learning
Neural Architecture Search (NAS) is a powerful automatic alternative to manual design of a neural network. In the zero-shot version, a fast ranking function is used to compare architectures without training them. The outputs of the ranking functions often vary significantly due to different sources of randomness, including the evaluated architecture's weights' initialization or the batch of data used for calculations. A common approach to addressing the variation is to average a ranking function output over several evaluations. We propose taking into account the variation in a different manner, by viewing the ranking function output as a random variable representing a proxy performance metric. During the search process, we strive to construct a stochastic ordering of the performance metrics to determine the best architecture. Our experiments show that the proposed stochastic ordering can effectively boost performance of a search on standard benchmark search spaces.
title Variation Matters: from Mitigating to Embracing Zero-Shot NAS Ranking Function Variation
topic Machine Learning
url https://arxiv.org/abs/2502.19657