Vision-Oriented Lightweight Neural Architecture Search with Budget-Adaptive Evaluation
Yi Fan, Yu-Bin Yang
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper30
- VMamba: Visual State Space ModelYue Liu, Yunjie Tian, Yuzhong Zhao, Hongtian Yu 等NeurIPS 2024 · 被引用 3,199 次
- Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space ModelLianghui Zhu, Bencheng Liao, Qian Zhang, Xinlong Wang 等ICML 2024 · 被引用 1,725 次
- Once-for-All: Train One Network and Specialize it for Efficient DeploymentHan Cai, Chuang Gan, Tianzhe Wang, Zhekai Zhang 等ICLR 2020 · 被引用 1,522 次
- TransReID: Transformer-based Object Re-IdentificationShuting He, Hao Luo, Pichao Wang, Fan Wang 等ICCV 2021 · 被引用 1,172 次
- Pruning neural networks without any data by iteratively conserving synaptic flowHidenori Tanaka, Daniel Kunin, Daniel L. K. Yamins, Surya GanguliNeurIPS 2020 · 被引用 884 次
相关 Paper
- Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired PerspectiveWuyang Chen, Xinyu Gong, Zhangyang WangICLR 2021 · 被引用 51 次
- GP-NAS: Gaussian Process Based Neural Architecture SearchZhihang Li, Teng Xi, Jiankang Deng, Gang Zhang 等CVPR 2020
- Robustifying and Boosting Training-Free Neural Architecture SearchZhenfeng He, Yao Shu, Zhongxiang Dai, Bryan Kian Hsiang LowICLR 2024 · 被引用 8 次
- Speedy Performance Estimation for Neural Architecture SearchRobin Ru, Clare Lyle, Lisa Schut, Miroslav Fil 等NeurIPS 2021 · 被引用 50 次
- Training-free Transformer Architecture SearchQinqin Zhou, Kekai Sheng, Xiawu Zheng, Ke Li 等CVPR 2022 · 被引用 51 次
