Tutorials › Multiple Imputation by Chained Equations (MICE) › Parameter And Residual Uncertainty In Continuous Imputation

Imputation Models by Variable Type · Tutorial 154 of 497

Parameter And Residual Uncertainty In Continuous Imputation

See how coefficient uncertainty and residual uncertainty play different roles in continuous MICE imputations, and what goes wrong when either is omitted.

Intermediate 13 min read

What You'll Learn

  • Distinguish uncertainty in a fitted regression mean from individual residual variation.
  • State the normal linear model assumptions behind continuous predictive draws.
  • Derive the two components of posterior predictive variance.
  • Compare complete stochastic draws with procedures that omit a source of uncertainty.
  • Explain why one coefficient draw is shared across missing cases in an imputation update.
  • Recognize diagnostics that can reveal under-dispersed or otherwise inappropriate imputations.

Why Two Kinds of Uncertainty Matter

A continuous imputation from a regression model is not just a predicted mean. For a case with predictors \(x\), the regression mean \(x^\mathsf{T}\beta\) is uncertain because \(\beta\) is estimated from a finite observed sample. Even if that mean were known exactly, an individual outcome would still vary around it. A proper predictive draw represents both sources: uncertainty about the regression parameters and residual variation among individuals.

These sources have different consequences. Coefficient uncertainty moves the fitted regression surface; it can shift predictions for many missing cases together. Residual uncertainty represents case-specific deviations around that surface. A procedure that includes one source but not the other generally produces imputations that are too concentrated in some respect.

Definition: In a continuous normal linear imputation model, parameter uncertainty is variation in the regression coefficients (and residual variance) across draws conditional on the observed data. Residual uncertainty is the variation in a new outcome around its conditional regression mean, represented by a residual draw. A predictive imputation includes both.

Model and Predictive Draws

Suppose \(y_o\) denotes the observed values of the target variable in the current MICE update, and \(X_o\) is the corresponding design matrix. For a case to be imputed, let \(x_m\) be its predictor vector, including an intercept if the model has one. The normal linear model is

$$ y_o = X_o\beta+\varepsilon_o,\qquad \varepsilon_o\sim N(0,\sigma^2 I). $$

Here \(\beta\) is a \(p\)-component coefficient vector and \(\sigma^2\) is the residual variance. The model assumes a correctly specified linear conditional mean, independent errors with constant variance, and normally distributed errors. The observed design matrix must have full column rank, and there must be more observed outcomes than coefficients. For this imputation model to be used without modeling the missingness process directly, the usual missing-data framework also requires an appropriate missing-at-random assumption and distinct model parameters.

One standard Bayesian draw uses the reference prior \(p(\beta,\sigma^2)\propto 1/\sigma^2\). Write \(\hat{\beta}\) for the least-squares estimate, \(SSE\) for the residual sum of squares, and \(\nu=n_o-p\), where \(n_o\) is the number of observed outcomes used in the fit. Then draw

$$ \sigma^{2*}=\frac{SSE}{Q},\qquad Q\sim\chi^2_{\nu}, \qquad \beta^*\mid\sigma^{2*},y_o\sim N\!\left(\hat{\beta},\sigma^{2*}(X_o^\mathsf{T}X_o)^{-1}\right). $$

The asterisk marks a draw, not a new fixed parameter. For each missing case, make the predictive draw

$$ y_m^*=x_m^\mathsf{T}\beta^*+\varepsilon_m^*, \qquad \varepsilon_m^*\mid\sigma^{2*}\sim N(0,\sigma^{2*}). $$

The coefficient draw and variance draw are shared across cases in the same model update; the residual draws are case-specific and independent conditional on those parameter draws. This distinction matters when imputing several values at once.

Why the Predictive Variance Has Two Parts

The separation follows directly from the law of total variance. Conditional on a coefficient and residual variance, a new response has mean \(x_m^\mathsf{T}\beta\) and variance \(\sigma^2\). Averaging over parameter uncertainty gives

$$ \begin{aligned} \operatorname{Var}(Y_m\mid y_o) &=E\!\left[\operatorname{Var}(Y_m\mid\beta,\sigma^2,y_o)\mid y_o\right]\\ &\quad+\operatorname{Var}\!\left(E[Y_m\mid\beta,\sigma^2,y_o]\mid y_o\right)\\ &=E(\sigma^2\mid y_o)+ x_m^\mathsf{T}\operatorname{Var}(\beta\mid y_o)x_m. \end{aligned} $$

