Tutorials › High-Dimensional Genomic Data Analysis › Defining the Genomic Prediction Target

Honest Evaluation and Leakage · Tutorial 626 of 795

Defining the Genomic Prediction Target

Learn to state a genomic prediction target precisely and understand which population-level performance a validation estimate can support.

Intermediate 10 min read

What You'll Learn

  • Specify the target population and prediction occasion for a genomic model.
  • Distinguish performance for new individuals in a cohort from performance at a new site.
  • Define prediction risk using a target distribution and a stated loss function.
  • Explain when an average validation loss estimates the intended risk.
  • Recognize how subgroup composition and outcome prevalence affect reported performance.
  • Diagnose mismatches between the validation sample and the intended deployment population.

Why the Prediction Target Comes First

A genomic prediction model can perform well for one intended use and poorly for another. A model evaluated on held-out individuals from the same cohort may be useful for predicting additional individuals recruited under similar conditions. That result does not automatically tell us how well the model will predict patients at a different hospital, in a later period, or under a different assay protocol. The populations and prediction circumstances differ, so the performance target differs.

The prediction target is not merely the name of an outcome. It specifies who or what will receive predictions, when predictions are made, what information is available then, and how prediction quality will be measured. In genomic applications, these choices matter because populations, clinical practice, genotyping platforms, and outcome definitions can vary.

Definition: A prediction target is the specified population of prediction occasions, the information available at each occasion, the outcome to be predicted over a stated period or condition, and the loss or performance measure used to judge predictions. The target population can be described by a probability distribution \(P^*\) over the relevant predictors and outcomes, or by a finite population with a defined sampling scheme.

Risk Depends on the Target Distribution

Let \(X\) denote the predictors available at the time of prediction, including genomic measurements when appropriate, and let \(Y\) be the outcome defined for the prediction task. Let \(f\) be a fitted prediction rule and \(L(Y,f(X))\) a loss, where smaller values mean better predictions. For example, squared-error loss is \(L(Y,f(X))=(Y-f(X))^2\); misclassification loss is 1 when the predicted class is wrong and 0 otherwise.

For a fixed fitted rule, its target risk is the expected loss for prediction occasions drawn from the target population. The distribution \(P^*\) must describe the population and circumstances specified by the prediction target.

$$ R(P^*,f)=E_{(X,Y)\sim P^*}\left[L\{Y,f(X)\}\right]. $$

The risk is conditional on the fitted rule \(f\). If the goal is instead to describe the performance of a training procedure, which can produce different fitted rules from different training samples \(D\), the target may average over those training samples as well:

$$ R_{\mathrm{procedure}}(P^*)=E_D\left[R\{P^*,A(D)\}\right], $$

where \(A\) is the specified procedure that maps training data to a fitted prediction rule. These are different quantities: one concerns a particular fitted model, and the other concerns the expected performance of the procedure. Either can be relevant, but the intended quantity should be stated.

Key takeaway: A validation score is meaningful only in relation to its target distribution, fitted model or training procedure, and loss measure. A score is not a universal property of a model.

When Does Validation Estimate the Intended Risk?

Suppose a fitted rule \(f\) is fixed before evaluation, and a validation sample \((X_i,Y_i)\), \(i=1,\ldots,m\), consists of independent draws from \(P^*\). If the loss has a finite expectation, the average validation loss estimates the target risk:

$$ \widehat R_{\mathrm{val}}=\frac{1}{m}\sum_{i=1}^{m}L\{Y_i,f(X_i)\}. $$

The justification is direct. Each validation loss has expectation \(R(P^*,f)\), so linearity of expectation gives:

$$ E(\widehat R_{\mathrm{val}}\mid f) =\frac{1}{m}\sum_{i=1}^{m}E\left[L\{Y_i,f(X_i)\}\mid f\right] =\frac{1}{m}\sum_{i=1}^{m}R(P^*,f) =R(P^*,f). $$

This result does not say that every validation sample will produce the exact risk. The estimate varies from sample to sample. It also depends on the assumptions: the validation observations must represent the target distribution, the fitted rule must not have been adapted using their outcomes, and the recorded predictors and outcomes must correspond to the stated prediction task. If validation observations are selected in a way that changes their distribution, an unweighted average generally estimates performance in that selected sample rather than in \(P^*\).

Worked Example: New Individuals From the Same Cohort

