Why a Changing Reference Trajectory Matters
A reference arm does not necessarily have a flat mean outcome over follow-up. Control participants may improve, worsen, or change direction between visits. A J2R prediction must use the reference-arm mean for the visit being predicted, together with the reference trajectory’s covariance structure. Reusing a single control mean at every visit can distort the predicted path even when the covariance calculation is otherwise correct.
This tutorial focuses on calculating those predictions when the reference-arm mean changes over time. The calculations concern hypothetical outcomes for participants assigned to treatment who discontinue treatment or otherwise have missing follow-up outcomes. They are model-based predictions under an assumption, not observed outcomes or guaranteed descriptions of what happened.
In the examples, the outcome is measured in points, \(R\) denotes the reference arm, and \(A\) denotes the treatment arm. A dropout time \(k\) means outcomes through visit \(k\) are observed and later outcomes are missing. For clarity, numerical examples treat the estimated means and covariance parameters as fixed. In multiple imputation, parameter uncertainty must also be represented when drawing imputations.
Conditional Prediction with Visit-Specific Means
Suppose the reference-arm outcome vector across visits follows a multivariate normal model with mean vector \(\boldsymbol{\mu}_R\) and covariance matrix \(\boldsymbol{\Sigma}_R\). Partition the visits into the observed history \(P\) and the future visits \(F\). The reference model’s conditional mean for the future outcomes, given the observed history \(\mathbf y_P\), is
The first term, \(\boldsymbol{\mu}_{R,F}\), contains the reference mean at each future visit. The second term adjusts that mean for how the observed history differs from the reference-arm history, with the adjustment determined by the reference covariance. Thus, a J2R conditional prediction is not generally just the unconditional reference mean at the future visit. It uses the time-varying reference means and the conditional relationship between past and future outcomes.
For a useful special case, suppose the reference outcomes have equal variance and an AR(1) correlation structure: the correlation between adjacent visits is \(\rho\), and the correlation between visits \(i\) and \(j\) is \(\rho^{|i-j|}\). This model is Markov, so the next visit’s conditional mean given the full history depends only on the most recent outcome:
For sequential predictions, replace \(y_{j-1}\) with the observed outcome if visit \(j-1\) is observed, or with the preceding conditional prediction when calculating the next expected value. The reference mean increment is therefore present through the two visit-specific means; the covariance term adjusts for the preceding outcome’s deviation from its own visit’s reference mean.
Worked Calculations for Rising and Falling Means
Worked Example: A Reference Mean That Rises
Suppose a hypothetical trial measures an outcome in points at four visits. The reference-arm means are 10, 12, 15, and 18 points. Assume equal reference-arm variances and AR(1) correlation \(\rho=0.50\). A treatment participant drops out after visit 2, with \(Y_2=14\) points observed. Calculate the sequential J2R conditional means for visits 3 and 4.
The visit-3 prediction uses the reference mean at visit 3 and the deviation at visit 2 from the visit-2 reference mean:
For the visit-4 expected value, the preceding predicted value is \(m_3=16\). Its deviation from the visit-3 reference mean is \(16-15=1\) point. Therefore,
The predictions are 16 points at visit 3 and 18.5 points at visit 4. The visit-specific reference means rise from 15 to 18 points, while the conditional adjustment reflects the preceding prediction’s deviation from the corresponding reference mean. The calculation would be different if the reference mean were incorrectly held at 12 points after dropout.
Worked Example: A Reference Mean That Falls
Now suppose the reference-arm means at four visits are 20, 18, 16, and 15 points, with the same AR(1) correlation \(\rho=0.60\). A treatment participant is observed only through visit 1 and has \(Y_1=23\) points. Calculate the conditional means at visits 2 through 4.
At visit 2, the observed outcome is 3 points above the reference mean at visit 1:
At visit 3, the visit-2 prediction is 1.8 points above the reference mean at visit 2. Thus,
At visit 4, the visit-3 prediction is 1.08 points above the reference mean at visit 3:
The conditional predictions are 19.8, 17.08, and 15.648 points. They move downward as the reference means fall, but each remains above its corresponding reference mean because the observed history began above the reference trajectory and the AR(1) model carries some of that deviation forward. A falling reference mean does not imply that every conditional prediction must fall by exactly the same amount.
The AR(1) recursion is convenient, but it is not a substitute for the general conditional formula when the covariance structure is not AR(1), when the observed history is irregular, or when the full pattern of correlations matters. The next example calculates a conditional prediction directly from a partitioned covariance matrix.
Worked Example: Conditioning on Two Observed Visits
Suppose a participant has observed outcomes \(Y_1=5\) and \(Y_2=8\) points. The hypothetical reference-arm means at visits 1, 2, and 3 are 4, 6, and 9 points. The reference covariance matrix is
For predicting visit 3, the observed-history covariance block and the future-to-history covariance block are
The inverse of the observed-history block is \(\frac{1}{12}\begin{pmatrix}4&-2\\-2&4\end{pmatrix}\). The observed residual vector is \(\begin{pmatrix}5-4\\8-6\end{pmatrix} =\begin{pmatrix}1\\2\end{pmatrix}\). The covariance multiplier is
The conditional prediction is therefore
The visit-3 reference mean is 9 points, and the observed history adds a conditional adjustment of 1 point. Using the reference mean at visit 2 instead of visit 3 would give the wrong baseline for this prediction.
Justification of the Conditional Formula
The conditional-mean expression follows from the joint normal model. Define the future residual after its linear adjustment for the observed history as
This residual has mean zero. Its covariance with the observed history is
Because \(\mathbf U\) and \(\mathbf Y_P\) are jointly normal, zero covariance implies independence. Hence the conditional expectation of \(\mathbf U\) given \(\mathbf Y_P=\mathbf y_P\) is still zero. Rearranging the definition of \(\mathbf U\) gives the stated conditional mean. This argument also shows why the covariance matrix must match the modeled visits and why the observed-history block must be invertible.
Checks, Alternatives, and Limitations
Several practical checks help catch errors before interpreting a J2R trajectory:
- Match the visit index. For a prediction at visit \(j\), use \(\mu_{R,j}\), not a baseline or dropout-visit mean by default.
- Check the residual reference visit. In the AR(1) recursion, subtract \(\mu_{R,j-1}\) from the preceding outcome or prediction, not \(\mu_{R,j}\).
- Confirm the covariance scale and ordering. The rows and columns must correspond to the same visits as the mean vector, and the covariance matrix must be valid for the modeled outcomes.
- Inspect the resulting path. Plot predictions alongside the visit-specific reference means and observed outcomes. Unexpected jumps, reversals, or extreme values can indicate a coding, indexing, or model-fit problem; they are not automatically proof that the assumption is wrong.
- Check support for estimation. Sparse reference-arm data or an inadequate covariance model can make conditional predictions unstable. A mathematically computable prediction is not necessarily a reliable one.
J2R is one assumption for missing outcomes, not a consequence of the observed data alone. The contrast with common alternatives is useful:
| Approach | Post-dropout prediction basis | How it differs here |
|---|---|---|
| J2R | Reference-arm conditional distribution after dropout | Uses the reference mean at each future visit and the specified reference covariance. |
| MAR-based prediction | A fitted outcome model conditional on observed data under a missing-at-random assumption | Does not impose the J2R reference-arm post-dropout distribution. |
| Copy-reference | A reference-based rule that retains a treatment-related component in the post-dropout trajectory | Does not use the same J2R assumption that post-dropout outcomes follow the reference conditional model. |
A common mistake is to set every missing outcome equal to the control mean at the dropout visit, or to repeat that one mean at all later visits. That ignores a changing reference trajectory and, in the conditional model, the covariance adjustment. The correction is to use each future visit’s reference mean and apply the appropriate conditional adjustment. Conversely, using a future reference mean without checking its visit index, outcome scale, and covariance model does not make a calculation valid.
The predictions also depend on the chosen reference arm, mean model, covariance structure, and J2R assumption. They do not establish that a participant’s unobserved outcomes truly followed the reference-arm process. Sensitivity analysis may be needed to assess how conclusions change under alternative assumptions; the J2R calculation itself does not remove that uncertainty.
Check Your Understanding
Answer each question, then compare your work with the solutions below.
- Under an AR(1) reference model with \(\rho=0.40\), the reference means at visits 1 and 2 are 7 and 10 points. If the observed visit-1 outcome is 12 points, what is the visit-2 J2R conditional mean?
- The reference means at visits 2 and 3 are 14 and 11 points. A preceding conditional prediction is 16 points, and \(\rho=0.50\). Calculate the next conditional mean.
- In the general conditional formula, what role does the reference covariance play in addition to specifying the reference means?
- Why is repeating the reference mean at the dropout visit for all later visits not generally a valid calculation when control means change over time?
Solutions. (1) \(10+0.40(12-7)=10+2=12\) points. (2) Use the visit-3 mean and the deviation from the visit-2 mean: \(11+0.50(16-14)=12\) points. (3) The covariance determines how deviations in the observed history adjust the conditional mean of future outcomes; in the formula this is the multiplier \(\boldsymbol{\Sigma}_{R,FP}\boldsymbol{\Sigma}_{R,PP}^{-1}\). (4) The reference mean can differ by future visit, and the conditional prediction may also depend on the observed history through the reference covariance. Freezing the dropout-visit mean discards both the changing visit-specific reference means and, if omitted, the conditional adjustment.