Data Curation and Synthetic Data › Trajectory Quality and Diversity
World Models HandbookAll chapters Data Curation and Synthetic Data › Trajectory Quality and DiversityOne section · Sections in this chapter2 / 9 ↓
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
A world model trains on trajectories , where is the observation at time , is the action at time , is the trajectory length, and each adjacent triple for is a transition. Two important properties of a trajectory collection are how trustworthy each trajectory is and how broadly the collection covers the situations the model will be asked about. They are distinct rather than strictly orthogonal: collection choices can affect both. A pristine recording of a robot waving its arm in place for an hour may have high signal quality and little behavioral diversity. A varied hour containing recoveries and falls may have broad behavioral coverage even if some transitions are corrupted. Dataset size alone therefore cannot summarize either quality or coverage.
What "quality" means for a world model
Quality is not a scalar property of a file. It is a set of separate failure modes, and each one demands a different detector. It helps to think of each failure mode as a specific corruption process acting on an otherwise clean signal, because then the question "is this defect present?" becomes "how would I detect this specific corruption?", which is a much more tractable engineering problem:
-
Signal-to-noise ratio. Sensor noise, motion blur, compression artifacts, and dropped frames can corrupt observations. Without a model that separates observation noise from process noise, the learned transition distribution may attribute measurement variability to the environment. That misattribution can distort uncertainty estimates used by a planner; the direction of the control error depends on how the planner uses uncertainty, as discussed in Stochasticity and Uncertainty.
-
Temporal alignment. The action paired with must match the timing convention used by the model. Actuation latency, clock drift between the control loop and the camera, and buffered logging can break this correspondence. The data may still look plausible frame by frame while the fitted model learns a delayed or smeared action effect. Detecting the defect therefore requires checking action-observation timing rather than individual frames alone.
-
Action fidelity. Commanded actions, measured joint torques, and realized end-effector velocities are different variables. Training with one action representation and querying the model with another creates a systematic interface mismatch. More data expressed through the same mismatched interface does not identify the missing mapping; calibration, paired measurements, or an explicit actuator model can.
-
Episode completeness. Truncated episodes can create false terminal targets if the training pipeline treats the final recorded frame as a genuine terminal state, pads it as stationary, or otherwise ignores censoring. If truncation correlates with failures, an unmodeled termination indicator can also leak logging behavior into the learned dynamics.
-
Label and state-estimate correctness. If states come from a filter or tracker, estimation failures can be absorbed into the learned dynamics unless uncertainty or invalidity is represented. A simulator can expose its internal state directly, avoiding a tracker in simulator coordinates, although those states and the simulated physics can still differ from reality.
The uncomfortable part is that quality is model-relative and task-relative. Repeated noisy observations can help a stochastic model identify a deployment-relevant noise process, but only when the noise is represented appropriately and can be distinguished from latent dynamics. Aggressive, jerky actions may be irrelevant to one task and essential for learning actuator saturation in another. Before you write a filter, name the downstream consumer, the decisions it will make, and which quality dimensions affect those decisions. Filtering removes information as well as corruption, so retain the raw, provenance-linked source whenever policy permits.
Diversity, coverage, and the difference between them
It is tempting to reduce diversity to a single number and optimize it. Don't. Two distinct quantities matter:
- Coverage asks: which regions of the reachable state-action space have support in the data? A query outside that support requires extrapolation, where accuracy is not constrained directly by observations.
- Density asks: how many samples fall in each region under a chosen representation and resolution? Density affects estimation error within a region, but sample count alone does not guarantee a good fit or calibrated uncertainty.
Coverage and density problems can produce different symptoms, but neither has a unique signature. Out-of-support queries may yield implausible rollouts or deceptively smooth ones, while sparse in-support regions may show high error without an explicit rise in predictive variance. Nearest-neighbor distances, histograms, and kernel-density estimates in a task-relevant embedding are diagnostics rather than proofs: their conclusions inherit the representation and distance metric.
The plot below puts the two quantities side by side on the raw dataset. Density is the number of points packed into a region; coverage is whether a region has any points at all. The narrow cloud saturates the nearby region, while the wide cloud reaches far from the origin but is thin.

The separation matters because the two problems suggest different interventions. More data in a covered region can increase density but does not extend empirical support. A simulator or generator can place samples in a previously empty region, but those samples add useful evidence only to the extent that the generator is faithful there.
For a dataset , where is the -th trajectory, a practical coverage diagnostic is the distribution of distances from each held-out query point to its nearest training state. If most held-out states have a nearest neighbor within and a few are far away, you have a localized hole. If the whole distribution shifts outward, you have a global mismatch.
Useful diversity summaries include:
- State-space effective dimension. Fit a PCA or an autoencoder to all observed states and look at how quickly the explained-variance curve saturates. A curve that saturates at two components while the environment is six-dimensional means your data is effectively planar.
- Action spread. For discrete actions, empirical entropy can summarize concentration. For continuous actions, differential entropy is coordinate- and scale-dependent, so pair it with units, marginals, or coverage over a discretized action grid. A concentrated marginal action distribution can still encode many state-dependent decisions; inspect when that distinction matters.
- Regime coverage. A hand-labeled or clustered partition of episodes into qualitatively different situations (walking, turning, recovering, standing still) and the number of episodes per regime.
- Trajectory-shape diversity. The distribution of episode-level summaries such as duration, total displacement, energy, and contact count. Datasets collected by a single scripted controller often have a suspiciously narrow shape distribution.
A frequent pathology is concentration of the behavior policy. If 95% of the data was collected by one scripted controller, the resulting state-action distribution may occupy a narrow region. A learned dynamics model can be accurate there while remaining unconstrained elsewhere. This is the collection-side failure discussed in Interaction Data and Passive Observation, restated as a metric that can be tracked as collection policies change.
There is a persistent tension between collecting data that is typical (representative of what the deployed agent will experience) and data that is informative (unusual enough to constrain the dynamics). A dataset pulled entirely from the deployment distribution is easy to fit but teaches the model nothing about the boundaries of its validity. A dataset pulled entirely from rare events is informative but leaves the routine regime underdense. Good mixtures deliberately contain both, and the mixture weight between them is a knob you tune, not a constant you inherit.