Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search

About

Neural Architecture Search (NAS) aims to automatically discover high-performing deep neural network (DNN) architectures. However, conventional algorithm-driven NAS relies on carefully hand-crafted search spaces to ensure executability, which restricts open-ended exploration. Recent coding-based agentic approaches using large language models (LLMs) reduce manual design, but current LLMs struggle to reliably generate complex, valid architectures, and their proposals are often biased toward a narrow set of patterns observed in their training data. To bridge reliable algorithmic search with powerful LLM assistance, we propose LLMasTool, a hierarchical tree-based NAS framework for stable and open-ended model evolution. Our method automatically extracts reusable modules from arbitrary source code and represents full architectures as hierarchical trees, enabling evolution through reliable tree transformations rather than code generation. At each evolution step, coarse-level planning is governed by a diversity-guided algorithm that leverages Bayesian modeling to improve exploration efficiency, while the LLM resolves the remaining degrees of freedom to ensure a meaningful evolutionary trajectory and an executable generated architecture. With this formulation, instead of fully agentic LLM approaches, our method explores diverse directions beyond the inherent biases in the LLM. Our method improves over existing NAS methods by 0.69, 1.83, and 2.68 points on CIFAR-10, CIFAR-100, and ImageNet16-120, demonstrating its effectiveness.

Masakazu Yoshimura, Zitang Sun, Yuiko Sakuma, Junji Otsuka, Atsushi Irie, Takeshi Ohashi• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 NAS-Bench-201 (test)
Accuracy95.31
225
Image ClassificationCIFAR-100 NAS-Bench-201 (test)
Accuracy77.83
198
Image ClassificationCIFAR-10 NAS-Bench-201 (val)
Accuracy93.06
169
Image ClassificationImageNet-16-120 NAS-Bench-201 (test)
Accuracy53.2
167
Image ClassificationCIFAR-100 NAS-Bench-201 (val)
Accuracy77.51
139
Image ClassificationImageNet 16-120 NAS-Bench-201 (val)
Accuracy52.3
123
Showing 6 of 6 rows

Other info

Follow for update