ERA: Scaling Scientific Discovery via LLM-Driven Tree Search

An AI system to help scientists write expert-level empirical software

2026-01-01
Eser Aygün, Anastasiya Belyaeva, Gheorghe Comanici, Marc Coram, Hao Cui, Julie Jake Garrison, Renee Johnston, Anton Kast, Cory Y. McLean, Peter C. Norgaard, Zahra Shamsi, David Smalling, Julie James Thompson, Subhashini Venugopalan, Brian P. Williams, Chujun He, Sarah Martinson, Martyna Plomecka, Lai Wei, Yuchen Zhou, Qian-Ze Zhu, Matthew Abraham, Erica Brand, Anna Bulanova, Julie Jeffrey A. Cardille, Chris Co, Scott Ellsworth, Grace Joseph, Malcolm Kane, Ryan Krueger, Julie Johan Kartiwa, Daniel J. Liebling, Julie Jan-Matthis Lueckmann, Paul Raccuglia, Xuefei Wang, Xuefei Wang, Katherine Chou, James Manyika, Yossi Matias, John C. Platt, Lizzie Dorfman, Shibl Mourad, Michael P. Brenner
Summary
Problem
Method
Results
Takeaways
Abstract

The paper introduces Empirical Research Assistance (ERA), an AI system that combines Large Language Models (LLMs) with Tree Search (TS) to autonomously develop expert-level scientific software. ERA achieved SOTA results in diverse fields, including discovering 40 novel single-cell RNA-seq analysis methods and outperforming the CDC ensemble in COVID-19 hospitalization forecasting.

TL;DR

Researchers from Google DeepMind and Harvard have unveiled Empirical Research Assistance (ERA), a system that automates the creation of high-performance scientific software. By framing software development as a "scorable task" and navigating it via Tree Search (TS), ERA doesn't just write code—it "evolves" it, discovering methodologies that outperform human experts in genomics, epidemiology, and time-series forecasting.

The Bottleneck: Human-in-the-Loop Discovery

The bottleneck of modern science isn't just data—it's the software required to process it. Historically, creating empirical software (models designed to maximize a specific metric) has been a tedious process of manual trial and error. Design choices are often governed by convenience rather than exhaustive search.

Current AI code assistants (like Github Copilot) are largely "one-shot" generators. While they can help write a function, they lack the long-range persistence to explore thousands of variations or "backtrack" when an approach plateaus. This is where ERA changes the game.

Methodology: The "Semantic Mutation" Loop

ERA treats the space of possible programs as a manifold to be searched. Unlike Genetic Programming (GP)—which relies on random bit-flips or subtree swaps—ERA uses an LLM (Gemini 2.5 Flash/Pro) as an intelligent, semantic-aware mutation operator.

1. The PUCT Tree Search

ERA uses a modified version of the PUCT (Predictor + Upper Confidence Bound applied to Trees) algorithm.

  • Exploitation: The system expands nodes (code versions) that have previously yielded high scores.
  • Exploration: It branches out to less-visited nodes to avoid local optima.

2. Infusing Domain Expertise

The system's most potent feature is its ability to ingest "Research Ideas." The authors provided ERA with summaries of highly-cited papers (e.g., scRNA-seq integration techniques). The LLM processes these ideas to guide its code generation, effectively standing on the shoulders of giants.

ERA Algorithm Architecture Figure 1: Schematic showing how ERA combines scorable tasks, research ideas, and Tree Search to optimize code in a sandbox environment.

Breakthrough Results across Disciplines

Genomics: Single-Cell RNA Sequencing

The challenge: Removing "batch effects" (noise from different labs) while preserving biological signals. ERA discovered a hybrid method combining ComBat and BBKNN, which achieved a 14% improvement over the previous SOTA. It didn't just replicate the paper; it found that applying ComBat-corrected PCA embeddings before the BBKNN graph construction was the secret sauce humans had missed.

Public Health: CDC COVID-19 Forecasting

In a retrospective study of U.S. hospitalizations, ERA-generated models outperformed the CDC's robust Ensemble model (the previous gold standard). The search discovered that hybridizing "climatology" (historical averages) with "epidemiological" autoregressive models provided the best-calibrated uncertainty.

COVID-19 Forecast Comparison Figure 2: Performance comparison showing ERA's superior accuracy (WIS) across the majority of U.S. states compared to the CDC ensemble.

Why It Works: The "Needle-in-the-Haystack" Effect

The authors highlight a critical phenomenon: Breakthrough Plots. As the tree search progresses, the score often remains flat until a specific "semantic mutation" (like a new feature transform or a change in optimizer) triggers an abrupt jump in performance. Humans lack the patience to perform the thousands of trial runs required to hit these "needle-in-the-haystack" solutions; ERA does not.

Breakthrough Plot Figure 3: A breakthrough plot for scRNA-seq batch integration showing how specific code modifications lead to stepwise increases in metric scores.

Critical Insight: Beyond Empirical Optimization

ERA represents a massive leap in Empirical Software Engineering, but the authors are careful to distinguish this from "Genuine Discovery." While ERA excels at maximizing a score, genuine science requires reasoning about causal mechanisms.

However, the paper hints at the future: by providing ERA with structural priors (like brain wiring diagrams in the ZAPBench task), the system began to recover causal effective connectivity. This suggests that if we can "score" a hypothesis, ERA can discover the underlying theory.

Conclusion & Future Outlook

ERA proves that LLMs, when unshackled from "one-shot" constraints and embedded in a rigorous search framework, can outperform human engineers. For any scientific field where a metric can be defined—from climate modeling to drug discovery—the "trial and error" phase of research is about to accelerate by orders of magnitude.

Takeaway: We are entering the era of "Agentic Discovery," where the scientist's role shifts from writing the code to defining the Metric and the Search Space.

Find Similar Papers

Try Our Examples

  • Find other recent papers that utilize MCTS or Tree Search with LLMs specifically for automated algorithm design or scientific code optimization.
  • Which paper first established the "PUCT" algorithm variant of Upper Confidence Bound for Trees, and how does ERA's "Flat UCB" modification differ from AlphaZero's implementation?
  • Identify studies exploring the application of LLM-driven autonomous agents in other "scorable" scientific tasks such as protein folding or chemical synthesis planning.
Contents
ERA: Scaling Scientific Discovery via LLM-Driven Tree Search
1. TL;DR
2. The Bottleneck: Human-in-the-Loop Discovery
3. Methodology: The "Semantic Mutation" Loop
3.1. 1. The PUCT Tree Search
3.2. 2. Infusing Domain Expertise
4. Breakthrough Results across Disciplines
4.1. Genomics: Single-Cell RNA Sequencing
4.2. Public Health: CDC COVID-19 Forecasting
5. Why It Works: The "Needle-in-the-Haystack" Effect
6. Critical Insight: Beyond Empirical Optimization
7. Conclusion & Future Outlook