OptiProxy-NAS: Optimization Proxy based End-to-End Neural Architecture Search

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Lyu, Bo, Cui, Yu, Shi, Tuo, Li, Ke
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915483050049536
author Lyu, Bo
Cui, Yu
Shi, Tuo
Li, Ke
author_facet Lyu, Bo
Cui, Yu
Shi, Tuo
Li, Ke
contents Neural architecture search (NAS) is a hard computationally expensive optimization problem with a discrete, vast, and spiky search space. One of the key research efforts dedicated to this space focuses on accelerating NAS via certain proxy evaluations of neural architectures. Different from the prevalent predictor-based methods using surrogate models and differentiable architecture search via supernetworks, we propose an optimization proxy to streamline the NAS as an end-to-end optimization framework, named OptiProxy-NAS. In particular, using a proxy representation, the NAS space is reformulated to be continuous, differentiable, and smooth. Thereby, any differentiable optimization method can be applied to the gradient-based search of the relaxed architecture parameters. Our comprehensive experiments on $12$ NAS tasks of $4$ search spaces across three different domains including computer vision, natural language processing, and resource-constrained NAS fully demonstrate the superior search results and efficiency. Further experiments on low-fidelity scenarios verify the flexibility.
format Preprint
id arxiv_https___arxiv_org_abs_2509_05656
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle OptiProxy-NAS: Optimization Proxy based End-to-End Neural Architecture Search
Lyu, Bo
Cui, Yu
Shi, Tuo
Li, Ke
Machine Learning
Artificial Intelligence
Neural architecture search (NAS) is a hard computationally expensive optimization problem with a discrete, vast, and spiky search space. One of the key research efforts dedicated to this space focuses on accelerating NAS via certain proxy evaluations of neural architectures. Different from the prevalent predictor-based methods using surrogate models and differentiable architecture search via supernetworks, we propose an optimization proxy to streamline the NAS as an end-to-end optimization framework, named OptiProxy-NAS. In particular, using a proxy representation, the NAS space is reformulated to be continuous, differentiable, and smooth. Thereby, any differentiable optimization method can be applied to the gradient-based search of the relaxed architecture parameters. Our comprehensive experiments on $12$ NAS tasks of $4$ search spaces across three different domains including computer vision, natural language processing, and resource-constrained NAS fully demonstrate the superior search results and efficiency. Further experiments on low-fidelity scenarios verify the flexibility.
title OptiProxy-NAS: Optimization Proxy based End-to-End Neural Architecture Search
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.05656