# Structured Scaling of AI Discovery Across Diverse Scientific Domains

> SimpleTES uses one open-source model to achieve state-of-the-art results across 28 scientific problems by structuring evaluation-driven search through parallel trajectories, iterative refinement, and selective history reuse.

- **Source:** [arXiv](https://arxiv.org/abs/2604.19341)
- **Published:** 2026-08-17
- **Permalink:** https://picx.dev/p/twKveT
- **Whiteboard:** https://picx.dev/p/twKveT/image

## Summary

# Structured Scaling of AI Discovery Across Diverse Scientific Domains

## Summary (Overview)

- **SimpleTES (Simple Test-time Evaluation-driven Scaling)** is a framework that structures AI-driven scientific discovery loops through a compact design tuple $(C, L, K, \Phi)$, where $C$ is the global width (parallel trajectories), $L$ is the refinement depth, $K$ is the local sample size, and $\Phi$ is the proposal constructor that selectively reuses evaluated histories.
- Using a **single open-source GPT-OSS model**, SimpleTES achieves **state-of-the-art results across 28 open-ended problems** in quantum physics, astronomy, biology, AI, and mathematics—surpassing systems that use larger proprietary models.
- Key breakthroughs include a **24.5% reduction in quantum circuit compilation overhead**, up to **23% lower propulsive cost for deep-space trajectories**, a **2.17× faster lasso-path solver**, an **8.5% lower-error whole-brain neural-activity predictor**, the **fastest reported TriMul kernel**, and **new mathematical constructions** beyond prior human or AI records.
- The paper introduces **trajectory-level post-training**, which credits each attempt by the final outcome of its trajectory rather than its immediate score, improving performance on both training and held-out mathematics problems.
- SimpleTES reframes discovery-loop scaling as a problem of **managing evaluated histories** so that additional evaluations compound into stronger trajectories rather than merely producing more attempts.

## Introduction and Theoretical Foundation

### Background and Motivation

Scientific discovery fundamentally requires **many cycles of proposing, testing, and refining** candidate solutions. While language models can increasingly participate in these loops, the central challenge is not merely scaling AI-driven discovery but **structuring that scaling so that evaluation signals compound over time**. Parallel searches may duplicate one another, and iterative refinement may become trapped in poor directions.

The paper draws an analogy to human research communities, which advance science collectively through:
- **Local refinement**: individual researchers test a few variants and carry forward the best
- **Selective learning**: researchers selectively learn from prior results based on relevance and performance
- **Collective building**: researchers build on one another's ideas
- **Parallel competition**: multiple lines of inquiry develop and compete

### Theoretical Foundation: Evaluation-Driven Scaling

The authors distinguish evaluation-driven scaling from:
- **Standard model scaling**: scaling parameters and training data
- **Classical test-time scaling**: scaling through longer reasoning traces or additional agent turns

Evaluation-driven scaling targets **open-ended problems that fundamentally require exploration**, where a larger number of loops does not automatically imply better results.

### Problem Formulation

Formally, a problem is specified by an instruction $x$, and the goal is to find a solution $y$ that maximizes a score $r$ output by an evaluator $V$. An LLM $G$ plays the role of a researcher, drawing candidates $y \sim G(\Phi)$ from a proposal built from the instruction plus accumulated history of previously evaluated solutions.

## Methodology

### The SimpleTES Design Tuple

SimpleTES organizes the discovery loop through the compact design tuple:

$$(C, L, K, \Phi) \tag{1}$$

where:
- **$C$** = number of parallel trajectories (global width) — counters the tendency of single deep trajectories to become trapped in narrow basins
- **$L$** = refinement steps per trajectory — lets feedback accumulate so later attempts build on earlier ones
- **$K$** = candidates drawn per step (local sample size) — addresses noise in LLM output by committing only the highest-scoring candidate
- **$\Phi$** = proposal constructor — selects which parts of evaluated history to reuse via a graph-based PUCT rule, favoring attempts that scored well or led to strong later attempts

The total evaluator-query budget is $N = C \times L \times K$.

### Proposal Constructor Φ

Because each trajectory's history becomes too long and noisy to feed back in full, Φ:
- Selects attempts through a **graph-based PUCT rule** (favoring high-scoring attempts while allowing promising rarely-used ones to be revisited)
- Folds in a **running summary of recurring failures** (compilation errors, verifier failures, timeouts) to steer the model away from unproductive directions

### Post-Training for Long-Horizon Discovery

The model is post-trained on its own successful trajectories, crediting each attempt by the trajectory's final outcome rather than its immediate score:

$$\mathcal{L} = -\mathbb{E}_{(x, \hat{y}, w) \sim \mathcal{D}}\left[w \cdot \sum_{i=1}^{|\hat{y}|} \log \pi_\theta(\hat{y}_i \mid x, \hat{y}_{<i})\right] \tag{2}$$

where $x$ is the proposal context, $\hat{y}$ is the model's reasoning and solution, and $w=1$ for trajectories in the top $R\%$ by outcome, $w=0$ otherwise. This is repeated over multiple rounds.

### Default Configuration

Unless otherwise stated, SimpleTES uses $C = 32$, $L = 100$, and $K = 16$ as its default configuration.

## Empirical Validation / Results

### Quantum Circuit Compilation

SimpleTES achieves substantial overhead reductions across two hardware settings:

**Superconducting qubit routing:**
- **21.7% lower average added SWAP overhead** versus SABRE and **14.9% versus LightSABRE**
- Largest reductions on Q20 reaching **33.3% and 24.5%** respectively
- Best policy maps high-traffic logical qubits toward high-degree physical qubits with local search refinement

**Zoned neutral-atom compilation:**
- **33.2% reduction** in geometric mean of total execution time versus ZAC-style baseline
- Achieved by reserving relative qubit order, reducing travel distance without losing AOD parallelism

### Astrodynamics Trajectory Design

Across five historical mission-window benchmarks, discovered optimizers reduce propulsive cost by **2.1–23.1%** relative to same-evaluator references:
- **Mariner 10**: 0.326993 (vs. 0.424147 historical)
- **Cassini**: 0.820129 (vs. 1.066682 historical)
- **Galileo**: finds Earth–Venus–Venus–Venus–Earth–Jupiter sequence (vs. reference Earth–Venus–Earth–Earth–Jupiter)

### Scientific Algorithms

**Lasso regularization path solver:**
- **2.17× faster than glmnet** on average, **14.08× faster than sklearn**
- Discovers a **geometry-aware switching policy**: uses exact LARS homotopy for tall designs ($p \leq 2000$, $n \geq p/4$), coordinate descent with strong-rule screening for wide/sparse designs
- Gains peak at **9.56× over glmnet** on tall DNA design, with all solutions passing strict correctness checks (objective gap < $10^{-6}$)

**ZAPBench whole-brain neural-activity forecasting:**
- Horizon-averaged test MAE of **0.0259 at 32 steps** (8.5% relative improvement over ERA's 0.0283)
- Discovers a residual-prediction architecture with identity skip connections, linear-trend baseline, and per-horizon closed-form least-squares weighting

### AI Foundations

**GPU kernel optimization (on NVIDIA H200, transferring to H100):**
- **TriMul**: 1.122 ms geometric-mean latency (vs. 1.140 ms human expert, 1.154 ms AI baseline); fastest on A100, H100, and AMD MI300
- **Asymmetric matmul**: up to **4.01× speedup** over PyTorch FP32 reference, 65.6% runtime reduction vs. CUDA Agent
- **Batched cumsum**: outperforms NVIDIA CUB on 4 of 6 shapes, up to **6.06× speedup** over PyTorch

**Scaling-law discovery (SLDBench):**
- Best average $R^2$ of **0.674**, exceeding SLDAgent+GPT-5 (0.572)
- Highest $R^2$ on U-shape task (−0.008 vs. −0.232 next-best) and lr&bsz task (0.712 vs. 0.611)
- Discovered law recommends hyperparameters only **0.058% above the true optimum** at 1B-parameter/100B-token target

### Mathematical Discovery

Without model training, SimpleTES establishes new state-of-the-art results:
- **Erdős minimum-overlap problem**: 0.380868 (vs. 0.380871 Together AI)
- **Second autocorrelation inequality**: 0.962694 (vs. 0.961206)
- **Third autocorrelation inequality**: 1.453675 (vs. 1.454555)
- **Sum–Difference Problem**: 1.143975 (vs. 1.121936 AlphaEvolve V2)

### Post-Training Results

- Post-training moves the **upper tail of the score distribution upward** at every level (top-10/25/50/75%)
- Gains **accumulate over iterations** rather than saturating
- Transfers to **held-out problems**, indicating a generalizable discovery skill
- Achieves a **new SOTA of 1.144887** on the held-out Sum–Difference Problem

## Theoretical and Practical Implications

### Theoretical Implications

1. **Structured scaling is independently powerful**: By keeping the proposer minimal (single LLM call, no multi-agent coordination, no model ensembles), SimpleTES demonstrates that the **organization of the evaluation loop is itself an important source of capability**.

2. **Compact design space**: The three dimensions ($C$, $L$, $K$) target distinct, independent failure modes—making them a small and non-redundant set of controls on search.

3. **Evaluation-driven scaling as a general mechanism**: One unchanged procedure produces discoveries across diverse domains, from abstract mathematics to executable scientific programs.

### Practical Implications

1. **Efficiency gains**: Trajectories proceed independently, and $K$ candidates conditioned on the same proposal can be produced in a single batched LLM request. Under matched budgets, SimpleTES reaches OpenEvolve's best score with **11.8× time-to-solution speedup** on one problem and **18.0× on another**.

2. **Early pruning**: Retaining only one trajectory after 25% of refinement steps preserves the best solution in 55% of problems, with relative score degradation below 0.01% in most configurations.

3. **Evaluator fidelity is critical**: SimpleTES optimizes the evaluator's score directly and can exploit implementation loopholes (numerical tolerances, timing manipulation, baseline corruption). This requires independent re-verification and hardened evaluators.

### Limitations

- Requires **automatic and inexpensive feedback**; not suited to subjective, experimentally costly, or slow physical/wet-lab measurements
- Fixed allocation across $C$, $L$, $K$; optimal allocation depends on task and search state
- Relies on **continuous, fine-grained scores**; less suited to sparse or binary feedback (e.g., formal theorem proving)

## Conclusion

SimpleTES reframes open-ended scientific discovery as a problem of **structured scaling**: how evaluator queries should be distributed across competing lines of inquiry, and how the resulting experience should guide subsequent attempts. The framework demonstrates that evaluator queries become substantially more powerful when structured so that exploration and refinement can compound.

Key takeaways:
1. **Structured scaling works across domains**: 28 SOTA results using a single open-source model with only the evaluator changing between problems
2. **Trajectory-level post-training enhances discovery**: Crediting attempts by trajectory outcomes rather than immediate scores instills a less myopic discovery skill that generalizes to held-out problems
3. **Future directions** include adaptive allocation policies, richer semantic feedback from failed attempts, and interfaces to higher-fidelity simulators and laboratory robotics

The complete SimpleTES codebase is publicly available at https://github.com/wq-will/SimpleTES.

---

_Markdown view of https://picx.dev/p/twKveT, served by PicX — AI-generated visual whiteboard summaries of research papers._
