mlsanityData to 7 Oct 2026

Adaptive Experimentation for Censored Survival Outcomes

arXiv 2605.18459 ↗First posted 18 May 2026cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Adaptive experimentation enables efficient estimation of causal effects, but existing methods are not designed for survival data with censoring, where event times are only partially observed (e.g., overall survival in cancer trials but with dropout). In this paper, we develop a novel framework for adaptive experimentation to estimate causal effects under right censoring. For this, we derive the semiparametric efficiency bound for the average survival effect curve as a function of the treatment allocation policy and thereby obtain a closed-form efficiency-optimal allocation policy. The policy generalizes classical Neyman allocation to survival settings by prioritizing patient strata where both event and censoring dynamics induce high uncertainty. Building on this, we propose the Adaptive Survival Estimator (ASE), an adaptive framework that learns the allocation policy and estimates the average survival effect curve sequentially. Our framework has three main benefits: (i) it accommodates arbitrary machine learning models for nuisance estimation; (ii) it is guided by a closed-form efficiency-optimal allocation policy; and (iii) it admits strong theoretical guarantees, including asymptotic normality via a martingale central limit theorem. We demonstrate our framework across various numerical experiments to show consistent efficiency gains over uniform randomization and censoring-agnostic baselines.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 67 references that are in the map. First posted on arXiv 18 May 2026.

Builds on

32 of 67 references.

References · 32

  1. 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
  2. 2016DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Networkcited by 354
  3. 2022Game-theoretic statistics and safe anytime-valid inferencecited by 145
  4. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  5. 2023Deep Learning for Survival Analysis: A Reviewcited by 75
  6. 2021Conformalized Survival Analysiscited by 53
  7. 2021Statistical Inference with M-Estimators on Adaptively Collected Datacited by 40
  8. 2020Estimating heterogeneous treatment effects with right-censored data via causal survival forestscited by 29
  9. 2021Time-uniform central limit theory and asymptotic confidence sequencescited by 29
  10. 2023Semiparametric Efficient Inference in Adaptive Experimentscited by 23
  11. 2022Conformalized survival analysis with adaptive cutoffscited by 17
  12. 2023Clip-OGD: An Experimental Design for Adaptive Neyman Allocation in Sequential Experimentscited by 16
20 more
  1. 2017Individualized Treatment Effects with Censored Data via Fully Nonparametric Bayesian Accelerated Failure Time Modelscited by 13
  2. 2020Estimating heterogeneous survival treatment effect in observational data using machine learningcited by 12
  3. 2024Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate Choicescited by 12
  4. 2020Response-adaptive randomization in clinical trials: from myths to practical considerationscited by 10
  5. 2025Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event Datacited by 10
  6. 2021Inference for treatment-specific survival curves using machine learningcited by 8
  7. 2021Estimating Heterogeneous Treatment Effects for General Responsescited by 7
  8. 2022BITES: Balanced Individual Treatment Effect for Survival datacited by 7
  9. 2024Anytime-Valid Inference for Double/Debiased Machine Learning of Causal Parameterscited by 5
  10. 2024Optimal Adaptive Experimental Design for Estimating Treatment Effectcited by 5
  11. 2024Estimating Heterogeneous Treatment Effects on Survival Outcomes Using Counterfactual Censoring Unbiased Transformationscited by 4
  12. 2017Learning the distribution with largest mean: two bandit frameworkscited by 3
  13. 2024Off-policy estimation with adaptively collected data: the power of online learningcited by 3
  14. 2024Logarithmic Neyman Regret for Adaptive Estimation of the Average Treatment Effectcited by 3
  15. 2025Optimistic Algorithms for Adaptive Estimation of the Average Treatment Effectcited by 2
  16. 2025Stronger Neyman Regret Guarantees for Adaptive Experimental Designcited by 2
  17. 2025Efficient Adaptive Experimentation with Noncompliancecited by 2
  18. 2025Efficient Statistical Estimation for Sequential Adaptive Experiments with Implications for Adaptive Designscited by 2
  19. 2025Sigmoid-FTRL: Design-Based Adaptive Neyman Allocation for AIPW Estimatorscited by 1
  20. 2026Design-Based Anytime-Valid Inference for Randomized Experiments with Delayed Outcomes and Staggered Entrycited by 1

Built on by

Not cited yet.