arXiv:2505.07300cs.CV2025-05CVPR被引 4

提出新指标L-SWAG,让零样本搜索在视觉Transformer上更准更快。

L-SWAG: Layer-Sample Wise Activation with Gradients information for Zero-Shot NAS on Vision Transformers

  • 设计分层样本激活与梯度融合的评估指标,适配卷积与Transformer架构
  • 在14个任务上验证有效,搜索仅用0.1 GPU天即达ImageNet1k 17.0%错误率
  • 引入代理组合策略LIBRA-NAS,提升零样本NAS性能,适合快速架构筛选

无训练神经架构搜索(NAS)通过零成本(ZC)代理高效识别高性能网络。与多轮和单轮NAS不同,ZC-NAS兼具时间效率(无需训练)和可解释性(代理设计常有理论基础)。尽管该领域发展迅速,现有最先进ZC代理仍主要局限于经典卷积搜索空间。随着大语言模型重塑深度学习未来,本文将ZC代理扩展至视觉变换器(ViTs)。我们构建了基于Autoformer搜索空间的新基准,在6个不同任务上评估,并提出一种新型通用指标L-SWAG,该指标能刻画卷积与变压器架构的激活与梯度特征,覆盖14个任务。此外,先前研究指出不同代理蕴含互补信息,推动了机器学习模型用于代理组合的需求。为此,本文进一步提出LIBRA-NAS(低信息增益与偏差重对齐),一种战略性组合代理的方法,以最优表征特定基准。集成至NAS搜索流程后,LIBRA-NAS在仅0.1 GPU天内即发现测试误差为17.0%的ImageNet1k架构,优于进化与梯度基NAS方法。

原文摘要 · Abstract (English)

Training-free Neural Architecture Search (NAS) efficiently identifies high-performing neural networks using zero-cost (ZC) proxies. Unlike multi-shot and one-shot NAS approaches, ZC-NAS is both (i) time-efficient, eliminating the need for model training, and (ii) interpretable, with proxy designs often theoretically grounded. Despite rapid developments in the field, current SOTA ZC proxies are typically constrained to well-established convolutional search spaces. With the rise of Large Language Models shaping the future of deep learning, this work extends ZC proxy applicability to Vision Transformers (ViTs). We present a new benchmark using the Autoformer search space evaluated on 6 distinct tasks and propose Layer-Sample Wise Activation with Gradients information (L-SWAG), a novel, generalizable metric that characterizes both convolutional and transformer architectures across 14 tasks. Additionally, previous works highlighted how different proxies contain complementary information, motivating the need for a ML model to identify useful combinations. To further enhance ZC-NAS, we therefore introduce LIBRA-NAS (Low Information gain and Bias Re-Alignment), a method that strategically combines proxies to best represent a specific benchmark. Integrated into the NAS search, LIBRA-NAS outperforms evolution and gradient-based NAS techniques by identifying an architecture with a 17.0% test error on ImageNet1k in just 0.1 GPU days.

零样本NAS视觉Transformer架构搜索零成本代理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。