The first term is residual uncertainty. The second is uncertainty in the fitted conditional mean. Under the reference-prior model above, let \(h_m=x_m^\mathsf{T}(X_o^\mathsf{T}X_o)^{-1}x_m\). Then \(\operatorname{Var}(\beta\mid y_o)=E(\sigma^2\mid y_o)(X_o^\mathsf{T}X_o)^{-1}\), so

$$ \operatorname{Var}(Y_m\mid y_o)=E(\sigma^2\mid y_o)(1+h_m). $$

The “1” represents residual variation; \(h_m\) scales coefficient uncertainty at the particular predictor values. Cases with predictor values far from the observed design can have larger \(h_m\), and therefore greater parameter uncertainty. If \(\nu>2\), then \(E(\sigma^2\mid y_o)=SSE/(\nu-2)\). The full predictive distribution is a Student \(t\) distribution with \(\nu\) degrees of freedom, center \(x_m^\mathsf{T}\hat{\beta}\), and scale \(s\sqrt{1+h_m}\), where \(s^2=SSE/\nu\). Its variance, when \(\nu>2\), is \(s^2(1+h_m)\nu/(\nu-2)\), consistent with the decomposition above.

Worked Example: Separating the Two Variance Components

Suppose a hypothetical regression fit has \(n_o=12\) observed outcomes, \(p=2\) coefficients, \(SSE=80\), and a missing case with \(x_m^\mathsf{T}\hat{\beta}=14\) and \(h_m=0.25\). The assumptions of the normal linear model are taken to hold for this illustration, and the design matrix is assumed to have full rank.

There are \(\nu=12-2=10\) residual degrees of freedom. The posterior expected residual variance is

$$ E(\sigma^2\mid y_o)=\frac{SSE}{\nu-2}=\frac{80}{10-2}=10. $$

Thus residual uncertainty contributes \(10\) squared response units to predictive variance. Coefficient uncertainty contributes \(h_m E(\sigma^2\mid y_o)=0.25(10)=2.5\) squared response units. Total predictive variance is \(10+2.5=12.5\), and the predictive mean is \(14\) response units. The predictive standard deviation is \(\sqrt{12.5}=3.536\) response units. Using only the regression mean would give no predictive variation at all; using residual draws around a fixed fitted mean would retain the \(10\) but omit the \(2.5\).

What Is Lost When a Source Is Omitted?

The following comparisons describe incomplete procedures, not interchangeable versions of the same predictive draw. A deterministic fitted-mean imputation sets every imputed value to \(x_m^\mathsf{T}\hat{\beta}\). It omits both sources and places cases exactly on the fitted regression surface. Adding residual draws while keeping \(\hat{\beta}\) fixed restores individual scatter, but still treats the estimated surface as known. Drawing coefficients but setting the residual to zero allows the surface to vary across imputations, but places each value exactly on whichever surface was drawn.

A coefficient draw is not a replacement for a residual draw: it describes uncertainty in the estimated conditional mean, not the spread of individual outcomes around that mean. Conversely, a residual draw around a fixed estimate does not account for how the fitted mean could change if a different plausible parameter value were used. In the standard normal model, both draws are needed for the predictive distribution.

Worked Example: Comparing Procedures at One Missing Case

Consider a hypothetical case for which the fitted mean is \(14\) units. Conditional on a particular residual-variance draw, suppose the predictive variance of the fitted mean due to coefficient uncertainty is \(1\) squared unit, and the residual variance is \(4\) squared units. These values describe the two components at this case; assume the model conditions for predictive draws are met.

With both sources included, predictive variance is \(1+4=5\) squared units. If the coefficient is fixed and only residuals are drawn, variance is \(4\) squared units, so coefficient uncertainty is missing. If coefficients are drawn but residuals are set to zero, variance is \(1\) squared unit, so individual scatter is missing. A deterministic fitted-mean imputation has variance zero across repeated imputations under this procedure.

For one illustrative realization, suppose the coefficient draw gives a predicted mean of \(15.2\) units and the residual draw is \(-1.1\) units. The imputed value is \(15.2-1.1=14.1\) units. The particular value is not required to be closer to \(14\) than another draw would be; the point is that the two random contributions enter in distinct places.

A further distinction appears when several missing cases are imputed using the same model update. Their coefficient uncertainty is generally shared, because one \(\beta^*\) is drawn for the update. Their residual errors are separate. Treating every case as if it had an independent coefficient draw would remove this shared component and give the wrong joint predictive behavior.

Worked Example: Shared Coefficient Uncertainty Across Two Cases

