Tutorials › Adaptive Clinical Trial Design › Diagnosing Joint Model Fit And Convergence

Phase 1/2 Designs With Efficacy and Toxicity · Tutorial 649 of 799

Diagnosing Joint Model Fit And Convergence

Use computational, likelihood-based, and residual diagnostics to locate problems in a joint dose model and choose remedies that address their actual source.

Advanced 11 min read

What You'll Learn

  • Distinguish sampler convergence from parameter identifiability and model fit.
  • Interpret trace plots, split-\(\hat R\), and effective sample size for joint-model parameters.
  • Recognize when sparse dose allocation leaves response parameters weakly identified.
  • Calculate and interpret cell-level residuals for joint efficacy-toxicity outcomes.
  • Diagnose posterior dependence without confusing it with efficacy-toxicity association.
  • Match remedies to computational, information, and fit problems.

Why Joint-Model Diagnostics Need More Than One Question

A joint efficacy-toxicity model can produce a dose recommendation even when its computation, information content, or fit deserves concern. Those are distinct issues. A sampler may fail to explore a well-defined posterior; a sampler may accurately explore a posterior that is broad because the data barely distinguish parameter values; or the model may fit poorly despite stable computation. Diagnostics are useful only when they identify which of these problems is plausible.

The model considered here assigns each participant an efficacy outcome and a toxicity outcome at an assigned dose. Both outcomes may be binary, ordinal, or otherwise modelled jointly. The particular likelihood can vary, but the diagnostics below require that the analyst state the fitted model and assumptions clearly: participants' outcome pairs are independent conditional on assigned dose and included covariates; the joint likelihood represents the recorded outcomes; and the dose-response and dependence structure have been specified. For Bayesian fitting, the prior must also be stated. If dose assignment adapts to prior outcomes, the assignment history should be recorded; when inference conditions on assigned doses and the allocation probabilities contain no unknown outcome-model parameters, the outcome likelihood can be evaluated conditional on those doses. Adaptive allocation can nevertheless leave some dose regions with little information.

Definition: Convergence concerns whether a computational procedure, such as a Markov chain Monte Carlo sampler, has adequately explored its target distribution. Identifiability concerns whether distinct parameter values imply distinct distributions for the observable data, and hence can in principle be distinguished by the likelihood. Practical identification concerns how precisely the available data constrain those parameters. Posterior precision can depend on both likelihood information and the prior; a precise posterior alone does not establish that the likelihood identifies a parameter. Fit diagnostics assess whether model-based predictions are compatible with observed data. These questions are related, but none answers the others.

In practice, diagnose in layers. First check computation, then ask whether the data provide information about the parameters, and then inspect whether fitted predictions reproduce important features of the observed joint outcomes. The earlier material on posterior predictive checks introduced comparison of observed data with replicated data; here the emphasis is on localizing patterns and choosing a response to them.

Checking Computation Before Interpreting Posterior Summaries

For simulation-based posterior inference, inspect trace plots by chain and parameter, and examine numerical summaries such as split-\(\hat R\) and effective sample size. A trace plot should show chains moving through the same posterior region rather than remaining in separate bands, drifting persistently, or making only rare jumps between regions. A satisfactory-looking trace is not a proof of convergence, so use multiple diagnostics together.

For \(m\) chains with \(n\) retained draws each, a basic variance comparison underlying \(\hat R\) uses the average within-chain variance \(W\), the between-chain variance \(B\), and the estimated marginal variance \(\hat V\):

$$ \hat V = \frac{n-1}{n}W+\frac{1}{n}B, \qquad \hat R = \sqrt{\frac{\hat V}{W}}. $$

If chains explore the same stationary distribution, their within-chain and between-chain variation should become similar, bringing \(\hat R\) close to 1. Modern implementations commonly use split chains and rank-based refinements, so software-reported values need not equal this basic calculation exactly. A value notably above 1 is a warning to investigate, not a direct measure of how wrong an estimate is. Effective sample size (ESS) estimates how many independent draws would provide comparable Monte Carlo information; strong autocorrelation can make ESS much smaller than the number of stored draws.

Worked Example: A Parameter That Has Not Mixed Across Chains

Suppose four chains each retain \(n=2{,}000\) draws for a dependence parameter in a joint model. For an illustrative basic variance calculation, the average within-chain variance is \(W=0.040\), and the between-chain variance estimate is \(B=0.120\), both in squared parameter units. Calculate \(\hat R\) from the formula above.

Substitution gives \(\hat V=(1-1/2{,}000)(0.040)+(1/2{,}000)(0.120) =0.03998+0.000060=0.04004\). Thus \(\hat R=\sqrt{0.04004/0.040}=1.0005\). This basic calculation would not flag a problem; with a large \(n\), the stated \(B\) contributes little to this particular uncorrected formula. That result illustrates why the exact diagnostic implementation matters: split-chain and rank-based procedures are designed to be more sensitive to chain differences than this simplified calculation in some situations.

