arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.02484cs.LGcs.AIcs.CVcs.NE

用于基于预测的神经架构搜索的编码

Encodings for Prediction-based Neural Architecture Search

Cornell University(康奈尔大学)

Yash Akhauri, Mohamed S. Abdelfattah

AI总结:

本文系统比较结构化、学习型和基于评分的NAS架构编码,提出统一编码与FLAN流注意力预测器,在多搜索空间逾150万架构上实现跨空间迁移并将预测器训练成本降低逾十倍。

摘要:

基于预测器的方法已显著增强了神经架构搜索(NAS)优化。这些预测器的有效性在很大程度上受到神经网络架构编码方式的影响。传统编码使用邻接矩阵描述神经网络的图结构,而新型编码则涵盖从潜在表示的无监督预训练到零成本代理向量等多种方法。本文将神经编码分为三大类并加以研究:结构化、学习型和基于评分的编码。此外,我们扩展了这些编码,并提出了统一编码,将NAS预测器扩展到多个搜索空间。我们的分析基于在NASBench-101(NB101)、NB201、NB301、Network Design Spaces(NDS)和TransNASBench-101等NAS空间中对超过150万个神经网络架构进行的实验。在研究基础上,我们提出了预测器FLAN:用于NAS的流注意力(Flow Attention for NAS)。FLAN整合了关于预测器设计、迁移学习和统一编码的关键洞见,使NAS精度预测器的训练成本降低超过一个数量级。我们的实现以及所有神经网络的编码已在https://github.com/abdelfattah-lab/flan_nas开源。

英文摘要:

Predictor-based methods have substantially enhanced Neural Architecture Search (NAS) optimization. The efficacy of these predictors is largely influenced by the method of encoding neural network architectures. While traditional encodings used an adjacency matrix describing the graph structure of a neural network, novel encodings embrace a variety of approaches from unsupervised pretraining of latent representations to vectors of zero-cost proxies. In this paper, we categorize and investigate neural encodings from three main types: structural, learned, and score-based. Furthermore, we extend these encodings and introduce \textit{unified encodings}, that extend NAS predictors to multiple search spaces. Our analysis draws from experiments conducted on over 1.5 million neural network architectures on NAS spaces such as NASBench-101 (NB101), NB201, NB301, Network Design Spaces (NDS), and TransNASBench-101. Building on our study, we present our predictor \textbf{FLAN}: \textbf{Fl}ow \textbf{A}ttention for \textbf{N}AS. FLAN integrates critical insights on predictor design, transfer learning, and \textit{unified encodings} to enable more than an order of magnitude cost reduction for training NAS accuracy predictors. Our implementation and encodings for all neural networks are open-sourced at \href{https://github.com/abdelfattah-lab/flan_nas}{https://github.com/abdelfattah-lab/flan\_nas}.

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

↑