adaptive covariate balancing

This paper addresses unbiased estimation of the average treatment effect (ATE) under sequential adaptive treatment allocation. We consider settings where assignment probabilities depend on past treatments and outcomes, violating the complete randomization assumption. To tackle this, we propose inverse probability weighting (IPW) and augmented IPW (AIPW) estimators, and introduce the novel concept of “design stability.” For the first time in this framework, we establish their asymptotic normality and derive a central limit theorem, along with a consistently estimable asymptotic variance expression. Our methodology unifies classical sequential designs—including Wei’s adaptive coin-tossing and Efron’s biased coin design—thereby transcending conventional randomization constraints. The resulting confidence intervals are asymptotically valid, substantially enhancing the reliability and applicability of ATE inference in sequential experiments.

This paper addresses the problem of efficiently estimating the average treatment effect (ATE) in causal inference. We propose an adaptive experimental design framework with three key contributions: (1) We formally define and dynamically learn the optimal treatment assignment probability that minimizes the semiparametric efficiency bound of the ATE estimator; (2) We introduce the A²IPW estimator, which achieves the theoretically optimal asymptotic variance in finite samples; and (3) We construct nonparametric confidence intervals that are valid at any stopping time, enabling rate-optimal sequential testing and early stopping. The method integrates adaptive randomization, semiparametric efficiency theory, and anytime-valid inference. It substantially reduces the sample size required to achieve a target statistical power while guaranteeing strict coverage probability—even in small-sample settings.

This paper addresses identification and inference for the average treatment effect (ATE) and average treatment effect on the treated (ATT) under covariate-adaptive randomization (CAR) with noncompliance. First, it precisely characterizes the sharp identification sets for ATE and ATT within the CAR framework—where sampling yields non-i.i.d. data—using extremal identification theory. Second, it proposes boundary estimators and confidence interval constructions that are both consistent and asymptotically efficient. Third, it develops a hybrid weighting strategy that jointly leverages empirical sampling frequencies and known randomization probabilities; it establishes that ATE estimation achieves asymptotic efficiency using only empirical frequencies, whereas ATT estimation requires the numerator to employ true compliance probabilities and the denominator empirical frequencies. These results provide both theoretical foundations and practical guidelines for robust causal inference in CAR-based randomized controlled trials.

To address the challenge of balancing treatment effect heterogeneity learning and decision timeliness in adaptive clinical trials, this paper proposes a covariate-adjusted response-adaptive design that accelerates randomization probability updates using surrogate outcomes. Methodologically, it introduces the first formal framework quantifying the dual benefits—accelerated learning and bias reduction—of surrogate outcomes in sequential adaptive trials; develops an Online-Superlearner–driven mechanism for dynamic surrogate selection; and establishes a model-agnostic, targeted minimum loss–based estimation (TMLE) inference method tailored to adaptive trial data. Extensive simulations—including scenarios calibrated to real trial data—demonstrate that the design significantly improves the expected outcome for newly enrolled participants while preserving asymptotic normality of estimators. Collectively, it provides a generalizable toolkit for evaluation, selection, and inference in adaptive clinical trials.

This paper investigates the impact of covariate balancing on average treatment effect (ATE) and average treatment effect on the treated (ATT) estimation, and establishes the theoretical foundation for numerical equivalence among inverse probability weighting (IPW), augmented IPW (AIPW), and IPW-regression adjustment (IPWRA) estimators. We rigorously prove that when propensity scores are estimated via covariate balancing methods—such as inverse probability tilting (IPT) for ATE or covariate-balancing propensity score (CBPS) for ATT—the three estimators are algebraically identical, and their weights are automatically normalized. This equivalence unifies the theoretical frameworks of weighted and doubly robust estimation, substantially improving finite-sample stability and precision. Moreover, the result enables a novel analytical pathway for identifying the local average treatment effect (LATE) under unmeasured confounding, thereby advancing model robustness and computational consistency in causal inference.

This study addresses the challenge in randomized experiments where conventional sample allocation methods fail to simultaneously account for deployment relevance and statistical precision for a target population. The authors propose TWNA, a two-stage stratified design: an initial pilot stage estimates stratum-specific treatment effect variances, which then inform a joint optimization of final-stage sample sizes and treatment probabilities to enhance estimation precision of the target-weighted group average treatment effect (GATE). TWNA is the first method to unify deployment weights and statistical difficulty within a single optimization framework, yielding a closed-form optimal allocation rule. It further extends robustly to complex settings involving uncertainty in target population composition, skewed outcomes, or rare events. Simulations and empirical analyses demonstrate that TWNA substantially improves estimation accuracy and resource efficiency, particularly for critical yet hard-to-estimate subgroups.

This study addresses a central challenge in clinical trials: achieving both covariate balance and response-adaptive allocation without relying on correct model specification. The authors propose CBARA, a novel method that extends the principles of covariate-adaptive randomization (CAR) to dynamically adjust target allocation proportions based on covariate information, thereby integrating the strengths of CAR and covariate-adjusted response-adaptive (CARA) designs. By introducing an imbalance vector and a three-component mechanism, CBARA simultaneously balances both observed and unobserved covariates. Through a pseudo-Markov chain framework, a new metric for transition kernel discrepancy, and continuity analysis of Poisson equation solutions, the authors theoretically establish that CBARA consistently attains the desired allocation targets and substantially enhances covariate balance—all without requiring correct model assumptions—thus offering improved ethical, statistical, and operational robustness.

This study addresses the lack of theoretical justification for covariate-adaptive randomization in high-dimensional settings where the number of covariates grows with the sample size. The authors develop a unified theoretical framework to analyze the imbalance properties of two classes of adaptive randomization procedures, both for specified and unspecified covariates. Leveraging high-dimensional probabilistic limit theory and imbalance analysis, they establish—for the first time—the convergence rates of covariate imbalance under this asymptotic regime and prove the asymptotic normality of the average treatment effect estimator. Furthermore, they derive valid confidence intervals based on these results. Numerical experiments corroborate the practical relevance of the theoretical findings, thereby providing a rigorous statistical foundation for high-dimensional causal inference.

This study addresses the challenge of constructing valid confidence intervals in two-stage adaptive enrichment clinical trials, where patient subgroups are selected based on interim data, thereby compromising the nominal coverage of conventional intervals. The authors propose a novel method that constructs confidence intervals conditional on the interim selection decision, leveraging conditional inference and inversion of uniformly most accurate unbiased (UMAU) tests to guarantee exact coverage within the selected subgroup. The approach is broadly applicable to various adaptive enrichment designs and is implemented via an efficient numerical algorithm. Extensive simulation studies demonstrate that the proposed intervals consistently achieve the desired coverage probability across diverse design configurations, substantially outperforming existing methods in both validity and precision.