Skip to content

Repository files navigation

WorldBench

Regression testing for video-based robotics world models.

WorldBench compares baseline and candidate checkpoints on the same prediction suite, then surfaces the individual episodes and future horizons that became worse.

Main question: did this new world-model checkpoint improve, and what became worse?

same fixed episodes
  -> baseline predictions vs candidate predictions
  -> per-episode and per-horizon deltas
  -> reproducible PASS or FAIL gate

Tests Python Version License

Demo video: artifacts/demo/worldbench-demo-web.mp4

Flagship Proof

The strongest committed proof compares two NanoWM-B/2 checkpoints on the same fixed 10-episode RT-1 / Fractal suite.

Field Value
Baseline checkpoint knightnemo/nanowm-b2-rt1-abl-pred-v-50k
Candidate checkpoint knightnemo/nanowm-b2-rt1-300k
Fixed episodes 10, IDs 0 through 9
Composite Score mean baseline 85.67, candidate 87.28
Composite Score delta +1.61
Improved / regressed / unchanged episodes 9 / 1 / 0
Gate result strict PASS; engineering-threshold PASS

WorldBench detected aggregate improvement while surfacing the regressed episode: episode_002.mp4 changed by -0.33. This is a fixed-suite validation proof, not a public model ranking and not evidence of universal robot capability.

Artifacts: artifacts/checkpoint_validation/ Method: docs/checkpoint_validation.md, docs/checkpoint_regression.md

One Video Pair

If you have one ground-truth future video and one generated prediction video:

python -m pip install "worldbench[video]"

worldbench eval-videos \
  --ground-truth ground_truth.mp4 \
  --prediction predicted_future.mp4 \
  --output results/

This writes:

results/
  result.json
  summary.md
  artifacts/comparison.png

eval-videos is a beginner-friendly single-prediction evaluation. It is not checkpoint regression by itself.

Try the local synthetic demo:

worldbench eval-videos --demo --output results/demo

Checkpoint Regression

worldbench eval-batch \
  --ground-truth eval_suite \
  --predictions baseline_predictions \
  --name baseline \
  --skip-context 4 \
  --output baseline.json

worldbench eval-batch \
  --ground-truth eval_suite \
  --predictions candidate_predictions \
  --name candidate \
  --skip-context 4 \
  --output candidate.json

worldbench gate \
  --baseline baseline.json \
  --candidate candidate.json \
  --strict-config-match \
  --max-episode-regressions 0

Optional paired bootstrap uncertainty is available in gate with --bootstrap-samples; confidence-bound gating is opt-in so existing gates remain backward compatible.

Example Output

Composite Score: 96.10/100
Metric coverage: 2 of 5 configured metrics
Available configured weight: 45%
Available: Visual Similarity, Temporal Stability
Unavailable: Action Consistency, Object Permanence, Contact Realism

The Composite Score is a weighted summary over available configured metrics. It is not accuracy, not task success, and not a universal robotics capability score.

Compatibility

Directly compatible:

  • aligned predicted future RGB frames or videos
  • action-conditioned robot video predictors
  • image-to-video robot models
  • latent models with deterministic RGB decoders
  • simulator-rendered futures exported as videos

Adapter required:

  • robot-specific action vectors
  • dataset layouts outside WorldBench's current readers
  • prediction folders or latent/state outputs that need conversion to aligned RGB futures
  • 3D, point-cloud, or occupancy outputs that need deterministic rendering

Unsupported as direct targets:

  • action-only VLAs
  • closed-loop real-robot policies
  • text-only environment models
  • task-success evaluation without saved future observations

Matrix: docs/COMPATIBILITY_MATRIX.md

Installation

python -m pip install worldbench
python -m pip install "worldbench[video]"

From a checkout:

python -m pip install -e ".[dev,video]"

Documentation

Honest Limits

WorldBench evaluates saved predicted visual futures. It does not run robot control, execute real-world tasks, infer undocumented action semantics, or prove scientific validity by repository edits alone.

External adoption, human calibration, second-model validation, and commercial validation require evidence outside this repository. Protocols for that work are documented, but not claimed complete.

About

Checkpoint regression testing for robotics world models

Topics

Resources

Code of conduct

Contributing

Security policy

Stars

7 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages