arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.19232cs.CVcs.LG
收藏

AZ-NAS:组装零成本代理用于网络架构搜索

AZ-NAS: Assembling Zero-Cost Proxies for Network Architecture Search

Yonsei University(延世大学);Korea Institute of Science and Technology (KIST)(韩国科学技术研究院)

Junghyup Lee, Bumsub Ham

AI总结:

AZ-NAS通过集成四种互补的零成本代理并提出非线性排名聚合方法,显著提升了免训练NAS中网络排名与真实性能的相关性,在标准基准上以合理运行成本超越现有方法。

摘要:

免训练的网络架构搜索(NAS)旨在利用零成本代理发现高性能网络,这些代理能够捕获与最终性能相关的网络特性。然而,以往免训练NAS方法估计的网络排名与性能之间仅表现出较弱的相关性。为了解决这一问题,我们提出了AZ-NAS,这是一种新颖的方法,利用多种零成本代理的集成,在性能方面大幅增强网络预测排名与真实排名之间的相关性。为此,我们引入了四种新颖的零成本代理,它们彼此互补,分别从表达性、渐进性、可训练性和复杂性等不同视角分析架构的不同特征。这些代理分数可以在一次前向和反向传播中同时获得,从而使整个NAS过程高度高效。为了有效整合我们的代理所预测的排名,我们引入了一种非线性排名聚合方法,该方法突出在所有代理中一致获得高排名的网络。实验结果明确证明了AZ-NAS的有效性和高效性,在标准基准上超越了最先进的方法,同时保持了合理的运行时间成本。

英文摘要:

Training-free network architecture search (NAS) aims to discover high-performing networks with zero-cost proxies, capturing network characteristics related to the final performance. However, network rankings estimated by previous training-free NAS methods have shown weak correlations with the performance. To address this issue, we propose AZ-NAS, a novel approach that leverages the ensemble of various zero-cost proxies to enhance the correlation between a predicted ranking of networks and the ground truth substantially in terms of the performance. To achieve this, we introduce four novel zero-cost proxies that are complementary to each other, analyzing distinct traits of architectures in the views of expressivity, progressivity, trainability, and complexity. The proxy scores can be obtained simultaneously within a single forward and backward pass, making an overall NAS process highly efficient. In order to integrate the rankings predicted by our proxies effectively, we introduce a non-linear ranking aggregation method that highlights the networks highly-ranked consistently across all the proxies. Experimental results conclusively demonstrate the efficacy and efficiency of AZ-NAS, outperforming state-of-the-art methods on standard benchmarks, all while maintaining a reasonable runtime cost.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