Datasets, Benchmarks, and Experimental Design › Ablations, Baselines, and Statistical Power
World Models HandbookAll chapters Datasets, Benchmarks, and Experimental Design › Ablations, Baselines, and Statistical PowerOne section · Sections in this chapter4 / 8 ↓
Reading tools Definitions & inline links
Choose your expertise level to adjust how many terms are explained. Beginners see more tooltips, experts see fewer to maintain reading flow. Hover over underlined terms for instant definitions.
Article links
Make inline references clickable
We now have a training set and the right held-out unit. The remaining questions are: what should you compare against, what does an ablation establish, and how many independent replicates do you need before a difference means anything?
What each baseline compares
A baseline defines a comparison, not automatically an isolated causal contribution. Specify its inputs, fitting and tuning procedure, and resource scope before interpreting the contrast.
- Simple dynamics baseline. Predict the next state from a scaled last action or fit a low-order linear model. The contrast measures improvement over that specified predictor; it does not separate capacity, nonlinearity, optimization, and feature effects unless those are controlled.
- Action-agnostic predictor. Fit a matched predictor that omits actions. Its contrast measures the incremental predictive value of supplied action information under the protocol. It does not by itself establish how an already-trained conditioned model uses actions. Controlled action changes or interventions address that different question, with support and identification caveats.
- Well-tuned feedback or model-free comparator. Compare control systems under a declared interaction budget. Architecture, optimization, and representation may still differ, so this is a whole-system comparison. Matched component ablations or controlled model substitutions can support more specific contribution claims; identified observational analysis or a justified structural derivation can provide other routes.
- True-dynamics oracle. Give a planner the exact simulator as a privileged diagnostic. Its achieved return is not generally an upper bound when search is approximate, the horizon is short, or information access differs. A solved optimum under the same constraints is a different object and can provide a bound when its assumptions are satisfied.
Giving a planner true state or true dynamics changes its information access and must be disclosed. It can help diagnose model-versus-search limitations, but the diagnosis is conditional on the planner and evaluation protocol. For example, a one-step greedy planner can take an immediate reward of 1 instead of investing now for reward 10 on the next step, even with exact dynamics. Suppose the model also predicts rewards and incorrectly assigns the investment an immediate reward of 2 instead of its true reward of 0. The biased reward prediction induces investment and an actual two-step return of 10. Dynamics bias alone cannot change this one-step ranking if the current-state reward is known exactly. The true-model planner's realized two-step return of 1 is not an upper bound on that result; the true constrained optimum is a separate calculation. Do not present privileged access unavailable at deployment as an ordinary deployable competitor.
A useful heuristic when reading your own results draft: for each row of a results table, ask what a skeptical reader would say if this baseline were absent. If the claim depends on that comparison, keep the baseline in the main table. If removing it leaves the claim unchanged, the comparison can move to an appendix. This keeps the main table focused on comparisons that constrain the claim.
Comparing fairly
Two questions get conflated under the heading of "fair comparison":
- Declared resource caps. Each method gets the specified limits relevant to the question, such as environment interactions or training wall-clock time. This asks what each attained under those caps. Equal gradient-step counts do not imply equal compute when batch sizes or per-step work differ.
- Reasonably tuned performance. Each method is reported at the best validated configuration found within a declared search procedure. This asks what that search attained, not what the method could achieve under every possible training and tuning procedure.
Both questions are legitimate. Matching parameter counts alone does not establish equal training compute, but it can answer a parameter-count-constrained question. For weight-memory constraints, also declare parameter precision and storage. For total deployment-memory constraints, measure or bound runtime state, caches, activations and buffers under the stated input and batch scope. Disclose training cost separately. You need not equalize every resource simultaneously; name the constraint being compared. Unequal tuning effort is part of the attained system contrast, not evidence of an isolated algorithmic effect.
The minimum disclosure is: what was searched, over what space, how it was selected, and how many configurations were tried for each method. Searching 200 configurations consumes compute and can make the winning validation estimate optimistic. A final test independent of that selection need not inherit the validation winner's bias. Report the search and preserve the fitting, selection, and testing boundaries.
One implication that papers sometimes miss: if you tuned method A over 200 configurations and method B over 20, then even with equal compute budgets the comparison is asymmetric in search effort. The honest fixes are to give both methods search budgets of the same order, to report the best-config performance and the default-config performance side by side, or to explicitly declare that the comparison is between one method's tuned configuration and another method's default configuration. Any of those is fine; silently mixing them is not.
Ablations: removal versus retraining
The word "ablation" covers two experiments that answer different questions.
- Removal ablation. Take a trained model and delete or mask a component at inference. This answers: how much does the trained system rely on this component?
- Retrained ablation. Train without the component from scratch using a declared training and tuning procedure. This estimates its contribution to the performance achieved by that procedure, not everything either architecture could learn under all procedures.
These can disagree. A component can help optimization during training yet be dispensable at inference. Conversely, a trained system can rely on a component while a separately retrained system compensates for its absence. Label the intervention and training procedure clearly; these possibilities do not establish a frequency of disagreement.
A second issue is confounding. If removing a component also changes the parameter count, training budget, or optimization difficulty, the ablation measures that combination. A one-factor-at-a-time design that leaves combinations unobserved cannot generally identify their interaction without additional assumptions. For two binary components, evaluating all four combinations is a simple factorial design for estimating the interaction as a difference of differences. Replication is still needed to estimate uncertainty. Additional combinations and replicates consume runs; state which interaction the design can identify rather than assuming every ablation reveals a universal mechanism.
A third issue is identifiability. The effect of a component depends on the rest of the system. An ablation does not automatically identify a universal causal mechanism; it identifies the role of that component in this architecture, at this scale, on this data. Writing "the recurrence is essential for memory" after one ablation is a much stronger claim than the experiment supports. A more defensible phrasing would be: "In our architecture, at this scale, on this suite, removing recurrence at inference degraded held-out error by X, and retraining without recurrence degraded it by Y." The narrower statement travels well; the broader one does not.
Finally, evaluating all subsets of six binary components gives candidate configurations and consumes a search budget. Selecting the largest noisy selection-set estimate can make that estimate optimistic; the candidate count alone does not determine the magnitude of the bias. A final test independent of that selection need not inherit it. Keep the final test set separate, and report the full search, including candidates other than the winner.
Independent replication and pairing
Consider an illustrative experiment with:
- 12 independent training runs per method,
- 30 evaluation episodes per run,
- 200 transitions per episode,
- 20 tasks.
How many independent replicates do you have? It depends entirely on what you are claiming.
| Level | What it is | Independent replicate? | Illustrative count range, not a recommendation |
|---|---|---|---|
| Transition | one row | Not a new training replicate; observation dependence depends on collection | to |
| Evaluation episode | one rollout under a fixed model and policy | Conditional on the fixed model and policy, if episodes are independently sampled; identical distributions are an additional assumption for a common-population analysis | 10 to |
| Planner stream | one sequence of decisions with a fixed seed | Only under an independently generated stream design; cross-method matching is a separate choice | 1 to 100 |
| Task | one environment or task instance | For independent-replicate population inference, justify independent task or task-cluster sampling; dependence-aware designs need their own sampling and variance assumptions | 1 to 100 |
| Training run | one fit from initialization to final checkpoint | Can be independent conditional on fixed data and independent randomization; not automatic | 3 to 20 |
Ten thousand transitions evaluated under one learned model are not ten thousand training replicates. They provide within-run observations, with a dependence structure determined by their collection. A transition-level interval cannot substitute for an interval over training procedures. It omits run variability if that variability is present; whether the omission dominates precision is an empirical question. The illustrative ranges in the table are not literature frequency estimates or required seed counts.
There are two kinds of uncertainty, and they are not interchangeable:
- Conditional uncertainty. Given a fixed fitted model, how variable are the evaluation outcomes? Independently sampled evaluation episodes estimate that variation. More such episodes can reduce Monte Carlo error for the conditional mean under finite-variance assumptions, but their cost depends on the environment and protocol.
- Procedure-level uncertainty. How variable is the outcome across the sources randomized in the target procedure? State whether the dataset, initialization, minibatch order, and model-selection process are fixed or varied. Independent fits conditional on one dataset estimate a different uncertainty from fresh-data fits that rerun selection.
If independent training runs are evaluated on identical fixed cases, inference on the run-level differences is conditional on those cases unless the analysis also models case sampling. To target fresh cases as well as fresh runs, describe the case population and sampling hierarchy. Shared cases create a crossed dependence structure rather than separate nested cases for every run. Accounting for case sampling can add a variance component, but does not guarantee that every estimated interval is wider than its fixed-case counterpart.
The paired-difference variance identity
Pairing can reduce variance in a comparison when matched outcomes have positive covariance. The variance identity states that condition and also shows when pairing can hurt.
Let the experimental unit be the independent replicate, and suppose for replicate you observe under method A and under method B, on the same cases. Define the paired difference
where:
- : the index over the independent replicate (for example, the training run)
- : the metric for replicate under method A
- : the metric for the same replicate under method B, evaluated on the same matched cases
For an error metric, positive means method A has larger error. For a return metric, positive instead means A has larger return; the algebra does not decide which direction is desirable. Write for the average of replicate differences. Variance measures how outcomes vary across repetitions; covariance measures the shared movement of two outcomes. If the are independent and identically distributed with finite variance , then
Uncorrelated equal-variance differences also suffice for this variance identity. Independence is a stronger assumption used in the subsequent interval analysis; the variance identity alone does not establish its distributional validity.
To expand in terms of the individual outcomes, assume both and have finite second moments. Finite variance of their difference alone does not ensure that the marginal variances and covariance exist. Under the marginal-moment condition,
so, retaining the preceding assumptions for the replicate differences and these finite marginal second moments,
Three consequences follow.
- Positive covariance helps. At the same and marginal outcome variances, positive covariance makes the variance of the paired mean difference smaller than an independent-sample comparison. Matching can produce this benefit, but does not establish the covariance's sign or size. Estimated interval widths also depend on the interval procedure and degrees of freedom.
- Negative covariance hurts. If the methods trade off, for example when one succeeds precisely where the other fails, pairing can be worse than not pairing.
- Matching must be defined. A seed number alone does not guarantee corresponding draws across different implementations. Feedback policies can follow different state-action paths while still sharing initial conditions, task difficulty, or exogenous disturbances. Path divergence does not force outcome covariance to zero. Measure the covariance or the paired-difference variability rather than assuming the benefit vanishes. Valid interval inference additionally needs the stated independence and distributional assumptions across replicate pairs.
Pairing removes shared outcome variation, which can include variation caused by shared training-data draws or coupled initialization. If and , the shared cancels in even if it came from training. Unshared variation remains. More independent runs are needed to resolve that residual uncertainty when it is large; matched evaluation cases alone do not guarantee a small residual.
Executable comparison: run-level paired differences
We now run the second experiment. The design is deliberately structured the way a real comparison should be:
- 12 independent runs. Each run simulates a fresh cohort with its own RNG stream, so the intercepts, noise, and actions differ.
- Within each run, a session-grouped split gives 28 training sessions and 12 held-out sessions.
- Both models are fit on the same training rows.
- Both models are evaluated on exactly the same held-out rows, which is what makes the comparison paired.
- The run-level metric is the mean absolute error over those rows, so each run contributes one number per method.
- We average matched cases within a run first, then treat the run-level differences as the replicates.
The two methods are deliberately from different families: an affine model using only the current state and action, and a k-NN model using a three-step history window. Their disagreement is what we will measure.
N_RUNS = 12
def run_comparison(seed, k=8):
"""One independent run: fresh cohort, fresh split, matched evaluation rows."""
cohort = simulate_cohort(seed=seed)
Xr, yr, sr = make_transitions(cohort)
fit_i, held_i, _, _ = session_grouped_split(sr, frac=0.7, seed=seed + 1)
X_train, y_train = Xr[fit_i], yr[fit_i]
X_eval, y_eval = Xr[held_i], yr[held_i]
mae_affine = mean_absolute_error(
y_eval, affine_fit_predict(X_train, y_train, X_eval)
)
mae_knn = mean_absolute_error(
y_eval, knn_predict(X_train, y_train, X_eval, k=k)
)
return mae_affine, mae_knn
run_results = np.array(
[run_comparison(seed=1000 + 7 * r) for r in range(N_RUNS)]
)
run_ids = np.arange(1, N_RUNS + 1)
diffs = run_results[:, 1] - run_results[:, 0] # k-NN window minus affineNow we summarize with a run-level paired Student- interval and a percentile bootstrap. The interval assumes the are approximately normal and i.i.d.; with 12 runs those assumptions are doing real work. Under the exact iid Gaussian model with positive population variance, a 95% procedure would contain the population mean difference in 95% of repeated experiments of this design. That is coverage of the mean, not a claim that 95% of future run differences lie in this interval, nor a posterior probability for the realized interval. For this training-procedure target the bootstrap resamples whole independent runs, not flattened transitions or episodes.
mean_d = float(diffs.mean())
sd_d = float(diffs.std(ddof=1))
se_d = sd_d / np.sqrt(N_RUNS)
t_crit = float(stats.t.ppf(0.975, df=N_RUNS - 1))
ci_t = (mean_d - t_crit * se_d, mean_d + t_crit * se_d)
boot_rng = np.random.default_rng(7)
boot_means = np.array(
[
diffs[boot_rng.integers(0, N_RUNS, size=N_RUNS)].mean()
for _ in range(10_000)
]
)
ci_boot = tuple(float(v) for v in np.percentile(boot_means, [2.5, 97.5]))
standardized = mean_d / sd_d
run_sd_a = float(run_results[:, 0].std(ddof=1))
run_sd_b = float(run_results[:, 1].std(ddof=1))Per-run MAE (affine, k-NN window, difference): run 1 0.0563 0.0980 +0.0417 run 2 0.0718 0.1389 +0.0671 run 3 0.0764 0.1654 +0.0890 run 4 0.0653 0.1135 +0.0482 run 5 0.0692 0.1142 +0.0450 run 6 0.0577 0.1230 +0.0653 run 7 0.0573 0.1175 +0.0602 run 8 0.0659 0.1108 +0.0449 run 9 0.0515 0.1002 +0.0487 run 10 0.0555 0.1573 +0.1018 run 11 0.0846 0.1279 +0.0434 run 12 0.0689 0.1627 +0.0938 Run-level mean difference (k-NN minus affine): +0.0624 Run-level SD of differences: 0.0215 95% paired t interval (11 df): [+0.0488, +0.0761] 95% percentile bootstrap over runs: [+0.0513, +0.0746] Standardized effect size (mean / SD): +2.91 Run-level SD of MAE, affine: 0.0098 k-NN window: 0.0235 Interval contains zero: False
Read the printed interval rather than assuming an outcome. If the interval excludes zero, the data favor one method under this estimand, these assumptions, and this number of independent runs. If it contains zero, the comparison has not resolved the difference at this replication level, and that is a statement about the experiment, not evidence that the methods are equivalent. Note also the practical effect size: a difference can be statistically detectable and too small to matter, or practically meaningful and statistically unresolved. The printed mean and interval give the statistical result and its size in state units. Judging practical importance still requires an application-specific criterion; the standardized mean-to-SD ratio is not an equivalence margin.
Two cautions about the bootstrap here. Its empirical sampling distribution uses only 12 observed run differences. Many resampled means can be numerically distinct; more resamples reduce Monte Carlo error but supply no new independent information about unobserved population tails. Resampling flattened transitions instead targets the wrong unit for this procedure-level comparison. Ignoring run variation or positive within-run dependence can make that interval too narrow; the size of the error depends on the variance components.
The run-level differences are the replicates. Two separate figures preserve distinct quantities: absolute MAE for each model, then the within-run MAE difference and its uncertainty. A confidence interval for a difference must not be overlaid on an absolute-error axis as though it were an interval for either model's MAE.


Seeds help repeatability when the relevant random streams and other nondeterminism are controlled. A seed alone is not a characterization of variation across draws. Independent runs or a justified stochastic model are needed for that uncertainty. The 12 fits in this executable example are a demonstration choice, not a standard or an assurance of adequate power. A borderline interval with few replicates warrants an unresolved conclusion rather than a categorical presence-or-absence claim.
Choosing a replication level prospectively
The right time to decide how many runs you need is before you run them. For a two-sided paired comparison, a standard approximate calculation is
where is the Type I error rate, the desired power, the smallest difference worth detecting, and the assumed standard deviation of the paired differences. The symbol is the quantile of a standard normal distribution. At and power , the two quantiles are approximately and . This is a normal approximation for planning. It is not an exact small-sample guarantee, it is not post-hoc observed power, and it is not a universal seed recommendation.
We can drive it with the pilot estimated from the 12 runs above, while keeping in mind that a pilot estimate from a small number of runs is itself uncertain.
def paired_normal_n(sigma_d, delta, alpha=0.05, power=0.80):
"""Approximate runs needed for a two-sided paired normal test."""
z_alpha = stats.norm.ppf(1.0 - alpha / 2.0)
z_beta = stats.norm.ppf(power)
return ((z_alpha + z_beta) * sigma_d / delta) ** 2
sigma_pilot = max(sd_d, 1e-9)
delta_multipliers = [0.25, 0.50, 1.00]
delta_values = [m * sigma_pilot for m in delta_multipliers]
sigma_grid = np.linspace(0.25 * sigma_pilot, 2.0 * sigma_pilot, 200)
curves = {}
for m, delta in zip(delta_multipliers, delta_values):
curves[m] = np.array([paired_normal_n(s, delta) for s in sigma_grid])
planning_table = []
for m, delta in zip(delta_multipliers, delta_values):
for factor in [0.5, 1.0, 2.0]:
n_est = paired_normal_n(sigma_pilot * factor, delta)
planning_table.append(
(m, delta, factor * sigma_pilot, int(np.ceil(n_est)))
)Pilot sigma_D from 12 runs: 0.0215 state units
target delta (x pilot SD) assumed sigma_D runs needed (rounded up)
0.25x 0.0107 32
0.25x 0.0215 126
0.25x 0.0430 503
0.50x 0.0107 8
0.50x 0.0215 32
0.50x 0.0430 126
1.00x 0.0107 2
1.00x 0.0215 8
1.00x 0.0430 32
All values are planning approximations, not guarantees.
Halving the target difference roughly quadruples the required runs.The table makes the sensitivity concrete: halving the target difference quadruples the unrounded normal-planning requirement, and doubling the assumed variance doubles it. Integer rounding can alter the exact ratios in the table. Both unrounded relations follow from the squared ratio in the formula. This is why three seeds or five seeds cannot be defended as universal defaults: the right replication budget depends on the variance of the measured quantity and the difference worth detecting, neither of which is a universal constant.
The planning formula is easier to read as a curve. The plot varies the assumed paired standard deviation for three target differences. At fixed target difference, error rate, and power, planned runs grow quadratically with SD and linearly with variance.