Now suppose a modern split-\(\hat R\) reports 1.12 and bulk ESS is 85. The chains' trace plots also show that two chains remain near one region and two near another. These diagnostics together are concerning even though the simplified calculation is near 1. The responsible conclusion is that posterior summaries for this parameter are not yet trustworthy. Increase computation only after checking for poor parameterization, constrained geometry, or a multimodal target; simply storing more highly correlated draws may not resolve the problem.

This example also shows why one number should not dictate the decision. Check the trace, \(\hat R\), bulk and tail ESS, sampler warnings, and whether relevant functions of parameters are adequately sampled. If a dose recommendation depends on a derived probability or the joint distribution of two parameters, assess computation for that quantity as well as for individual coefficients. A stable mean for one parameter does not guarantee adequate exploration of a tail or joint region.

Identifiability and Sparse Dose Information

Structural identifiability is a property of the model's mapping from parameters to observable-data distributions. If distinct parameter values always imply distinct distributions, the parameter is structurally identifiable under that model. Practical identification is different: a model can be structurally identifiable yet have data so sparse or unbalanced that the likelihood is nearly flat along some parameter combinations. This often occurs in dose finding when few participants are observed at some doses, when outcomes are rare, or when several flexible dose-response and dependence parameters are estimated together.

A prior can regularize a weakly informed parameter and make posterior computation stable. It cannot make the data themselves distinguish parameter values. Therefore, compare prior and likelihood information, inspect likelihood profiles or posterior contours where appropriate, and check whether estimates change substantially under reasonable prior alternatives. This is a diagnostic of how much the data contribute, not a reason to choose a prior merely because it produces a narrow interval.

Worked Example: A Dose Slope With No Dose Variation

Consider the efficacy component of a joint model, \(Y_E\mid x \sim \operatorname{Bernoulli}(p_E(x))\), with \(\operatorname{logit}\{p_E(x)\}=\alpha_E+\beta_E x\), where \(x\) is a coded dose. Suppose all 30 participants receive the reference dose \(x=0\), and 8 have an efficacy response. The efficacy likelihood is \(L(\alpha_E,\beta_E)=p_E(0)^8\{1-p_E(0)\}^{22}\).

Since \(p_E(0)=\operatorname{logit}^{-1}(\alpha_E)\), the likelihood becomes \(\{\operatorname{logit}^{-1}(\alpha_E)\}^8 \{1-\operatorname{logit}^{-1}(\alpha_E)\}^{22}\). The parameter \(\beta_E\) does not appear. For every fixed \(\alpha_E\), all values of \(\beta_E\) give the same likelihood. The efficacy slope is therefore not identified by these data, regardless of how well a sampler runs.

A prior on \(\beta_E\) could produce a proper posterior distribution, but the resulting slope uncertainty would be driven by that prior rather than by dose contrasts in the observed outcomes. A matched remedy is to obtain information at multiple doses if the design permits, or to simplify the dose-response structure if the data cannot support the planned complexity. Reporting a highly precise slope solely because a restrictive prior narrowed it would not resolve the information problem.

For less extreme cases, look for strong posterior correlations, broad or irregular likelihood profiles, and estimates that move substantially when a small number of observations or plausible prior choices change. These are warnings of weak practical identification, not automatic proof that the model is invalid. They should prompt a substantive review of which parameters the available dose allocation and outcome counts can support.

Residual Patterns in Joint Outcomes

A joint model can reproduce each marginal response rate while failing to reproduce the pairing of efficacy and toxicity outcomes, or it can fit the joint table at one dose while missing a dose trend. Inspect observed and model-expected counts for the joint outcome categories across dose groups. For an observed count \(O\) and its fitted expected count \(E\), the Pearson residual is \(r=(O-E)/\sqrt{E}\). Its sign indicates whether the observed count exceeds or falls below expectation; its magnitude gives a scale for discrepancy relative to the model's count variability. For sparse expected counts, treat these residuals descriptively rather than relying on a large-sample reference distribution.

Worked Example: A Dose-Localized Joint Residual Pattern

At each of two dose groups, suppose 40 hypothetical participants have complete binary efficacy and toxicity outcomes. The fitted joint model gives the following expected counts, and the observed data give the counts shown. Cell order is no efficacy/no toxicity, efficacy/no toxicity, no efficacy/toxicity, efficacy/toxicity.

Dose groupExpected countsObserved counts
Lower24, 8, 6, 225, 7, 6, 2
Higher12, 8, 12, 88, 12, 8, 12

At the lower dose, the residuals are \((25-24)/\sqrt{24}=0.204\), \((7-8)/\sqrt{8}=-0.354\), \((6-6)/\sqrt{6}=0\), and \((2-2)/\sqrt{2}=0\). These are small descriptive discrepancies.

