Tutorials › Multi-Arm Multi-Stage (MAMS) Designs › Covariate-Adjusted Estimates After Arm Selection

Treatment-Effect Estimation After Selection · Tutorial 282 of 499

Covariate-Adjusted Estimates After Arm Selection

Learn when covariate adjustment improves precision and how to quantify its effect on bias when an adjusted interim contrast determines which treatment arm continues.

Advanced 12 min read

What You'll Learn

  • State the linear-model assumptions for covariate-adjusted treatment-control contrasts.
  • Derive the variance of an adjusted contrast and compare it with the unadjusted difference in means.
  • Calculate selection bias for a normally distributed adjusted estimate crossing a threshold.
  • Relate post-selection bias to the covariance between an interim selection statistic and a final estimate.
  • Identify model and design diagnostics that can undermine precision gains or bias calculations.

Why Adjustment Matters After Arm Selection

In a multi-arm multi-stage trial, interim outcomes may be used to decide which treatment arms continue. If the decision is based on covariate-adjusted treatment-control contrasts, the final estimate for a retained arm is affected by two features at once: adjustment changes the contrast’s precision, and the interim selection rule makes the retained arm’s estimate non-representative of an unselected estimate. These effects are connected, but they are not interchangeable. Lower variance does not by itself eliminate selection bias.

The key quantity for post-selection bias is the dependence between the interim statistic used to select an arm and the estimate eventually reported. Covariate adjustment changes that dependence as well as the standard errors. A useful analysis therefore states the adjustment model, identifies which data contribute to selection and final estimation, and then evaluates the selection rule under those assumptions. The introductory material on selection correction establishes why selection can bias estimates; here the focus is how covariate adjustment alters that calculation.

Definition: For a continuous outcome, a covariate-adjusted treatment-control contrast is the treatment coefficient (or treatment-control coefficient contrast) in a regression model that includes prespecified covariates, such as baseline measurements. In the model \(Y_i=\mu+\tau T_i+\beta X_i+\varepsilon_i\), \(Y_i\) is the outcome, \(T_i\) indicates treatment assignment, \(X_i\) is a baseline covariate, and \(\tau\) is the treatment effect conditional on the covariate under the model.

Model and Precision of the Adjusted Contrast

For one active arm and a control group, suppose the outcome follows the common-slope linear model \(Y_i=\mu+\tau T_i+\beta X_i+\varepsilon_i\). Assume treatment assignment is randomized, \(X_i\) is measured before treatment, the model is correctly specified, and the errors are independent with mean zero and common variance \(\sigma^2\). Normal errors give exact normal finite-sample inference when the design matrix has full rank; with sufficiently large samples, approximate normal inference may be reasonable under weaker error distributions. The treatment effect is interpreted as a mean difference at a common covariate value, and is constant across covariate values under this model.

With several active arms, replace \(\tau T_i\) with \(\sum_k \tau_k D_{ik}\), where \(D_{ik}\) indicates assignment to arm \(k\); the contrast of interest is \(\tau_k\), the mean difference between arm \(k\) and control. The following precision argument applies to each contrast, with the design matrix accounting for all included arm indicators and covariates.

For the single-arm case, let \(Z\) contain the intercept and covariate columns, and let \(M_Z=I-Z(Z^\mathsf{T}Z)^{-1}Z^\mathsf{T}\) be the residual-maker matrix. Under the stated model and fixed design, the variance of the fitted treatment coefficient is

$$ \operatorname{Var}(\hat{\tau}_{\mathrm{adj}}\mid T,X) =\frac{\sigma^2}{T^\mathsf{T}M_ZT}. $$

This follows from the partial-regression identity: after removing the linear effects of the intercept and covariate from both outcome and treatment indicator, the coefficient is the regression of residualized outcome on residualized treatment. The residualized outcome has error variance \(\sigma^2\), so the coefficient variance is that variance divided by the sum of squared residualized treatment indicators, \(T^\mathsf{T}M_ZT\). The design must retain treatment variation after adjustment; otherwise the denominator is zero and the treatment coefficient is not identifiable.