Suppose a hypothetical intercept-only model is used for two missing outcomes. Conditional on a variance draw, let the shared intercept draw have variance \(4\) squared units and let each independent residual draw have variance \(9\) squared units. Assume the normal model holds and the two cases are conditionally independent given the shared parameters.

For either case, \(Y_i=\beta+\varepsilon_i\), so its predictive variance is \(4+9=13\) squared units. For the two cases together,

$$ \operatorname{Cov}(Y_1,Y_2) =\operatorname{Cov}(\beta+\varepsilon_1,\beta+\varepsilon_2) =\operatorname{Var}(\beta)=4. $$

The residuals contribute no covariance because they are independent. The predictive correlation is \(4/\sqrt{13\cdot13}=4/13=0.3077\). If the intercept were held fixed, the shared covariance of \(4\) would disappear. If instead one incorrectly gave each case an independent intercept draw, that shared parameter component would also be lost. This is why parameter draws are made at the model-update level, while residual draws are made for individual cases.

Checks, Alternatives, and Limitations

After fitting a continuous imputation model, check that the fitted model is numerically stable and that the draws behave as intended. The design matrix must not be rank deficient, and the residual degrees of freedom must be positive. Extremely large coefficient draws, implausibly large or small residual-variance draws, or unstable chains can signal weak information, influential observations, or model problems. Such draws should be investigated rather than silently discarded.

Also compare the conditional spread of imputed values with the observed data in relevant predictor regions. A cloud of imputations that is much narrower than observed residual scatter can indicate that residual variation was omitted or understated. Imputations collapsing onto the regression line are a direct warning that predictive residuals were not added. Conversely, excessive spread can arise from an inappropriate model or poorly estimated variance. These checks do not prove that the conditional model is correct: the assumptions of linearity, normal errors, and constant variance still require substantive assessment.

The main alternatives differ in which uncertainty they represent. Deterministic regression imputation is simple but omits both predictive components. Fixed-coefficient regression with residual draws represents individual variation but omits coefficient uncertainty. Parameter-draw-only procedures represent uncertainty in the fitted surface but omit residual variation. A bootstrap-based approach can represent parameter uncertainty by refitting to resampled observed cases, but it still needs predictive residual variation for individual outcomes. Predictive mean matching is another commonly used approach for continuous variables; it does not use this same normal residual-draw construction. The appropriate choice depends on the conditional model and the imputation goal.

The normal linear predictive draw has limitations. It can produce values outside a variable’s meaningful range, and its linear, homoscedastic, normal-error assumptions may be unsuitable for a particular outcome. Transformations, bounded-variable models, two-part models, or models allowing changing variance address different structures and require their own assumptions. Do not try to correct a poor model merely by changing how its coefficient and residual draws are combined.

Common Mistake

Common mistake: “The regression coefficients were estimated with uncertainty, so drawing coefficients is enough to make the imputations realistic.” A coefficient draw varies the predicted mean, but it does not generate the individual scatter around that mean. The correction is to draw a residual for each missing case as well, using the residual variance draw, while sharing the coefficient draw across cases in the same model update.
1
Fit the conditional model.
Use observed target values and the predictors available in the current MICE update; check the design and residual degrees of freedom.
2
Draw model parameters.
Draw the residual variance and then the coefficient vector conditional on that variance and the observed data.
3
Draw missing outcomes predictively.
For each missing case, add an independent residual draw to its mean under the shared coefficient draw.

Check Your Understanding

Answer each question, then compare with the solutions below.

  1. In a normal linear imputation, what does a coefficient draw represent, and what does a residual draw represent?
  2. If residual draws are used with fixed regression coefficients, which predictive-variance component is omitted?
  3. A case has \(h_m=0.10\) and \(E(\sigma^2\mid y_o)=5\). What are the coefficient, residual, and total predictive variance components?
  4. Why can two missing outcomes have positively correlated imputations even when their residual draws are independent?
  5. Name two model assumptions that should be checked before relying on normal linear predictive draws.

Solutions. (1) A coefficient draw represents uncertainty in the fitted conditional mean; a residual draw represents case-specific variation around that mean. (2) It omits coefficient uncertainty in the fitted mean. (3) Coefficient variance is \(0.10(5)=0.5\), residual variance is \(5\), and total predictive variance is \(5.5\), all in squared response units. (4) The same coefficient draw is shared across cases, so uncertainty in that shared fitted surface induces covariance. (5) Examples include linearity of the conditional mean, normally distributed errors, independent errors, and constant residual variance.

Clinical Biostats