At the higher dose, the four residuals are \((8-12)/\sqrt{12}=-1.155\), \((12-8)/\sqrt{8}=1.414\), \((8-12)/\sqrt{12}=-1.155\), and \((12-8)/\sqrt{8}=1.414\). The repeated signs show that the observed higher-dose table has fewer participants in both no-efficacy cells and more in both efficacy cells than expected. This pattern suggests that the fitted model's higher-dose efficacy distribution may not track the observed counts well. It does not, by itself, identify the cause or establish a formal test result; inspect the model's assumptions, uncertainty, and other doses before deciding what revision is warranted.

Plot residuals or observed-versus-predicted summaries by dose and outcome category rather than examining only a single overall discrepancy. A pattern that repeats across neighbouring doses is more informative than one isolated cell, especially when expected counts are small. If posterior predictive summaries show a systematic pattern, first verify coding, outcome definitions, dose labels, and the calculation of expected values. Then consider whether the model structure or the available information can plausibly account for the pattern. Do not use convergence remedies to address a lack of fit: they answer different questions.

Posterior Dependence and Remedies Matched to the Problem

Joint models often have correlated posterior parameters. For example, a steeper efficacy dose slope and a shifted efficacy intercept can produce similar predictions over the observed dose range. Dependence parameters can also be associated with marginal dose-response parameters. Correlation between parameter draws is not the same thing as association between the efficacy and toxicity outcomes: the former describes uncertainty about model parameters, while the latter is a feature represented by the joint outcome distribution.

Worked Example: Interpreting Correlated Parameter Draws

Suppose posterior draws for an efficacy slope and a toxicity slope have standard deviations 0.15 and 0.13, respectively, and covariance \(-0.018\). Their posterior correlation is \[ \operatorname{Corr}(\beta_E,\beta_T\mid\text{data}) =\frac{-0.018}{(0.15)(0.13)} =\frac{-0.018}{0.0195} =-0.923. \]

The draws show a strong negative dependence: draws with a larger efficacy slope tend to have a smaller toxicity slope. This may reflect a weakly constrained parameter combination, or a genuine feature of the posterior geometry; the correlation alone cannot decide which. Inspect joint scatterplots, trace plots, marginal intervals, and sensitivity to plausible prior choices. If chains explore the same elongated region adequately, high posterior correlation is not itself a convergence failure. Centering dose, using a better-scaled or otherwise suitable parameterization, or simplifying unsupported model complexity may improve computation; more informative data may be needed to improve identification.

A practical comparison of remedies is useful. For chain separation or poor mixing, investigate initialization and sampler diagnostics, then address parameter scaling, parameterization, or model geometry before extending runs. For weak identification, assess dose and outcome information, simplify unsupported parameters, and report prior sensitivity rather than presenting prior-driven precision as data-driven evidence. For patterned residuals, verify the data and inspect the model's ability to represent the relevant outcome pattern; changing sampler settings will not repair systematic prediction discrepancies. These remedies are not interchangeable.

A common mistake is to say, “The chains converged, so the joint model fits and its parameters are identified.” This reasoning is wrong: chains can accurately sample a broad, weakly identified posterior, and stable computation does not guarantee predictions compatible with observed data. The correction is to report the three assessments separately: computational adequacy, information about parameters, and predictive fit. Likewise, a narrow posterior interval should not be described as proof of identification unless the likelihood itself meaningfully distinguishes the parameter values.

Key takeaway: Diagnose the failure before choosing a remedy. Use chain diagnostics for computational exploration, likelihood and sensitivity assessments for identification, and outcome-level residual or predictive checks for fit. A reassuring result in one category does not substitute for evidence in the others.

Check Your Understanding

Answer each question, then compare with the concise solutions below.

  1. If all participants receive the same coded dose \(x=0\) in a logistic dose-response model, which coefficient is not informed by the efficacy likelihood, and why?
  2. What does a high posterior correlation between efficacy and toxicity slope draws describe? Does it establish strong association between the clinical outcomes?
  3. A joint outcome cell has observed count 15 and fitted expected count 10. Calculate its Pearson residual and state its direction.
  4. Suppose chains mix well, but observed joint outcome counts show a repeated dose-specific residual pattern. Which diagnostic question remains concerning, and why will simply running more iterations not necessarily fix it?

Solutions. (1) The dose slope is not informed because \(x=0\) makes the linear predictor \(\alpha+\beta x=\alpha\), so the likelihood does not contain \(\beta\). (2) It describes dependence between posterior parameter draws; it does not, by itself, establish the efficacy-toxicity association in outcomes. (3) \(r=(15-10)/\sqrt{10}=5/3.162=1.581\), a positive residual indicating more observed cases than expected. (4) Predictive fit remains concerning. More iterations may improve exploration of the same posterior but do not necessarily change the model's fitted predictions or remove a systematic residual pattern.

Clinical Biostats