LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search
summary
The gist
Detailed Research Summary of LLMasTool: A Hybrid NAS Framework for Open-Ended Architecture Search As a diligent researcher, I have meticulously analyzed the provided text snippets from the arXiv
In short
LLMasTool is a hybrid NAS framework that uses code mining to extract reusable modules and represents architectures as hierarchical trees. It employs a coarse-to-fine evolution strategy where an algorithm guides broad exploration, and an LLM acts only as a tool to make precise, valid tree transformations. This approach overcomes LLM biases by ensuring reliable architectural evolution without relying on the LLM to generate complex code directly.
Key concepts
- Hierarchical Tree Representation
- Architectures are structured like trees where nodes represent components and branches represent connections. This structure allows for precise, rule-based modifications rather than free-form text generation. It makes it possible to perform valid, executable edits on the architecture step-by-step.
- Algorithm-Based Module Mining
- This component automatically searches arbitrary source code to identify and extract reusable neural network modules. This process builds a reliable foundation for the search space by providing pre-defined, high-quality building blocks that can be reliably integrated into new architectures.
- Coarse-to-Fine Evolution Strategy
- The search process is split into two parts: a coarse level guided by an algorithm using Bayesian modeling for broad exploration, and a fine level where the LLM assists in resolving the remaining small details. This balances algorithmic control with LLM flexibility to ensure meaningful evolutionary steps.
Terminology used across episodes
This episode discusses
- LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search · Paper Radio
- A Downsampled Variant of ImageNet as an Alternative to the CIFAR datasets
- CoLLM-NAS: Collaborative Large Language Models for Efficient Knowledge-Guided Neural Architecture Search
- MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer
- -DARTS: Mitigating Performance Collapse by Harmonizing Operation Selection among Cells
- LeMo-NADe: Multi-Parameter Neural Architecture Discovery with LLMs
- Mixed-precision Supernet Training from Vision Foundation Models using Low Rank Adapter
- Qwen3 Technical Report
- SF-Mamba: Rethinking State Space Model for Vision · Paper Radio
- Can GPT-4 Perform Neural Architecture Search?
The paper
LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search · Read on arXiv
Sony Group Corporation
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.
Transcript
Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.
Tom: We're looking at the paper: "LLM as a Tool, Not an Agent".
Jane: Detailed Research Summary of LLMasTool: A Hybrid NAS Framework for Open-Ended Architecture Search As a diligent researcher,
Tom: First, who's behind it and why it matters.
Paper summary: Tom: So we’re talking about this paper now called "LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search". Basically they’re saying current AI tools that try to design whole architectures are messy and biased.
Jane: That sounds like a big problem because designing models is supposed to be automated, right? So what’s the main idea here for this paper?
Tom: The core thesis is that we shouldn't treat the LLM as an autonomous architect that just spits out a finished design. Instead, it’s better used as a tool inside a solid algorithmic structure.
Lu: Exactly. They argue that conventional search methods are too restricted, and agentic LLMs aren't reliable enough to guarantee you get something actually buildable for your computer to run >
Meng: So what’s the mechanism they use instead of just letting the LLM design everything? How do they keep it controlled?
Tom: They propose a framework called LLMasTool which uses hierarchical trees to represent the architectures. This lets them do reliable tree transformations instead of relying on messy code generation >
Jane: Hierarchical trees sound more structured than just looking at lines of code. Can you explain what that means for building a neural network?
Lu: It means they start by mining reusable modules from arbitrary source code to build a module database, which acts as the foundation for the search space >
Lalam: And this module database feeds into the tree representation where full architectures are built as these hierarchical trees, allowing evolution through tree transformations rather than just code generation >
Tom: So they take an existing structure, say a ResBlock or a ConvBlock from some source code, and they can swap, insert, or delete parts of it using these defined rules >
Jane: It sounds like they are using the LLM to help with the fine-grained stuff while the algorithm handles the big picture choices. Is that right?
Tom: That’s spot on. At each step of evolution, they have a coarse-level planning guided by a diversity-guided algorithm using Bayesian modeling to improve exploration efficiency >
Meng: How does that Bayesian part help them explore better when you don't want to just try everything randomly? What are the numbers we can look at?
Tom: They use Hierarchical Bayesian-Thompson Sampling, which they say improves the efficiency of open-ended exploration by using a history of past transformations within the tree structure >
Jane: So it’s not just throwing random ideas at it; it’s learning from what worked in previous steps during the search process >
Lu: The authors are showing that this approach bridges reliable algorithmic evolution with LLM assistance for stable and open-ended model evolution, moving past the limitations of current agentic approaches >
Tom: And on the results side, they show improvements over existing NAS methods. They claim gains of zero point six nine points on CIFAR-ten one point eight three points on CIFAR-one hundred and two point six eight points on ImageNet16-one hundred twenty >
Meng: Those are decent gains compared to the baselines they ran, but we have to remember those are just benchmarks. What about the limitations they admit? Where does this system stop working?
Jane: They mention that LLM-driven NAS often suffers from limited controllability and difficulty guaranteeing executability for complex designs, especially when the LLM is used as an autonomous agent for end-to-end generation >
Tom: And they also flag that proposals from LLMs tend to be biased toward recurring patterns in their training corpus, which limits truly open-ended exploration >
Lalam: They make it clear they aren't trying to merge the two paradigms into one thing, but integrating them through this tree representation is what makes the stability work >
Jane: So if I’m listening just for my commute or cooking instructions, what does this actually change for me?
Lu: It means we are getting architectures that are more robust and stable because they pass a check for executability via those tree transformations, not just a text prompt >
Tom: Basically, they prove that using the LLM as a precise editing tool inside an executable framework is better than letting it be the sole generator >
Meng: For deployment at scale, I’m still worried about how much work goes into building that initial module database and how stable those tree transformations are when you have millions of architectures to test >
Jane: That’s a fair point. So, what do we actually get from this paper as a concept for the future of AI design?
Lu: It suggests that instead of chasing full autonomy from the LLM, we should focus on creating these structured frameworks where the LLM assists with local decisions inside those constraints >
Tom: Yeah, so it’s about using code mining to get reliable starting points and then using that structure to guide a more intelligent search process >
Meng: It’s still complex setup work, but it shows a clear way to get better performance on those specific image classification tasks they tested, like ImageNet16-one hundred twenty >
Jane: We’ve seen how this paper uses code mining and tree representations to make the search space manageable and executable, which is a big step for practical AI design.
Conclusion: Tom: So we’re looking at the conclusion of this paper, "LLM as a Tool, Not an Agent: Code-Mined Tree Transformations for Neural Architecture Search." Basically, they're wrapping up how they used LLMs to search for neural network designs.
Jane: They’re arguing that you shouldn't just think of the LLM as a robot that builds the whole thing from scratch. They want it to be a precise tool working within a structured system.
Lu: That’s right, they re-framed the whole search process around tree transformations instead of just asking the LLM for code generation. It’s about controlled evolution.
Meng: So what does this mean for us when we're actually shipping models? Does it solve the problem of getting architectures that are reliable and executable?
Tom: It tackles that directly by using these tree structures, which guarantees validity through transformation rules rather than hoping the LLM writes perfect code.
Jane: It also uses a mix of things to guide the search. The algorithm handles the big picture exploration, and then the LLM just fills in the small gaps for those specific steps.
Lalam: I’m confident in this direction because it builds on established ways we represent architectures, but it adds that layer of executable constraint that current agentic systems lack.
Tom: They showed performance gains on CIFAR-ten and ImageNet16-one hundred twenty which is the number you have to look at. It proves the structure helps exploration more than just letting the LLM run wild.
Lu: The real impact here is showing that this hybrid approach—combining rule-based mining with LLM assistance in a tree framework—is a much more stable way to do open-ended NAS.
Jane: So, for someone who only listens, it means we’re looking at architectures that are more predictable and less likely to fail when we try to deploy them.
Meng: It still seems like there's a lot of setup work involved in building that initial module database before you can even start the search process.
Tom: Yeah, yeah, the infrastructure cost is real, but they proved the methodology works for getting those better results on standard tasks.
Lu: The future direction here is seeing how we scale these tree transformations across much larger models and more complex architectures.
More episodes
- Daily Summary for 2026-10-21
- 2504.02767-How Deep Do Large Language Models Internalize Scientific Literature and Citation Practices?
- 2512.14220-Estimating problem difficulty without ground truth using Large Language Model comparisons
- 2601.19532-Benchmarks Saturate When The Model Gets Smarter Than The Judge
- 2601.23163-Probing the Trajectories of Reasoning Traces in Large Language Models
- 2602.18918-Early Evidence of Vibe-Proving with Consumer LLMs: A Case Study on Spectral Region Characterization with ChatGPT-5.2 (Thinking)
- 2606.26130-Thinking Like a Scientist? A Structural Study of LLM-Generated Research Methods
- 2603.13083-Human-in-the-Loop LLM Grading for Handwritten Mathematics Assessments
- 2508.15842-Lexical Hints of Accuracy in LLM Reasoning Chains
- 2603.13562-Scalable Classification of Course Information Sheets Using Large Language Models: A Reusable Institutional Method for Academic Quality Assurance