Tutorials › Causal Inference › Continuous Treatment Error and Regression Calibration

Missing Data and Measurement Error · Tutorial 918 of 972

Continuous Treatment Error and Regression Calibration

Use calibrated treatment values in confounding-adjusted outcome models, and assess when this correction identifies a causal effect.

Advanced 10 min read

What You'll Learn

  • Define the calibrated treatment as its conditional expectation given the error-prone measurement and measured confounders.
  • Estimate that conditional expectation from a validation sample or under an explicit replicate-measurement model.
  • Derive why regression calibration works for a linear conditional outcome mean.
  • State the causal and measurement assumptions needed to interpret the calibrated coefficient causally.
  • Recognize why calibration alone does not generally correct nonlinear outcome models.
  • Account for uncertainty introduced by estimating the calibration model.

Why Calibrate a Continuous Treatment?

Suppose an observational study records a continuous treatment such as a daily dose, exposure level, or number of treatment hours. The recorded value may be an imperfect measurement of the treatment that matters for the outcome. Treating that recorded value as exact can distort an adjusted association, often pulling a linear treatment coefficient toward zero under classical nondifferential error. The distortion can also affect how confounding adjustment is carried out.

Regression calibration replaces the error-prone treatment with its expected true value given the recorded measurement and measured confounders. It is useful because the calibration can be estimated from a validation sample, where both true and recorded treatment are available, or from replicate measurements under a specified measurement model. Calibration is not a universal correction: its causal interpretation depends on the outcome model and on assumptions about confounding and measurement error.

Definition: Let \(A\) be the true continuous treatment, \(W\) its error-prone measurement, \(X\) a vector of measured pretreatment confounders, and \(Y\) the outcome. The regression-calibrated treatment is $$ \widehat{A}_{\mathrm{cal}} = E(A\mid W,X). $$ In practice, the conditional expectation is unknown and is estimated from validation or replicate data. The estimated value is then used in place of \(A\) in an outcome model.

The conditioning on \(X\) is important. People with the same recorded value \(W\) may have different expected true treatments because their covariates predict treatment. Calibrating on \(W\) alone can therefore discard information needed for confounding adjustment.

Estimating the Calibration Function

In a validation sample, the true treatment \(A\) and error-prone measurement \(W\) are both observed, along with \(X\). One can fit a regression of \(A\) on \(W\) and \(X\), possibly including nonlinear terms or interactions when needed. The fitted conditional mean is then evaluated for each participant in the main analysis sample. The validation sample must provide information relevant to the target sample: in particular, the conditional relationship between \(A\) and \((W,X)\) must transport, or any validation-sampling differences must be appropriately handled.

Replicate measurements can provide information about measurement-error variability. A commonly used model assumes \(W_j=A+U_j\), where the replicate errors \(U_j\) have mean zero, are independent of \(A\) and \(X\), and are independent across replicates. Replicates alone do not automatically determine \(E(A\mid W,X)\): estimating it also requires a model for the distribution of \(A\) given \(X\), or other identifying assumptions. The following normal model gives one explicit route.

Worked Example: Calibration From a Validation Sample

In a hypothetical validation sample, a fitted conditional-mean model for true daily dose is \(E(A\mid W,X)=8+0.6W+1.5X\), where \(W\) and \(A\) are in milligrams per day and \(X\) is a measured, centered baseline severity score. For a participant with \(W=10\) mg/day and \(X=2\), calculate the calibrated dose.

Substitute the participant's values into the fitted calibration model: $$ \widehat{A}_{\mathrm{cal}}=8+0.6(10)+1.5(2)=8+6+3=17\ \text{mg/day}. $$

The recorded value is 10 mg/day, but the conditional expected true dose for someone with this measurement and severity score is 17 mg/day under the fitted model. This is a model-based prediction, not a claim that the participant's true dose is known to be exactly 17 mg/day. The example assumes the validation model is correctly specified and applicable to this participant's population.

Worked Example: Calibration From Replicate Measurements