Practical guidance that follows:
- Round up, always. Fractional runs do not exist.
- Explore plausible SD values. A larger assumed gives a larger planned at fixed target difference, error rate, and power. This is conservative relative to the pilot value, not a certified lower bound under the unknown population distribution.
- Distinguish exact inference from approximate planning. For iid Gaussian differences with positive population variance and , the Student- interval has exact nominal coverage. Small samples can make robustness to nonnormality poor and make the normal planning formula inaccurate. A noncentral- calculation under the Gaussian model, or simulation under a justified alternative model, can refine planning.
- Check heavy-tailed scenarios. Heavy tails can make finite-sample normal planning inaccurate in either direction. Infinite variance invalidates a formula requiring . Simulate plausible distributions and assess Type I error and power rather than assuming the approximation always underestimates the requirement.
- Separate episodes from runs. More independent evaluation episodes can reduce within-run measurement noise and improve procedure-level precision, but cannot remove a nonzero between-run component. When that component dominates, extra episodes buy little compared with extra independent runs.
Equivalence, practical significance, and multiplicity
Several statistical reflexes are wrong, and each has a specific correct counterpart.
Overlapping confidence intervals do not settle whether a paired difference is zero. Two marginal intervals can overlap substantially while the paired difference is confidently nonzero, because overlap ignores the covariance. Always compute the interval on the difference, not by eyeballing two separate intervals.
Absence of statistical significance is not equivalence. Failing to reject zero leaves a range of effects compatible with the estimate and uncertainty; it neither establishes equivalence nor measures power. Prespecify meaningful equivalence bounds. A symmetric choice is , although asymmetric bounds are possible. Two one-sided tests (TOST), each at level , reject the two boundary nulls when the corresponding two-sided interval lies strictly inside those bounds. At , that is a 90% interval, not the 95% interval used above to assess a difference from zero. A 95% interval wholly inside the same bounds is sufficient but more conservative. State the test, interval level, and practical bounds together. Lakens's equivalence-test primer explains this distinction.
Statistical and practical significance are different questions. With enough runs, a difference of 0.001 state units can be statistically significant. That does not make it worth deploying. Report the effect size in interpretable units alongside the interval.
Multiplicity matters. Ten independent true-null tests, each with actual Type I error probability 0.05, have probability of at least one rejection. Nominal level 0.05 alone does not imply that each test attains that probability; dependence also changes the calculation. Define the confirmatory family and error target. Holm controls familywise error, the probability of at least one false rejection. Benjamini-Hochberg controls false discovery rate under its stated dependence assumptions, an expected proportion among rejected hypotheses. These are different guarantees. Prespecifying one primary comparison limits that primary family but does not automatically control additional claims. Labeling other analyses exploratory makes their status clear; disclosure alone does not change their statistical error probabilities.
Prespecify the primary metric, comparison, split, and stopping rule before confirmatory evaluation. Outcome-informed choices can compromise an unadjusted test interpretation. Not every later action is such a choice: transparently correcting a verified implementation error under a declared protocol is different from selecting the most favorable observed result.