For a simple comparison, if the covariate is similarly distributed in the randomized groups and adjustment removes a fraction \(R_X^2\) of outcome variance, a useful large-sample approximation is \(\operatorname{Var}(\hat{\tau}_{\mathrm{adj}})\approx(1-R_X^2)\operatorname{Var}(\hat{\tau}_{\mathrm{unadj}})\). This is an approximation, not a guarantee for every realized randomization. It explains why a prognostic baseline covariate can improve precision: less residual outcome variation remains to obscure the treatment contrast.

Worked Example: Precision Gain From a Prognostic Baseline Measure

Suppose a hypothetical two-arm trial assigns 50 participants to treatment and 50 to control. The outcome variance is 100 points squared, and a baseline covariate explains 36% of that variance under a correct common-slope model. Assume balanced covariate distributions and use the large-sample variance approximation. Compare the standard errors of the unadjusted and adjusted differences in means.

The unadjusted variance is

$$ 100\left(\frac{1}{50}+\frac{1}{50}\right) =100(0.04)=4\text{ points}^2, \qquad SE_{\mathrm{unadj}}=\sqrt{4}=2.000\text{ points}. $$

The residual variance after adjustment is approximately \(100(1-0.36)=64\) points squared. Thus

$$ 64\left(\frac{1}{50}+\frac{1}{50}\right) =64(0.04)=2.56\text{ points}^2, \qquad SE_{\mathrm{adj}}=\sqrt{2.56}=1.600\text{ points}. $$

Under these assumptions, adjustment reduces the standard error by \(0.400\) points, or 20% relative to the unadjusted standard error. The calculation does not imply that every covariate-adjusted analysis is more precise: a weakly prognostic covariate, an unbalanced realized design, or an incorrect model can reduce or remove the gain.

Selection Bias Under a Threshold Rule

Consider an adjusted interim estimate \(S\) for an arm’s treatment effect. Suppose \(S\) is normally distributed with mean \(\theta\) and standard deviation \(\sigma_S\), and the arm is retained when \(S>c\). The parameter \(\theta\) is the true treatment effect. The retained arms tend to have positive interim estimation errors because low values are less likely to pass the threshold. This is a conditional property of the selection rule, not evidence that the true effect has changed.

Write \(a=(c-\theta)/\sigma_S\), and let \(\phi\) and \(\Phi\) denote the standard normal density and cumulative distribution function. Standardizing \(S\) gives \(P(S>c)=1-\Phi(a)\). The expected excess of \(S\) over its mean among selected cases is

$$ E(S-\theta\mid S>c) =\sigma_S E\left(\frac{S-\theta}{\sigma_S}\,\middle|\,\frac{S-\theta}{\sigma_S}>a\right) =\sigma_S\frac{\phi(a)}{1-\Phi(a)}. $$

