arXiv:2605.15649cs.LGcs.NE2026-05中稿 · GECCO 2026

用代码文本让语言模型直接提取网络结构特征,省去调优开销。

Towards Code-Oriented LM Embeddings for Surrogate-Assisted Neural Architecture Search

论文配图:Towards Code-Oriented LM Embeddings for Surrogate-Assisted Neural Architecture Search
图 1 · 摘自论文原文
  • 将网络结构转为PyTorch代码文本,用现成语言模型提取特征。
  • 在NAS-Bench-201上,用代码输入比其他文本编码更准,提升预测性能。
  • 替换传统结构编码后,找最优模型所需计算量减少34%。

为神经架构搜索(NAS)构建高效代理模型通常需要昂贵的微调或复杂表示工程。本文提出一种低成本嵌入策略,利用语言模型(LM)的归纳偏置,消除这些开销。通过将架构表示为PyTorch类定义文本,我们证明无需针对NAS微调的现成语言模型即可作为竞争性特征提取器。最终预测器由提取的代码导向语言模型嵌入(COLE)通过轻量回归头构建。我们还研究了提升嵌入质量与利用率的策略。在NAS-Bench-201和einspace搜索空间上的实验表明,使用冻结语言模型时,原始代码输入比其他文本编码(如ONNX-to-text)表现更优。在NAS-Bench-201上,使用BANANAS算法时,COLE驱动的代理辅助搜索效果更佳。当优化CIFAR-100性能时,用COLE替代结构路径编码,使达到搜索空间中最佳架构99%准确率所需的评估预算减少34%。由于任何神经网络均可表示为代码,这些发现确立了COLE作为推进NAS的通用高效基础。

原文摘要 · Abstract (English)

Developing effective surrogates (performance predictors) for Neural Architecture Search (NAS) typically requires expensive fine-tuning or the engineering of complex representations. We propose a low-cost embedding strategy that leverages the inductive bias of Language Models (LMs) to eliminate these overheads. By representing architectures as PyTorch class definition text, we demonstrate that off-the-shelf LMs act as competitive feature extractors without NAS-specialized fine-tuning. The final predictor is constructed by passing the extracted Code-Oriented LM Embeddings (COLE) through a lightweight regression head. We also investigate strategies to improve embedding quality and utilization. Our experiments on the NAS-Bench-201 and einspace search spaces reveal that raw code inputs yield higher predictive performance than other text-based encodings (e.g., ONNX-to-text encodings) when using frozen LMs. We also observe COLE drives superior surrogate-assisted search using the BANANAS algorithm in NAS-Bench-201. When optimizing for CIFAR-100 performance, replacing structural path encodings with COLE for architecture representation allows for a 34% decrease in the evaluation budget required to reach within 1% of the fittest architecture in the search space (by test accuracy). As any neural architecture can be represented as code, these findings establish COLE as a versatile and efficient foundation for advancing NAS.

神经架构搜索代码嵌入语言模型代理模型

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