Tutorials › Pattern-Mixture and Selection Models › Deriving J2R Conditional Means from Gaussian Models

Reference-Based Imputation for Clinical Trials · Tutorial 275 of 499

Deriving J2R Conditional Means from Gaussian Models

Learn how J2R combines an active-arm observed history with a reference-arm post-dropout mean trajectory to form a conditional predictive mean.

Advanced 12 min read

What You'll Learn

  • State the Gaussian and covariance assumptions needed for a J2R conditional mean.
  • Derive the conditional mean from a pattern-specific mean vector and covariance matrix.
  • Calculate predictions from one or several observed visits.
  • Distinguish J2R mean-centering from conditioning entirely under a reference-arm model.
  • Diagnose unstable predictions and identify what the J2R mean restriction leaves unspecified.
  • Check calculations and interpretations using solved practice problems.

Why a J2R Conditional Mean Needs More Than a Mean Restriction

Jump-to-reference (J2R) specifies that, after an active-arm participant drops out, the post-dropout mean trajectory follows the reference-arm mean trajectory. But a participant’s missing outcomes are predicted using their observed history as well. The prediction therefore depends on how observed and missing outcomes covary, not just on the post-dropout means.

This tutorial derives that prediction for a Gaussian pattern-mixture model. The key distinction is between the mean restriction, which sets the post-dropout mean, and the covariance model, which determines how the participant’s observed history shifts their conditional prediction. J2R’s mean restriction alone does not specify every covariance needed for that calculation.

Model and Assumptions

Let \(Y=(Y_1,\ldots,Y_K)^\mathsf{T}\) be a participant’s outcomes at \(K\) ordered visits. Consider a participant assigned to the active arm who drops out after visit \(s\), so their observed history is \(Y_o=(Y_1,\ldots,Y_s)^\mathsf{T}\) and their post-dropout outcomes are \(Y_m=(Y_{s+1},\ldots,Y_K)^\mathsf{T}\). We condition on this dropout pattern. Write \(y_o\) for the participant’s realized observed history.

For this pattern, suppose the joint distribution of observed and missing outcomes is multivariate normal with mean vector \(\mu^{(s)}\) and covariance matrix \(\Omega^{(s)}\). Partition them according to the observed and missing visits:

$$ \begin{pmatrix}Y_o\\Y_m\end{pmatrix} \Bigm|\text{pattern }s \sim N\left[ \begin{pmatrix}\mu_o^{(s)}\\\mu_m^{(s)}\end{pmatrix}, \begin{pmatrix} \Omega_{oo}^{(s)} & \Omega_{om}^{(s)}\\ \Omega_{mo}^{(s)} & \Omega_{mm}^{(s)} \end{pmatrix} \right]. $$

Assume \(\Omega_{oo}^{(s)}\) is nonsingular. For the standard J2R mean restriction, the observed-history means for this active-arm pattern are the active-arm means, while the post-dropout means are the corresponding reference-arm means:

$$ \mu_o^{(s)}=\mu_{A,o}, \qquad \mu_m^{(s)}=\mu_{R,m}. $$

Here \(A\) denotes the active arm and \(R\) the reference arm. The covariance blocks must also be specified by the model. For example, an analysis might use a common covariance matrix across arms or choose covariance blocks from a reference-arm model. The J2R mean restriction does not, by itself, choose between these options or imply that their covariance blocks are equal.

Definition: Under the Gaussian J2R mean restriction, the conditional predictive mean for an active-arm participant with observed history \(y_o\) is the conditional expectation of \(Y_m\) given \(Y_o=y_o\) in the specified dropout-pattern distribution. Its location combines the reference-arm post-dropout mean with a covariance-based adjustment for the participant’s deviation from the active-arm observed-history mean.

Deriving the Conditional Predictive Mean

For a jointly Gaussian vector partitioned into observed and missing components, the conditional expectation is

$$ E(Y_m\mid Y_o=y_o,\text{pattern }s) = \mu_m^{(s)} + \Omega_{mo}^{(s)} \left(\Omega_{oo}^{(s)}\right)^{-1} (y_o-\mu_o^{(s)}). $$

Substitute the J2R pattern means. The result is

$$ E(Y_m\mid Y_o=y_o,\text{pattern }s) = \mu_{R,m} + \Omega_{mo}^{(s)} \left(\Omega_{oo}^{(s)}\right)^{-1} (y_o-\mu_{A,o}). $$

The multiplication is dimensionally consistent: \(\Omega_{mo}^{(s)}\) has one row per missing visit and one column per observed visit, while \(\left(\Omega_{oo}^{(s)}\right)^{-1}(y_o-\mu_{A,o})\) is a vector with one entry per observed visit. If there is only one missing visit, the result is a scalar. With several missing visits, it is a vector of conditional means.

A short derivation explains the covariance adjustment. Set \(B=\Omega_{mo}^{(s)}(\Omega_{oo}^{(s)})^{-1}\) and define \(Z=Y_m-\mu_m^{(s)}-B(Y_o-\mu_o^{(s)})\). Its covariance with \(Y_o\) is

