arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2211.12759cs.CVcs.AIcs.LG
收藏

NAS-LID: Efficient Neural Architecture Search with Local Intrinsic Dimension

Hong Kong Baptist University(香港浸会大学);Shanghai Jiao Tong University(上海交通大学);Shanghai AI Laboratory(上海人工智能实验室);NVIDIA AI Tech Center(英伟达人工智能技术中心);The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州));Mahindra University(马恒达大学);Coventry University(考文垂大学)

Xin He, Jiangchao Yao, Yuxin Wang, Zhenheng Tang, Ka Chu Cheung, Simon See, Bo Han, Xiaowen Chu

英文摘要:

One-shot neural architecture search (NAS) substantially improves the search efficiency by training one supernet to estimate the performance of every possible child architecture (i.e., subnet). However, the inconsistency of characteristics among subnets incurs serious interference in the optimization, resulting in poor performance ranking correlation of subnets. Subsequent explorations decompose supernet weights via a particular criterion, e.g., gradient matching, to reduce the interference; yet they suffer from huge computational cost and low space separability. In this work, we propose a lightweight and effective local intrinsic dimension (LID)-based method NAS-LID. NAS-LID evaluates the geometrical properties of architectures by calculating the low-cost LID features layer-by-layer, and the similarity characterized by LID enjoys better separability compared with gradients, which thus effectively reduces the interference among subnets. Extensive experiments on NASBench-201 indicate that NAS-LID achieves superior performance with better efficiency. Specifically, compared to the gradient-driven method, NAS-LID can save up to 86% of GPU memory overhead when searching on NASBench-201. We also demonstrate the effectiveness of NAS-LID on ProxylessNAS and OFA spaces. Source code: https://github.com/marsggbo/NAS-LID.

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

↑