DIAMOND vs Real BattleZone — World Model Prediction

DIAMONDNeurIPS 2024 Spotlight·64x64 @ 15fps

Paper GitHub Weights

Compare with:DIAMOND TWISTER STORM Δ-IRISNEW IRIS

Ground Truth

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

Prediction

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

Higher is better

0.894

Lower is better

38.5

Lower is better

0.0171

FramesPrediction horizon

100

Prediction Quality Over Horizon

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

LPIPSSSIM

Better ↑