$$ \operatorname{Cov}(Z,Y_o) = \Omega_{mo}^{(s)}-B\Omega_{oo}^{(s)} = \Omega_{mo}^{(s)} -\Omega_{mo}^{(s)}(\Omega_{oo}^{(s)})^{-1}\Omega_{oo}^{(s)} =0. $$

Because \(Z\) and \(Y_o\) are jointly Gaussian, zero covariance implies independence. Thus the conditional expectation of \(Z\) given \(Y_o\) is its mean, zero. Rearranging the definition of \(Z\) gives the conditional expectation formula above. In J2R, substituting \(\mu_m^{(s)}=\mu_{R,m}\) and \(\mu_o^{(s)}=\mu_{A,o}\) yields the stated expression.

Worked Example: One Observed Visit

Suppose a hypothetical symptom score is modeled for an active-arm participant who drops out after visit 1. The active-arm mean at visit 1 is 50 points, the reference-arm mean at the missing visit 2 is 46 points, and the specified covariance model gives \(\operatorname{Var}(Y_1)=16\) points squared and \(\operatorname{Cov}(Y_2,Y_1)=8\) points squared. The participant’s observed score is 54 points. Assume the corresponding two-visit Gaussian pattern model and positive observed variance.

For one observed visit, the conditional regression coefficient is \(8/16=0.5\). The J2R conditional mean is

$$ \mu_{R,2} + \frac{\operatorname{Cov}(Y_2,Y_1)}{\operatorname{Var}(Y_1)} (y_1-\mu_{A,1}) = 46+\frac{8}{16}(54-50) = 46+2 = 48\text{ points}. $$

The participant is 4 points above the active-arm visit 1 mean, and the specified covariance model translates half of that difference into a 2-point adjustment. The predicted mean is therefore 48 points, not 46. The reference-arm mean supplies the J2R post-dropout location; it is not necessarily the participant-specific conditional mean.

Worked Example: Two Observed Visits Predict One Missing Visit

Consider a hypothetical three-visit symptom score. An active-arm participant drops out after visit 2 and has observed scores \(y_o=(12,16)^\mathsf{T}\). The active-arm means at visits 1 and 2 are \((10,14)^\mathsf{T}\), and the reference-arm mean at visit 3 is 12 points. Suppose the specified covariance blocks, in points squared, are

$$ \Omega_{oo}= \begin{pmatrix}4&2\\2&9\end{pmatrix}, \qquad \Omega_{mo}= \begin{pmatrix}1&3\end{pmatrix}. $$

The determinant of \(\Omega_{oo}\) is \(4(9)-2(2)=32>0\), and its leading diagonal entry is positive, so this observed-history covariance block is positive definite and invertible. The inverse and regression-weight row are

$$ \Omega_{oo}^{-1} = \frac{1}{32} \begin{pmatrix}9&-2\\-2&4\end{pmatrix}, \qquad \Omega_{mo}\Omega_{oo}^{-1} = \begin{pmatrix}1&3\end{pmatrix} \frac{1}{32} \begin{pmatrix}9&-2\\-2&4\end{pmatrix} = \begin{pmatrix}3/32&10/32\end{pmatrix}. $$

The participant’s deviations from the active-arm means are \((12-10,16-14)^\mathsf{T}=(2,2)^\mathsf{T}\). Therefore,

$$ E(Y_3\mid y_o,\text{pattern }2) = 12+ \begin{pmatrix}3/32&10/32\end{pmatrix} \begin{pmatrix}2\\2\end{pmatrix} = 12+\frac{26}{32} = 12.8125\text{ points}. $$

Both observed visits contribute, with weights \(3/32\) and \(10/32\). The computed conditional mean is above the reference-arm visit 3 mean because the observed history is above its active-arm mean and the specified cross-covariances produce a positive adjustment.

What the Formula Does—and Does Not—Say

The centering term is \(y_o-\mu_{A,o}\), not automatically \(y_o-\mu_{R,o}\). This follows from the stated J2R pattern mean: the observed visits are modeled with their active-arm means, while the missing visits use reference-arm means. If instead one conditions entirely under the reference-arm joint model, the resulting conditional mean is

$$ \mu_{R,m} + \Omega_{R,mo} \left(\Omega_{R,oo}\right)^{-1} (y_o-\mu_{R,o}). $$

That expression uses reference-arm centering and reference-arm covariance blocks. It is a distinct modeling choice unless the relevant means and covariance blocks make the two expressions equal. Similarly, continuing the active-arm mean trajectory after dropout would replace the reference post-dropout mean and would not implement the J2R mean restriction.

These distinctions matter because a mean restriction alone is not a full conditional model. Two analyses may agree on \(\mu_{R,m}\) but give different participant-specific predictions if they use different \(\Omega_{mo}\) or \(\Omega_{oo}\). If the covariance blocks are assumed common across arms, use those common blocks in the J2R formula; do not silently substitute reference-arm covariance blocks unless that is the specified model.

