Data-driven Design of Randomized Control Trials with Guaranteed Treatment Effects
Santiago Cortes-Gomez, Naveen Janaki Raman, Aarti Singh, Bryan Wilder
Abstract
Randomized controlled trials (RCTs) can be used to generate guarantees on treatment effects. However, RCTs often spend unnecessary resources exploring sub-optimal treatments, which can reduce the power of treatment guarantees. To address these concerns, we develop a two-stage RCT where, first on a data-driven screening stage, we prune lowimpact treatments, while in the second stage, we develop high probability lower bounds on the treatment effect. Unlike existing adaptive RCT frameworks, our method is simple enough to be implemented in scenarios with limited adaptivity. We derive optimal designs for two-stage RCTs and demonstrate how we can implement such designs through sample splitting. Empirically, we demonstrate that two-stage designs improve upon single-stage approaches, especially in scenarios where domain knowledge is available in the form of a prior. Our work is thus, a simple, yet effective, method to estimate high probablility certificates for high performant treatment effects on a RCT.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ca290346-2225-4516-b712-2b754dd37f22Cited by top-tier papers1
Ask how each one uses itRelated papers
- Task-specific experimental design for treatment effect estimationBethany Connolly, Kim Moore, Tobias Schwedes, Alexander Adam et al.ICML 2023 · 4 citations
- Treatment Effect Estimation for Optimal Decision-MakingDennis Frauen, Valentyn Melnychuk, Jonas Schweisthal, Mihaela van der Schaar et al.NeurIPS 2025 · 8 citations
- Estimation of Bounds on Potential Outcomes For Decision MakingMaggie Makar, Fredrik D. Johansson, John V. Guttag, David A. SontagICML 2020 · 11 citations
- A Two-Stage Pretraining-Finetuning Framework for Treatment Effect Estimation with Unmeasured ConfoundingChuan Zhou, Yaxuan Li, Chunyuan Zheng, Haiteng Zhang et al.KDD 2025 · 6 citations
- Learning Treatment Allocations with Risk Control Under Partial IdentifiabilitySofia Ek, Dave ZachariahICML 2026
