Given the recent impressive accomplishments of language models (LMs) for code generation, we explore the use of LMs as adaptive mutation and crossover operators for an evolutionary neural architecture search (NAS) algorithm. While NAS still proves too difficult a task for LMs to succeed at solely through prompting, we find that the combination of evolutionary prompt engineering with soft prompt-tuning, a method we term EvoPrompting, consistently finds diverse and high performing models. We first demonstrate that EvoPrompting is effective on the computationally efficient MNIST-1D dataset, where EvoPrompting produces convolutional architecture variants that outperform both those designed by human experts and naive few-shot prompting in terms of accuracy and model size. We then apply our method to searching for graph neural networks on the CLRS Algorithmic Reasoning Benchmark, where EvoPrompting is able to design novel architectures that outperform current state-of-the-art models on 21 out of 30 algorithmic reasoning tasks while maintaining similar model size. EvoPrompting is successful at designing accurate and efficient neural network architectures across a variety of machine learning tasks, while also being general enough for easy adaptation to other tasks beyond neural network design.
翻译:鉴于近期语言模型(LM)在代码生成方面取得的瞩目成就,我们探索将LM作为进化神经架构搜索(NAS)算法中的自适应变异和交叉算子。尽管NAS仍是一项对LM而言仅通过提示难以独立完成的挑战性任务,但我们发现,将进化式提示工程与软提示调优相结合(我们称之为EvoPrompting方法)能够持续发现多样且高性能的模型。我们首先在计算高效的MNIST-1D数据集上验证了EvoPrompting的有效性:该方法生成的卷积架构变体在准确率和模型规模上均优于人类专家设计的方案以及简单的少样本提示方法。随后,我们将该方法应用于CLRS算法推理基准上的图神经网络搜索,发现EvoPrompting能够在30个算法推理任务中的21个上设计出超越当前最先进模型的新型架构,同时保持相似的模型规模。EvoPrompting成功为多种机器学习任务设计了精准高效的神经网络架构,且具备足够通用性,可轻松适配神经网络设计以外的其他任务。