One-shot neural architecture search (NAS) is an efficient and widely adopted method for powerful networks. However, it suffers from inconsistency, where sub-networks inheriting supernet weights exhibit different performance rankings compared to networks trained independently from scratch. To address this issue, we present a theoretical analysis showing that the discrepancy in performance ranking between sub-networks and their optimal counterparts is strongly correlated with the gradient norm and weight variance of the supernet. Based on these insights, we propose two regularization techniques and introduce Oneshotimizer, a novel method designed to enhance the consistency and effectiveness of one-shot NAS. Experiments demonstrate that Oneshotimizer significantly improves both ranking consistency and search performance. Specifically, it achieves a 14.8% increase in Kendall’s \(\uptau \) over SPOS on NAS-Bench-201, and a state-of-the-art 77.8% Top-1 accuracy with only 395M FLOPs on ImageNet.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Oneshotimizer: Consistent and Effective NAS via Regularizing Gradient Norm and Weight Variance

  • Longxing Yang,
  • Xiaofeng Li,
  • Xiaogang Dong,
  • Bin Gu

摘要

One-shot neural architecture search (NAS) is an efficient and widely adopted method for powerful networks. However, it suffers from inconsistency, where sub-networks inheriting supernet weights exhibit different performance rankings compared to networks trained independently from scratch. To address this issue, we present a theoretical analysis showing that the discrepancy in performance ranking between sub-networks and their optimal counterparts is strongly correlated with the gradient norm and weight variance of the supernet. Based on these insights, we propose two regularization techniques and introduce Oneshotimizer, a novel method designed to enhance the consistency and effectiveness of one-shot NAS. Experiments demonstrate that Oneshotimizer significantly improves both ranking consistency and search performance. Specifically, it achieves a 14.8% increase in Kendall’s \(\uptau \) over SPOS on NAS-Bench-201, and a state-of-the-art 77.8% Top-1 accuracy with only 395M FLOPs on ImageNet.