Worked Example: How the Covariance Choice Changes the Adjustment

For a hypothetical one-visit prediction, suppose the observed score is 54 points, its active-arm mean is 50 points, and the reference-arm mean at the missing visit is 46 points. Under covariance model 1, let \(\operatorname{Var}(Y_o)=16\) and \(\operatorname{Cov}(Y_m,Y_o)=8\), both in points squared. Under covariance model 2, let these be 16 and 4, respectively. Both models have a positive observed variance; assume each is part of a valid joint Gaussian covariance model.

Under model 1, the J2R conditional mean is \(46+(8/16)(54-50)=48\) points. Under model 2, it is

$$ 46+\frac{4}{16}(54-50) = 46+1 = 47\text{ points}. $$

The post-dropout mean remains 46 points in both calculations, but the history-based adjustment differs by 1 point. This is why the covariance assumption must be reported alongside the J2R mean restriction.

Diagnostics, Limitations, and Common Pitfalls

Before using the formula, check that the Gaussian pattern model and the covariance blocks are appropriate for the intended analysis. In particular, \(\Omega_{oo}^{(s)}\) must be invertible. A nearly singular block can produce unstable weights, so small changes in estimated covariances may cause large changes in predicted means. Check the covariance estimates for symmetry and positive definiteness, and examine whether the resulting conditional predictions are plausible in the outcome’s units and range. An implausible result is a reason to investigate the model and data, not to silently alter the formula.

The calculation is conditional on the specified dropout pattern. It does not establish that dropout is unrelated to unobserved outcomes, nor does it make the J2R assumption a fact verified by the observed data. The reference-arm post-dropout trajectory is a modeling restriction. As in other pattern-mixture analyses, conclusions can depend on that untestable choice.

  • Common mistake: “J2R sets the missing conditional mean equal to the reference-arm mean, so the prediction is just \(\mu_{R,m}\).” Correction: J2R sets the post-dropout component of the pattern mean vector to \(\mu_{R,m}\). Conditioning on the participant’s observed history adds the covariance adjustment unless that adjustment is zero.
  • Common mistake: Centering the observed history at \(\mu_{R,o}\) while claiming to use the hybrid J2R pattern mean. Correction: With active-arm observed-history means and reference-arm post-dropout means, the adjustment is centered at \(\mu_{A,o}\). Reference-arm centering corresponds to a different conditional-model specification.
  • Common mistake: Treating the J2R mean rule as if it also specified every covariance. Correction: State which covariance blocks are used. The conditional mean depends on them.
  • Common mistake: Reporting the conditional mean as a realized missing outcome. Correction: It is a model-based expected value given the observed history and pattern, not a guarantee about an individual outcome or a complete predictive distribution.

Compared with continuing the active-arm mean after dropout, J2R substitutes a reference-arm post-dropout mean. Compared with conditioning wholly under a reference-arm model, the hybrid J2R expression retains active-arm centering for the observed history. The choice is substantive: neither the reference trajectory nor the covariance specification is determined by the observed participant history alone.

Key takeaway: Under a Gaussian pattern-specific model, J2R’s conditional predictive mean equals the reference-arm post-dropout mean plus a covariance-weighted deviation of the observed history from its active-arm pattern mean. State both the mean restriction and the covariance model.

Check Your Understanding

Answer each question, then compare with the solutions below.

  1. A participant has one observed visit, \(y_o=7\), with active-arm mean 6. The reference-arm mean at the missing visit is 5, \(\operatorname{Var}(Y_o)=4\), and \(\operatorname{Cov}(Y_m,Y_o)=2\). Find the J2R conditional mean.
  2. For two observed visits, the covariance-based regression weights are \((0.25,0.50)\), the observed history is \((11,13)^\mathsf{T}\), the active-arm means are \((10,12)^\mathsf{T}\), and the reference-arm missing mean is 20. Find the conditional mean.
  3. Why does a J2R mean restriction not uniquely determine the conditional predictive mean?
  4. In what circumstance is the covariance adjustment zero?

Solutions. (1) \(5+(2/4)(7-6)=5.5\), so the predicted mean is 5.5 outcome units. (2) The deviations are \((1,1)^\mathsf{T}\), so the adjustment is \(0.25(1)+0.50(1)=0.75\); the conditional mean is \(20.75\) outcome units. (3) The adjustment depends on the specified cross-covariance and observed-history covariance blocks as well as on the observed deviation; the J2R mean restriction sets the post-dropout mean but does not, by itself, specify those covariance blocks. (4) It is zero when \(\Omega_{mo}^{(s)}(\Omega_{oo}^{(s)})^{-1}(y_o-\mu_{A,o})=0\), for example when the observed history equals its active-arm mean or when the relevant missing-observed covariance is zero.

Clinical Biostats