Suppose a hypothetical study fits a model to some patients from one cohort and reserves four other patients from that same cohort for evaluation. The target is new patients recruited under the same cohort conditions, and the outcome is a laboratory measurement in mg/dL. Assume the four validation patients were not used to fit or choose the model, and that they represent the intended cohort population. Their prediction errors, in mg/dL, are \(1,-2,0,3\).

For squared-error loss, the observed errors squared are \(1,4,0,9\) in \((\mathrm{mg/dL})^2\). Thus:

$$ \widehat R_{\mathrm{val}} =\frac{1^2+(-2)^2+0^2+3^2}{4} =\frac{14}{4} =3.50\;(\mathrm{mg/dL})^2. $$

The validation RMSE is \(\sqrt{3.50}=1.871\) mg/dL. Under the stated representativeness and independence assumptions, the mean squared error estimates the model's squared-error risk for new patients from this cohort. The RMSE summarizes error on the outcome's original scale. Neither figure, by itself, establishes performance for patients at a different site.

Same Cohort and New Site Are Different Targets

The phrase “generalize to new patients” is incomplete unless it says what those patients are like. A same-cohort target treats future patients as drawn from the cohort's population of prediction occasions. A new-site target concerns patients from a site not represented in the same way during model fitting or evaluation. Differences between sites can include patient mix, clinical practice, outcome measurement, and genomic assay procedures. These differences can change the distribution of prediction errors.

For a target that draws sites and then patients within sites, overall risk can be written as an average of site-specific risks. If site \(s\) has target weight \(w_s\), with nonnegative weights summing to one, then:

$$ R(P^*,f)=\sum_s w_s R_s(f), \qquad \sum_s w_s=1, $$

where \(R_s(f)\) is the expected loss for a patient at site \(s\). The weights encode the target: for example, they could reflect the proportions of patients expected from each site, or assign equal importance to each site. Those choices need not produce the same risk. Data from one site cannot, on their own, establish performance for a population of future sites.

Worked Example: Patient-Weighted Versus Site-Weighted Risk

Consider hypothetical squared prediction errors, in \((\mathrm{mg/dL})^2\), from two sites. Four validation patients at Site A have losses \(1,1,4,4\), giving a mean loss of \((1+1+4+4)/4=2.50\). Two patients at Site B have losses \(9,16\), giving a mean loss of \((9+16)/2=12.5\).

If the target is a patient chosen from these six patients' site mixture, the pooled estimate weights patients equally:

$$ \widehat R_{\mathrm{patient}} =\frac{1+1+4+4+9+16}{6} =\frac{35}{6} =5.833\;(\mathrm{mg/dL})^2. $$

If the target instead gives the two sites equal importance, average the two site-specific means:

$$ \widehat R_{\mathrm{equal\ site}} =\frac{2.50+12.5}{2} =7.50\;(\mathrm{mg/dL})^2. $$

These values differ because the targets weight sites differently. The calculations do not establish how well the model will perform at an arbitrary new site: the two hypothetical sites may not represent the distribution of future sites. The target must determine the weighting, and the evaluation data must support that target.

Population Composition and the Performance Measure

Even when the model, outcome, and loss definition are unchanged, changing the mix of individuals can change average performance. This is especially important for classification. The fraction of individuals with each outcome class affects overall misclassification error, and measures such as sensitivity and specificity answer different questions from overall error. The measure must therefore be chosen to match the intended use, rather than selected after seeing which one looks most favorable.

Worked Example: A Different Disease Prevalence Changes Error

Suppose a hypothetical classifier has a 5% misclassification rate in a lower-risk group and a 30% rate in a higher-risk group. Assume these rates remain applicable to the target subgroups. In a target population that is 80% lower-risk and 20% higher-risk, the expected misclassification rate is:

$$ 0.80(0.05)+0.20(0.30)=0.04+0.06=0.10. $$

The target error rate is 10%, corresponding to 90% accuracy under zero-one loss. If a validation sample instead contains equal numbers from the two groups, its unweighted error estimate is \(0.50(0.05)+0.50(0.30)=0.175\), or 17.5%. Neither calculation is arithmetically wrong: they summarize performance under different group mixtures. The first is relevant to the specified 80%/20% target, while the second corresponds to an equal-mixture target. In practice, subgroup rates are themselves estimated with uncertainty, and weighting does not fix a model whose subgroup performance changes in the target setting.

