EO-WM: A Physically Informed World Model for Probabilistic Earth Observation Forecasting
Paper β’ 2606.27277 β’ Published β’ 3
Official checkpoints for EO-WM: Toward Physically Informed World Modeling for Probabilistic Earth Observation Forecasting (NeurIPS 2026).
EO-WM is a weather-conditioned latent diffusion model for multispectral Earth observation forecasting. It uses 10 observed Sentinel-2 RGBN frames and meteorological forcing to predict 20 future frames at 128 Γ 128 resolution.
This repository provides the EO-WM master weights and the EarthNet2021-finetuned EO-VAE with its matching configuration.
eo-wm/master.safetensors
eo-vae/eo-vae.ckpt
eo-vae/model_config.yaml
Please follow the GitHub README for data preparation, training, and inference instructions.