HyperNAS: Enhancing Architecture Representation for NAS Predictor via Hypernetwork

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lv, Jindi, Zhou, Yuhao, Tian, Yuxin, Ye, Qing, Feng, Wentao, Lv, Jiancheng
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912599880237056
author Lv, Jindi
Zhou, Yuhao
Tian, Yuxin
Ye, Qing
Feng, Wentao
Lv, Jiancheng
author_facet Lv, Jindi
Zhou, Yuhao
Tian, Yuxin
Ye, Qing
Feng, Wentao
Lv, Jiancheng
contents Time-intensive performance evaluations significantly impede progress in Neural Architecture Search (NAS). To address this, neural predictors leverage surrogate models trained on proxy datasets, allowing for direct performance predictions for new architectures. However, these predictors often exhibit poor generalization due to their limited ability to capture intricate relationships among various architectures. In this paper, we propose HyperNAS, a novel neural predictor paradigm for enhancing architecture representation learning. HyperNAS consists of two primary components: a global encoding scheme and a shared hypernetwork. The global encoding scheme is devised to capture the comprehensive macro-structure information, while the shared hypernetwork serves as an auxiliary task to enhance the investigation of inter-architecture patterns. To ensure training stability, we further develop a dynamic adaptive multi-task loss to facilitate personalized exploration on the Pareto front. Extensive experiments across five representative search spaces, including ViTs, demonstrate the advantages of HyperNAS, particularly in few-shot scenarios. For instance, HyperNAS strikes new state-of-the-art results, with 97.60\% top-1 accuracy on CIFAR-10 and 82.4\% top-1 accuracy on ImageNet, using at least 5.0$\times$ fewer samples.
format Preprint
id arxiv_https___arxiv_org_abs_2509_18151
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle HyperNAS: Enhancing Architecture Representation for NAS Predictor via Hypernetwork
Lv, Jindi
Zhou, Yuhao
Tian, Yuxin
Ye, Qing
Feng, Wentao
Lv, Jiancheng
Machine Learning
Artificial Intelligence
Time-intensive performance evaluations significantly impede progress in Neural Architecture Search (NAS). To address this, neural predictors leverage surrogate models trained on proxy datasets, allowing for direct performance predictions for new architectures. However, these predictors often exhibit poor generalization due to their limited ability to capture intricate relationships among various architectures. In this paper, we propose HyperNAS, a novel neural predictor paradigm for enhancing architecture representation learning. HyperNAS consists of two primary components: a global encoding scheme and a shared hypernetwork. The global encoding scheme is devised to capture the comprehensive macro-structure information, while the shared hypernetwork serves as an auxiliary task to enhance the investigation of inter-architecture patterns. To ensure training stability, we further develop a dynamic adaptive multi-task loss to facilitate personalized exploration on the Pareto front. Extensive experiments across five representative search spaces, including ViTs, demonstrate the advantages of HyperNAS, particularly in few-shot scenarios. For instance, HyperNAS strikes new state-of-the-art results, with 97.60\% top-1 accuracy on CIFAR-10 and 82.4\% top-1 accuracy on ImageNet, using at least 5.0$\times$ fewer samples.
title HyperNAS: Enhancing Architecture Representation for NAS Predictor via Hypernetwork
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.18151