AIDE: an LLM-driven agent that searches the space of code via tree search to automate ML engineering and beat many human baselines

February 18, 20258 min

Overview

Decision SnapshotNeeds Validation

AIDE is practical now for automating many ML engineering tasks if you can call LLM APIs. Results are supported by multiple benchmarks, but there are caveats (data contamination risk, uneven per-task gains, and challenges with large codebases).

Citations3

Evidence Strength0.70

Confidence0.80

Risk Signals10

Trust Signals

Findings with numeric evidence: 5/5

Findings with evidence refs: 5/5

Results with explicit delta: 3/5

Reproducibility

Status: Partial assets available

Open source: Partial

At A Glance

Cost impact: 50%

Production readiness: 60%

Novelty: 60%

Authors

Zhengyao Jiang, Dominik Schmidt, Dhruv Srikanth, Dixing Xu, Ian Kaplan, Deniss Jacenko, Yuxiang Wu

Links

Abstract / PDF / Code

Why It Matters For Business

AIDE can automate the repetitive trial-and-error of ML engineering, producing competitive models faster and often cheaper than manual work or traditional AutoML when you have LLM API access.

Who Should Care

Summary TLDR

AIDE is an agent that uses large language models to search directly in the space of code (Python scripts) with a tree-search workflow. It keeps a tree of code variants and applies three LLM-driven operations—draft, debug, improve—while summarizing past attempts to keep prompts small. Evaluations show AIDE outperforms traditional AutoML and other agent baselines on curated Kaggle tasks and improves medal rates on MLE-Bench. It is publicly released on GitHub but requires LLM APIs and compute. Key risks: training-data contamination, local optima from greedy search, and limits on large codebases.

Problem Statement

Building high-performance ML models is dominated by trial-and-error coding and tuning. Classic AutoML searches fixed hyperparameter spaces and needs expert design. With modern LLMs that can generate and fix code, we can instead search the space of scripts directly, but must manage prompt size and iterative history effectively.

Main Contribution

Introduce AIDE: an LLM-powered agent that frames ML engineering as code-space optimization using a solution tree.

Define practical operators: draft, debug, improve; plus a summarization operator to keep LLM prompts concise.

Key Findings

On 16 tabular Kaggle tasks (Weco-Kaggle Lite), AIDE outperformed about half of human competitors on average.

NumbersExceeds % of humans = 51.38%; Above Median = 50.0%

Practical UseRun AIDE on small-to-medium tabular problems to reach at-or-above median human performance without hand tuning.

Evidence RefTable 1 / Table 2 (Weco-Kaggle Lite)

On the larger Weco-Kaggle set AIDE remains competitive but variable across tasks.

NumbersAverage Exceeds % = 48.23%; Above Median = 49.21%

Practical UseExpect strong results on many datasets but uneven performance; validate per-task rather than assume uniform gains.

Evidence RefFull Weco-Kaggle results (Section 4.1, Figure 2)

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Exceeds % of humans (Weco-Kaggle Lite)51.38%Weco-Kaggle Lite (16 tabular tasks)AIDE average across 16 tabular Kaggle tasksTable 1
Above Median (%) (Weco-Kaggle Lite)50.0%Weco-Kaggle LiteFraction of tasks where AIDE exceeded human medianTable 1

What To Try In 7 Days

Run AIDE on a small internal tabular task and compare leaderboard/validation to your current pipeline.

Replace a time-consuming debugging loop with AIDE's draft→debug→improve flow for a single project.

Measure per-task LLM cost and net time saved versus manual engineering for 3 representative problems.

Agent Features

Memory
external solution tree storing scripts and scoressummarization operator to compress history
Planning
hard-coded search policy for draft/debug/improveatomic-change proposals to measure effect
Tool Use
LLM APIs for code generationcode execution environment for evaluation and submissionKaggle leaderboard submission flow
Frameworks
Contrast to ReACT-style agentsintegration with AutoML baselines (H2O) for comparison
Is Agentic

Yes

Architectures
LLM-based agenttree-search over code space
Collaboration
single-agent workflow (no multi-agent coordination)

Optimization Features

Token Efficiency
summarization operator reduces prompt size and history included
System Optimization
reuse and refine promising code nodes to trade compute for performance

Reproducibility

Code AvailableYes
Data AvailableNo
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

Potential data contamination: LLMs may have seen competition data during training.

Performance varies by dataset; not uniformly superior across all tasks.

When Not To Use

When working on large multi-module codebases needing deep architecture-level redesigns.

When strict reproducibility/privacy forbids use of external LLM APIs or possible training-data leakage.

Failure Modes

Repeating local patches without discovering new strategies (local optima).

Generating invalid submissions due to environment mismatch or hidden dependencies.

Core Entities

Models

GPT-4 TurboGPT-4oo1-previewLlama 3.1Claude 3.5GPT-4

Metrics

Exceeds % of humansAbove Median (%)Valid Submission (%)Any Medal (%)Gold (%)pass@k

Datasets

Weco-Kaggle (full)Weco-Kaggle Lite (16 tasks)MLE-Bench (75 Kaggle competitions)RE-Bench (METR R&D tasks)various Kaggle competitions (e.g., bike-sharing-demand)

Benchmarks

Weco-KaggleMLE-BenchRE-Bench