For the last equality, if \(Z\sim N(0,1)\), then \(\int_a^\infty z\phi(z)\,dz=\phi(a)\), since \(\phi'(z)=-z\phi(z)\). Dividing this tail integral by \(P(Z>a)=1-\Phi(a)\) gives the conditional mean. The factor \(\phi(a)/[1-\Phi(a)]\) is positive, so selection above a threshold creates upward conditional bias in the selected interim estimate.

Worked Example: Conditional Bias at the Same Standardized Boundary

Suppose a hypothetical interim selection rule retains an arm when its estimate exceeds a boundary 1.5 standard errors above its true mean. For illustration, take the adjusted estimate’s standard error to be 1.789 points and the unadjusted estimate’s standard error to be 2.236 points. These are alternative estimates under the same standardized boundary; the comparison does not hold the absolute cutoff fixed.

At \(a=1.5\), \(\phi(1.5)\approx0.1295\) and \(1-\Phi(1.5)\approx0.0668\), giving a ratio of approximately \(1.939\). Therefore the conditional bias in the selected adjusted estimate is

$$ 1.789\frac{\phi(1.5)}{1-\Phi(1.5)} \approx1.789(1.938677) =3.468\text{ points}. $$

For the unadjusted estimate at the same standardized boundary, the corresponding bias is

$$ 2.236\frac{\phi(1.5)}{1-\Phi(1.5)} \approx2.236(1.938677) =4.335\text{ points}. $$

The adjusted estimate has a smaller conditional bias in this comparison because its standard error is smaller, while the standardized selection boundary is held fixed. If instead the rule uses the same absolute cutoff for both estimates, the standardized boundary \(a\) changes too, and the bias comparison must be recalculated. The figures are a mathematical illustration, not results from a real trial.

From Interim Selection to the Final Estimate

The interim statistic and final estimate need not be the same random variable. Let \(F\) be the final adjusted estimate for the arm, and suppose \(S\) and \(F\) are jointly normal with means \(\mu_S\) and \(\mu_F\), standard deviations \(\sigma_S\) and \(\sigma_F\), and covariance \(\operatorname{Cov}(S,F)\). This model is exact for linear estimates under jointly normal errors; it can also serve as a large-sample approximation. If selection is \(S>c\), then

$$ E(F\mid S>c) =\mu_F+\frac{\operatorname{Cov}(S,F)}{\sigma_S} \frac{\phi(a)}{1-\Phi(a)}, \qquad a=\frac{c-\mu_S}{\sigma_S}. $$

To justify this, the conditional mean of a jointly normal \(F\) given \(S\) is \(E(F\mid S)=\mu_F+\{\operatorname{Cov}(S,F)/\sigma_S^2\}(S-\mu_S)\). Averaging this expression over \(S>c\), and substituting \(E(S-\mu_S\mid S>c)=\sigma_S\phi(a)/[1-\Phi(a)]\), yields the displayed result. When \(F\) and \(S\) estimate the same treatment effect, \(\mu_F=\mu_S=\theta\), so the second term is the final estimate’s conditional bias relative to \(\theta\).

Worked Example: Bias in a Correlated Final Estimate

Suppose a hypothetical trial selects an arm if its interim adjusted contrast \(S\) exceeds 6.25 points. Assume \(S\) has mean 4.00 points and standard deviation 1.50 points. The final adjusted estimate \(F\) also has mean 4.00 points and standard deviation 1.20 points, and the correlation between \(S\) and \(F\) is 0.60. Assume the joint normal model is adequate.

The standardized selection boundary is \(a=(6.25-4.00)/1.50=1.50\). The covariance is

$$ \operatorname{Cov}(S,F)=0.60(1.50)(1.20)=1.08\text{ points}^2. $$

Thus the conditional bias of the final estimate among selected arms is

$$ \frac{1.08}{1.50} \frac{\phi(1.50)}{1-\Phi(1.50)} =0.72(1.938677) \approx1.396\text{ points}. $$

The expected final estimate conditional on this selection event is therefore approximately \(4.00+1.396=5.396\) points. This is not the expected estimate for every possible selection mechanism; it follows from the specified joint distribution and threshold rule. If final data were independent of the interim statistic, the covariance would be zero and this particular conditional-bias term would be zero. In a MAMS design, however, interim observations are often included in the final estimate, creating positive dependence.

What Adjustment Changes—and What It Does Not

Adjustment can change the interim standard deviation, the final standard deviation, and their covariance. In the joint-normal formula, bias depends on \(\operatorname{Cov}(S,F)/\sigma_S\) as well as on the standardized boundary \(a\). A smaller \(\sigma_S\) does not guarantee smaller final bias if adjustment also changes the covariance or the effective selection boundary. For example, holding the absolute boundary \(c\) fixed while changing \(\sigma_S\) changes \(a\), and therefore changes the tail ratio as well.

A useful comparison is between three approaches. An unadjusted contrast is simple but may waste prognostic baseline information. A covariate-adjusted contrast can improve precision when the model is appropriate, but it remains subject to selection if arms are chosen using its interim values. A selection-aware estimator or interval accounts for the selection mechanism; it may use the adjusted joint distribution, but adjustment alone is not a selection correction.

ApproachPotential advantageKey limitation
Unadjusted treatment-control contrastDirect interpretation and fewer outcome-model assumptionsMay have larger variance when baseline covariates predict outcome
Covariate-adjusted contrastCan reduce residual variance and improve precisionRequires an adequate model; selection bias remains if this contrast drives selection
Selection-aware estimation using adjusted contrastsCan account for dependence between selection and estimationRequires a correctly specified selection rule and joint distribution or another justified correction

Diagnostics should focus on assumptions that support both the precision and selection calculations. Check whether baseline covariates were measured before treatment, whether the treatment arms have sufficient covariate overlap, and whether the treatment effect appears to vary with the covariate. Examine residual-versus-fitted and residual-versus-covariate plots for nonlinearity or unequal variance, and look for influential observations. Confirm that the analysis uses the same covariate definition and adjustment rule at interim and final analyses if the derivation assumes that it does. If the model includes treatment-by-covariate interactions, the common-slope interpretation and simple variance comparison no longer apply unchanged.

The normal tail calculation is also limited. It treats a single threshold event and a specified joint distribution. Selecting the largest of several noisy arm contrasts, applying multiple futility and efficacy boundaries, or using a complex adaptive rule generally creates a more complicated selection region. The single-threshold formula can explain the mechanism, but it is not a substitute for a calculation that matches the actual MAMS rule. In those settings, selection-aware methods need the full rule and the relevant joint distribution.

Common Mistakes

  • Equating precision gain with bias correction. A lower standard error can reduce the scale of selection bias in a particular comparison, but the selected estimate is still conditioned on passing the rule. Use the selection mechanism to assess bias.
  • Comparing biases without specifying the boundary. A fixed standardized boundary and a fixed absolute cutoff are different comparisons. State which is held constant and recalculate \(a\) when the standard error changes.
  • Using the interim standard error alone for the final estimate. Final conditional bias depends on the covariance of \(S\) and \(F\), not just on \(\sigma_S\). Specify which observations enter each estimate.
  • Assuming covariate adjustment always helps. Precision gains rely on prognostic information and an adequate model. Poor overlap, a misspecified relationship, or unnecessary parameters can compromise the analysis.
Key takeaway: Covariate adjustment can improve treatment-effect precision by removing predictable outcome variation. After arm selection, however, the expected bias of the final adjusted estimate depends on the selection boundary and on its covariance with the adjusted interim statistic. Estimate those quantities under the actual design rather than treating adjustment as a correction for selection.

Check Your Understanding

Answer each question, then compare your reasoning with the solutions below.

  1. A baseline covariate explains 25% of outcome variance. Under the balanced-design approximation, what is the ratio of adjusted to unadjusted variances? What is the ratio of their standard errors?
  2. An interim adjusted estimate has mean 3 points and standard deviation 2 points. An arm is selected when the estimate exceeds 6 points. Find the standardized boundary and state the direction of the conditional bias.
  3. A final estimate \(F\) is jointly normal with interim statistic \(S\), and \(\operatorname{Cov}(S,F)=0\). What is the conditional mean of \(F\) after selection on \(S>c\), under the joint-normal model?
  4. Why can the single-threshold formula not automatically be used for the arm with the largest estimate among several arms?

Solutions. (1) The variance ratio is approximately \(1-0.25=0.75\); the standard-error ratio is \(\sqrt{0.75}\approx0.866\). (2) \(a=(6-3)/2=1.5\). Since selection is above the mean and the tail ratio is positive, the selected interim estimates have positive conditional bias; its value is \(2(1.938677)\approx3.877\) points. (3) The covariance term is zero, so \(E(F\mid S>c)=\mu_F\); selection on \(S\) does not shift the mean of an independent jointly normal \(F\). (4) Selecting the largest estimate imposes a multivariate selection event involving all arms, not a single event \(S>c\). The single-threshold derivation does not represent that selection region or its dependence structure.

Clinical Biostats