AI-guided design of efficient clinical trials with BATON
Naim Rashid
Department of Biostatistics · Lineberger Comprehensive Cancer Center · UNC Chapel Hill
Computational Medicine Seminar · September 4, 2026
Section 1
The problem
Why isn’t simulation alone enough?
The question
Does the treatment beat a benchmark or a control?
How patients are used
Single-arm (everyone gets the treatment) or randomized (treatment versus control).
Two errors
Advance a treatment that fails, or drop one that works.
Two choices dominate: how many patients, and which rule decides.
Maximum N, look timing, futility and efficacy thresholds, priors, margins: 8 to 13 interacting choices, and every one changes how the trial behaves.
Type I error (false go)
Share of simulated trials declaring efficacy under the null. Cap: 0.10.
Power (true go)
Share declaring efficacy when the treatment works as hypothesized. Floor: 0.80.
| Typical trial | EVOLVE | |
|---|---|---|
| Designs needed | One | Many subtrials |
| Design parameters | Few | Many, interacting |
| Power and type I error | Often closed-form | Simulation only |
Not one calibration problem. Ten to fifteen of them, each with its own design, opened on a short clock.
2 weeks
manual tuning of one subtrial’s design
8 interacting settings
power stalled at 0.78 with 39.2 expected patients when the treatment is inactive
Several more subtrials were waiting. And even if you find a feasible design, is it efficient under the objective you actually care about?
Phase I dose-finding
escalation and safety rules tuned by simulation
Phase II Bayesian designs
posterior thresholds and interim looks tuned by simulation
Platform trials
every added feature adds settings that interact
Same shape every time: many interacting settings, operating characteristics evaluated by simulation, and calibration that still requires search.
In two dimensions a grid is easy. As dimensions grow, the number of combinations explodes. Random search spent 93% of its 2,000 evaluations on designs that fail. (Darker = better design.)
Section 2
BATON
What should we simulate next?
Machine learning
hyperparameters
↓
expensive training run
↓
validation performance
Trial design
design parameters
↓
expensive Monte Carlo simulation
↓
power, type I error, enrollment
The same computational shape as hyperparameter tuning, except feasibility is defined by hard statistical constraints.
BATON: Bayesian Adaptive Trial OptimizatioN
Constrained Bayesian optimization: surrogates learn power, type I error, and enrollment; they choose the next candidate.
You specify: a power floor, a type I error cap, what to minimize.
Your simulator: settings and scenarios in, operating characteristics out.
A closed loop: simulate, learn, choose, repeat.
Design settings in, trial behavior out, and the model says how sure it is.
Balance the potential improvement in the objective against the probability that the design satisfies the constraints.
Initial points learn broadly; later points concentrate where the answer may improve.
Cheap screening for the many, expensive re-checks for the few, multi-seed verification for the one design you report.
Section 3
The evidence
Feasibility, efficiency, and richer designs
Simon’s two-stage design: enroll n1, stop if too few responses, otherwise enroll to n. Every candidate in this design class can be enumerated, so the exact constrained optimum is known.
10 random starts: 8 return Simon’s exact optimal design; 2 land 0.15 patients away.
One run per scenario; ratio = BATON’s expected null enrollment / Simon’s exact optimum.
| Claim | Value |
|---|---|
| Feasible design returned | 12 / 12 |
| Exact Simon design returned | 4 / 12 |
| Median ratio, BATON / Simon | 1.025 |
| Worst-case ratio | 1.135 |
| Best-case ratio | 1.000 |
Feasibility is verified directly. Objective optimization remains approximate.
MANUAL
2 weeks
power 0.78
39.2 expected patients if inactive
infeasible
BATON
~4-hour workflow
power 0.86
11.7 expected patients if inactive
feasible
Illustrative comparison, not a controlled benchmark across analysts. Same simulator, same constraints.
Many designs meet power ≥ 0.80 and type I error ≤ 0.10. Which should we optimize for?
EVOLVE’s TNBC cohort, different objectives:
| Design A | Design B | |
|---|---|---|
| Maximum patients | 47 | 30 |
| Expected patients, inactive treatment | 11.7 | 22.8 |
| Power | 0.86 | 0.94 |
| Type I error | 0.001 | 0.021 |
One quits losers quickly; one caps the worst case. Neither is better.
Minimize expected N under the null; larger maximum N.
Minimize maximum N; more patients under the null.
A weighted blend, explicit.
Two-stage designs made these tradeoffs explicit for decades; BATON does so at scale.
The platform’s TNBC cohort
For one additional patient of maximum capacity, Admissible saves about eleven patients on average when the treatment is inactive. That made the compromise especially attractive.
Screen first with a single-arm stage, then demand randomized evidence from the cohorts that earn it. Thirteen interacting parameters, calibrated by automated search.
Minimax and Admissible calibrated to the identical design. That is not a coincidence, and the next slide shows why.
When the treatment is inactive, only ~4% of trials reach stage 2. Expected enrollment therefore changes very little with stage-2 size, so the maximum-enrollment component drives the Admissible solution toward Minimax. H0-Optimal can tolerate a larger stage 2 because its objective never penalizes maximum N.
Capacity: maximum N locked 3 months before first enrollment, from the frontier.
Program planning: enrollment and timeline projections for 10 to 15 subtrials.
Interim calendar: look spacing set the safety board’s schedule.
Turnaround: a feasible design in under 45 minutes; workflow about 4 hours.
We generally chose Admissible designs: efficient enrollment, early stopping, bounded capacity.
Manuscript under review at JASA
Open-source R package
Bring your own trial simulator
54%
of 640 therapeutics reaching phase 3 or pivotal trials failed in late-stage development; 57% of failures for inadequate efficacy
Locally reasonable decisions can leave problems that only become visible downstream.
Hwang et al., JAMA Internal Medicine 2016
Sotorasib (KRAS G12C)
960 mg: highest phase I dose tested
↓ carried forward
selected, developed, approved
↓ after approval
FDA required a 240 vs 960 mg comparison
↓
median PFS was similar at 240 mg and 960 mg
The optimal-dose question remained unresolved until after approval.
Source: CodeBreaK 100 dose comparison; Ann Oncol 2023, Eur J Cancer 2024
What if we reverse the logic?
final therapeutic claim + its evidentiary requirements
↓
simulate alternative development architectures
↓
phased · seamless · potentially phaseless
Phase structure becomes a design variable.
Optimize over a prespecified family of plausible truths, not one.
Optimize dose-finding rules for safety and for what later trials inherit.
Optimize dose, population, interim, and confirmatory decisions as one program.
For complex adaptive trials, simulation tells us how a candidate behaves; design is the search for what to try next.
BATON searches the simulator for feasible, efficient designs and makes the patient-use tradeoffs explicit.
Tractable calibration lets us consider richer adaptive trials; the same search logic extends to other decisions across development.
FUNDING · ARPA-H 140D042590009 · NCI P50-CA257911 (UNC SToP SPORE) · DOD HT9425241103100121 · NCI U01-CA274298
Thank you
Questions
naim@unc.edu · rashidlab.github.io
BATON R package: github.com/naimurashid/BATON · manuscript under review at JASA
Backup
Reference slides
Always feasible: 12 of 12. Exact optimum in 4 of 12; median within 2.5%; worst case 13.5%.
Primary target: verified feasibility. Objective optimization is approximate in an expensive, noisy, discrete space.
Calibrated (searched) parameters, single-arm:
Seamless adds conversion-gate thresholds and second-stage sample-size and stopping parameters (13 total).
Constraints (headline scenarios):
Objectives, by philosophy: expected null enrollment (H0-Optimal), maximum enrollment (Minimax), weighted blends (Admissible family), expected alternative enrollment (H1-Optimal).
Scenario A truths: null median 3.8 months, alternative 9 (hazard ratio 0.42), Weibull shape 1.3, accrual 2 per month; maximum enrollment searched in [30, 120]. Two Simon scenarios use a 0.05 ceiling with 0.90 power.
Existing approaches
What BATON adds, jointly in one framework
It exposes the trade-off surface, and reveals when philosophies coincide and why.
The single-arm and between-arm headline scenarios use futility-only interim stopping (interims can stop for futility; efficacy is decided at the end). The seamless design is calibrated with bidirectional stopping throughout, because its conversion gate is efficacy-driven. The paper’s supplement repeats the single-arm and between-arm calibrations with bidirectional stopping, where interims may also stop early for efficacy.
The whole pipeline, rerun end to end at ten independent random seeds:
| Runs returning a feasible design | 10 / 10 |
| Runs returning Simon’s exact optimal design | 8 / 10 |
| Expected null enrollment, mean (spread) | 17.77 (0.06) |
| Evaluations used, median | 250 of a 300 budget |
The other two seeds return a near-neighbor design (capacity 29 instead of 25) whose expected enrollment differs by 0.15 patients. What reproduces is the design’s behavior, and usually the design itself.
For a method meant to sit inside regulatory submissions, this is not a nicety. That is the bar for a design going into a regulatory submission.
Cutting the worst-case cap from 124 to 104 costs about 8 extra patients on average when the treatment does not help; the next 5 cost about 9 more.
excerpts from the ARPA-H solicitation and program-manager correspondence
UNC Lineberger · Computational Medicine Seminar 2026