State the Target Before Choosing an Evaluation Design

A useful target statement is concrete enough that another analyst could determine whether a candidate validation sample is relevant. It should identify:

  • Prediction unit and population: for example, patients newly presenting to specified clinics, rather than samples or repeated records treated as independent people.
  • Prediction time and available information: predictors must be available when the prediction would actually be made. A variable recorded only after the outcome is known is not a valid predictor for that use.
  • Outcome and horizon: define what is being predicted and when or how it is ascertained.
  • Performance measure: specify the loss or summary, including any clinically meaningful weighting.
  • Population mixture: state whether patients, sites, or subgroups receive weights proportional to their expected frequency or by another justified rule.

These statements distinguish targets that are easy to blur together. For instance, “predict disease status using genomic data” leaves out the intended patients, the time of prediction, the disease definition, and the measure of success. A more useful description might specify newly presenting patients at a defined set of clinics, genomic and clinical predictors measured at the initial visit, disease status assessed at a stated follow-up, and patient-weighted misclassification loss.

Diagnostics, Alternatives, and Limitations

Before interpreting validation performance, compare the validation sample with the target description. Check whether relevant populations, sites, calendar periods, outcome definitions, and measurement procedures match. Examine whether the validation set includes the kinds of individuals expected at deployment and whether important subgroups are represented. Large discrepancies in predictor distributions or outcome frequencies are warnings that an unweighted validation average may not estimate the stated target risk. Similar observed distributions are reassuring but do not prove that all relevant differences have been ruled out.

There is no universally best evaluation population. A random held-out sample from the same cohort is a natural choice for a same-cohort target. Evaluation at a separate site is more directly relevant to a target involving that site or similar sites, but a single external site may not represent all future sites. A time-separated evaluation can be relevant when the intended use is in a later period, while changes in practice or measurement can still limit transfer. These are alternatives for answering different questions, not interchangeable tests of model quality.

If the validation sample and target population differ, weighting can sometimes align the average with a specified target mixture when subgroup risks can be estimated and the needed groups are observed. It cannot supply evidence for populations absent from the data, guarantee that conditional prediction performance is unchanged, or correct a changed outcome definition. Transporting a performance estimate therefore requires assumptions about how the validation and target populations relate, as well as adequate data to assess those assumptions.

A common failure is to report a single score as “the model's accuracy” without saying which population it describes. Another is to define the target broadly as “future patients” while evaluating only one cohort, or to count multiple records from the same individual as if they represented independent future individuals. The correction is to define the prediction occasions and target distribution first, then ask whether the evaluation data and its analysis support an estimate for that target. The exact validation split or resampling unit is a separate design decision; it cannot repair an unclear target.

Key takeaway: Define the prediction target before interpreting validation results. The target fixes the population, available predictors, outcome, weighting, and performance measure; validation estimates only the performance supported by those choices and the evaluation data.

Check Your Understanding

Answer each question by identifying the target and the implication for performance interpretation.

  1. A model is evaluated on held-out patients from the same cohort in which it was trained. What target does this most directly address, and what does it not establish?
  2. Two sites have mean squared errors of 4 and 16 \((\mathrm{mg/dL})^2\). If the target gives Site 1 weight 0.75 and Site 2 weight 0.25, calculate the target risk estimate.
  3. A validation set is 50% high-risk patients, but the intended population is 20% high-risk. Why might the unweighted error rate fail to estimate target-population error?
  4. Why must predictors be restricted to information available at the intended prediction time?
  5. A model has been selected after repeatedly comparing its performance on the same validation outcomes. Does the simple unbiasedness justification for the average validation loss still apply? Explain.

Solutions. (1) It most directly addresses new individuals from a population represented by that cohort under similar conditions; it does not establish performance at a new site or in a different population. (2) The estimate is \(0.75(4)+0.25(16)=3+4=7.00\;(\mathrm{mg/dL})^2\). (3) The validation mixture gives the high-risk group more weight than the target does, and if subgroup error rates differ, the two weighted averages differ. (4) Otherwise the evaluated rule uses information unavailable when real predictions must be made, so it evaluates a different, potentially infeasible task. (5) No. The fitted or selected rule depends on the validation outcomes, so validation losses are not simply independent observations evaluated by a fixed rule; the stated conditional expectation argument does not apply as written.

Clinical Biostats