DIAMOND vs Real ChopperCommand — World Model Prediction

DIAMONDNeurIPS 2024 Spotlight·64x64 @ 15fps

Paper GitHub Weights

Compare with:DIAMOND TWISTER STORM Δ-IRISNEW IRIS

Ground Truth

Atari ChopperCommand real gameplay (ground truth frames, 64x64 @ 15fps)

Prediction

Atari ChopperCommand predicted by DIAMOND world model (64x64 @ 15fps)

Higher is better

0.9481

Lower is better

4.3

Lower is better

0.0096

FramesPrediction horizon

100

Prediction Quality Over Horizon

H=1/4/16/64. Two stacked charts; higher is better in both.

LPIPSSSIM

Better ↑