Suppose a hypothetical study has two replicate measurements, \(W_1\) and \(W_2\), for each participant. Assume \(A\mid X\sim N(\mu_X,16)\), and \(W_j=A+U_j\), where the independent errors satisfy \(U_j\sim N(0,9)\), independently of \(A\) and \(X\). These variances are in squared dose units. For a participant with \(\mu_X=10\) mg/day, \(W_1=14\) mg/day, and \(W_2=16\) mg/day, calculate the calibrated dose.

The replicate average is \(\bar W=(14+16)/2=15\) mg/day. Its measurement-error variance is \(9/2=4.5\). Under the stated normal model, the conditional mean of \(A\) given this average and \(X\) is $$ E(A\mid \bar W,X)=\mu_X+ \frac{\sigma_A^2}{\sigma_A^2+\sigma_U^2/2}(\bar W-\mu_X), $$ where \(\sigma_A^2=16\) and \(\sigma_U^2=9\). The coefficient is $$ \frac{16}{16+9/2}=\frac{16}{20.5}\approx 0.7805. $$

Therefore, $$ \widehat{A}_{\mathrm{cal}}=10+0.7805(15-10) =10+3.9025\approx 13.90\ \text{mg/day}. $$

This prediction lies between the covariate-specific mean of 10 mg/day and the observed replicate average of 15 mg/day. That shrinkage follows from the model: the replicate average contains measurement error, so it is not treated as the true dose. In an application, the distributional and independence assumptions, as well as the variance parameters, must be justified or assessed using the available data.

Why Calibration Can Recover a Linear Causal Effect

Calibration is most transparent with a continuous outcome and a correctly specified linear conditional mean. Let \(Y(a)\) denote the outcome that would be observed if treatment were set to \(a\). For a causal interpretation, suppose consistency holds, treatment is conditionally exchangeable given \(X\), and positivity holds for the treatment levels being compared. In words, the observed outcome equals the potential outcome at the received treatment, there is no unmeasured confounding after conditioning on \(X\), and the treatment levels of interest occur across the relevant covariate values.

In addition, suppose the conditional mean is linear with a constant treatment effect: \(E(Y\mid A,X)=\beta_0+\beta_A A+\beta_X^{T}X\). Assume that the measurement process is nondifferential for the outcome in the sense that \(E(Y\mid A,W,X)=E(Y\mid A,X)\). This condition says that once true treatment and \(X\) are known, the measurement error does not carry further information about the outcome. These conditions are substantive; calibration by itself does not establish them.

Result: Under the stated linear outcome and nondifferential measurement assumptions, $$ E(Y\mid W,X)=\beta_0+\beta_A E(A\mid W,X)+\beta_X^{T}X. $$ Thus, if \(E(A\mid W,X)\) is correctly specified, regressing \(Y\) on the calibrated treatment and \(X\) recovers the treatment coefficient \(\beta_A\) at the population level, subject to adequate variation and a correctly specified fitted model. Under consistency, conditional exchangeability, and positivity, \(\beta_A\) is the constant causal mean change in \(Y\) per one-unit increase in treatment.

The identity behind this result follows by iterated expectation. First condition on the true treatment, the recorded measurement, and covariates. Then average over the conditional distribution of the true treatment given the observed variables:

$$ \begin{aligned} E(Y\mid W,X) &=E\{E(Y\mid A,W,X)\mid W,X\}\\ &=E\{\beta_0+\beta_A A+\beta_X^{T}X\mid W,X\}\\ &=\beta_0+\beta_A E(A\mid W,X)+\beta_X^{T}X. \end{aligned} $$

The final equality holds because \(\beta_0\), \(\beta_A\), \(\beta_X\), and \(X\) are fixed when conditioning on \(W,X\). It is this linearity that lets the conditional expectation of \(A\) stand in for \(A\) in the outcome mean. If the causal effect varies with \(X\), a suitable interaction model may also be used: for example, the conditional expectation of \(AX\) given \(W,X\) is \(X E(A\mid W,X)\) when \(X\) is included in the conditioning set. The outcome and calibration models must then represent the relevant interactions correctly.

Worked Example: Interpreting a Calibrated Causal Effect

In a hypothetical observational analysis, a validation-based calibration model is used to construct \(\widehat{A}_{\mathrm{cal}}\). A fitted outcome model is \(E(Y\mid \widehat{A}_{\mathrm{cal}},X)=20+1.2\widehat{A}_{\mathrm{cal}}+0.5X\), where \(Y\) is a symptom score, treatment is measured in mg/day, and \(X\) is a measured baseline covariate. Calculate and interpret the model-implied mean contrast for a five mg/day treatment increase at the same \(X\).

For a change from dose \(a\) to \(a+5\), the \(X\) term and intercept cancel: $$ \Delta E(Y)=1.2\{(a+5)-a\}=1.2(5)=6.0\ \text{symptom-score points}. $$

If the calibration model, linear outcome model, measurement assumptions, and causal assumptions above are valid, the model implies a mean symptom score 6.0 points higher under a treatment level five mg/day greater, at the same covariate value. This is a causal interpretation only under those assumptions. Without them, 6.0 points is an adjusted model-based association, not necessarily a causal effect. The units are essential: the coefficient is 1.2 symptom-score points per mg/day.

Limits, Inference, and Common Pitfalls

The key limitation is that conditional expectation does not pass through a nonlinear function. If the outcome mean is \(m(A,X)\), then in general \(E\{m(A,X)\mid W,X\}\ne m\{E(A\mid W,X),X\}\). For example, if the outcome mean depends quadratically on treatment, knowing the conditional mean of treatment alone does not generally determine the conditional mean outcome; the conditional variance and other features of \(A\mid W,X\) can matter. Replacing \(A\) by its calibrated mean in a nonlinear outcome model is therefore not generally an exact correction. Regression calibration may still be used as an approximation in some settings, but its accuracy must not be assumed from the linear result.

A second issue is uncertainty. The calibrated treatment is estimated, not observed. Treating its estimated values as fixed can understate uncertainty in the final causal estimate. A bootstrap that repeats the calibration and outcome-model fitting in each resample can propagate this uncertainty when the sampling design permits it. Other approaches may jointly estimate model components or use design-appropriate variance methods. If the validation sample has a separate sampling process, that process also needs to be reflected in estimation and uncertainty assessment.

A common mistake is to say, “I replaced the noisy dose with its predicted true dose, so the treatment coefficient is now causal.” This reasoning skips two separate requirements. Calibration addresses a specified treatment-measurement problem; it does not remove unmeasured confounding, make a misspecified outcome model correct, or guarantee that a nonlinear outcome model is corrected. The causal interpretation additionally needs consistency, conditional exchangeability, positivity, appropriate measurement assumptions, and a suitable outcome model.

1
Specify the target and assumptions.
Define the treatment contrast, confounders, outcome scale, and causal assumptions. State what \(W\) measures and why the error model is plausible.
2
Estimate the calibration function.
Use validation observations of \(A,W,X\), or an explicit replicate-measurement model, to estimate \(E(A\mid W,X)\). Check its fit and applicability to the analysis sample.
3
Fit and interpret the outcome model.
Use the calibrated treatment with the measured confounders in an outcome model justified by the outcome scale and scientific question. Interpret a coefficient causally only when the identification and model assumptions support it.
4
Reflect calibration uncertainty.
Use an inference procedure that accounts for estimating the calibration function and, where relevant, the validation-sample design.
Key takeaway: Regression calibration replaces a continuous error-prone treatment with \(E(A\mid W,X)\). It recovers a confounding-adjusted constant causal effect under a correctly specified linear conditional outcome mean and appropriate causal and measurement assumptions. It is not a general-purpose correction for nonlinear models or unmeasured confounding.

Check Your Understanding

Use the assumptions and results in this tutorial to answer the following.

  1. Why can two people with the same recorded treatment \(W\) have different calibrated treatments when their covariates \(X\) differ?
  2. In the replicate example, what role did the normal distribution assumptions play in calculating the calibrated dose?
  3. State the iterated-expectation step that gives the linear regression-calibration identity.
  4. Why does replacing \(A\) by \(E(A\mid W,X)\) not generally give an exact correction for a nonlinear outcome mean?
  5. Name two causal assumptions needed before interpreting a calibrated outcome-model coefficient as a causal effect.
  6. Why may a standard error that treats calibrated treatment values as fixed be too small?
Clinical Biostats