Vision-Oriented Lightweight Neural Architecture Search with Budget-Adaptive Evaluation
Yi Fan, Yu-Bin Yang
Abstract
In the deep-learning-based computer vision community, Neural Architecture Search (NAS) has become the defacto tool for acquiring task-optimal network structures. Nevertheless, NAS methods are trapped in a fundamental accuracy-efficiency dilemma: training-based approaches deliver reliable performance but incur prohibitive search costs, whereas training-free strategies are ultra-fast but often yield relatively unreliable rankings. To reconcile this conflict, we propose a vision-oriented lightweight trainingbased NAS framework. We first design six micro vision tasks whose training time is negligible, yet together they probe a broad spectrum of representational capacities. Built upon these tasks, we introduce a budget-adaptive performance evaluator to produce the most accurate ranking attainable within the limit. Experiments on popular NAS benchmarks show that our method achieves a ranking correlation higher than existing methods. Furthermore, we construct a search space from prevalent neural blocks and run our method at a cost close to training-free methods; the discovered architecture surpasses the current state-of-the-art under identical training recipes. Our codes are available at https: //github.com/fanyi-plus/tf-nas.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 01be2ef4-73a3-41b8-b975-b2f96698be75Builds on30
- VMamba: Visual State Space ModelYue Liu, Yunjie Tian, Yuzhong Zhao, Hongtian Yu et al.NeurIPS 2024 · 3,199 citations
- Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space ModelLianghui Zhu, Bencheng Liao, Qian Zhang, Xinlong Wang et al.ICML 2024 · 1,725 citations
- Once-for-All: Train One Network and Specialize it for Efficient DeploymentHan Cai, Chuang Gan, Tianzhe Wang, Zhekai Zhang et al.ICLR 2020 · 1,522 citations
- TransReID: Transformer-based Object Re-IdentificationShuting He, Hao Luo, Pichao Wang, Fan Wang et al.ICCV 2021 · 1,172 citations
- Pruning neural networks without any data by iteratively conserving synaptic flowHidenori Tanaka, Daniel Kunin, Daniel L. K. Yamins, Surya GanguliNeurIPS 2020 · 884 citations
Related papers
- Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired PerspectiveWuyang Chen, Xinyu Gong, Zhangyang WangICLR 2021 · 51 citations
- GP-NAS: Gaussian Process Based Neural Architecture SearchZhihang Li, Teng Xi, Jiankang Deng, Gang Zhang et al.CVPR 2020
- Robustifying and Boosting Training-Free Neural Architecture SearchZhenfeng He, Yao Shu, Zhongxiang Dai, Bryan Kian Hsiang LowICLR 2024 · 8 citations
- Speedy Performance Estimation for Neural Architecture SearchRobin Ru, Clare Lyle, Lisa Schut, Miroslav Fil et al.NeurIPS 2021 · 50 citations
- Training-free Transformer Architecture SearchQinqin Zhou, Kekai Sheng, Xiawu Zheng, Ke Li et al.CVPR 2022 · 51 citations
