Latent-Foresight: End-to-End Learning Predictable Representations for Latent World Models
Organizations: Archimedes, Athena Research Center, Greece · National Technical University of Athens · University of Crete · IACM-Forth
Abstract
Predicting the future evolution of a scene is a fundamental capability for world modeling. Recent work has shown that operating in the feature space of Vision Foundation Models (VFMs) yields semantically rich representations that support diverse future scene understanding tasks. However, existing approaches rely on two-stage pipelines, where VFM features are first compressed using fixed dimensionality reduction (e.g., PCA) or independently trained autoencoders, and a separate predictor is trained on top of the resulting frozen latent space. This decoupling between representation learning and temporal prediction, as well as approaches that apply predictors directly on raw VFM features, provides no guarantee that the latent space is structured for predictable dynamics. In this work, we propose Latent-Foresight, an end-to-end framework that jointly learns a latent tokenizer and a flow-based generative dynamics model, explicitly shaping the representation to support temporal predictability. To enable stable joint optimization, we introduce several key design choices that prevent latent collapse and align reconstruction with generative objectives. Extensive experiments show that our approach learns more temporally coherent latent representations and consistently outperforms two-stage baselines across multiple future scene understanding tasks and prediction horizons, while eliminating separate training stages, including during high-resolution adaptation. We provide the implementation code and model weights at https://github.com/Sta8is/Latent-Foresight
Figures & tables
| Method | Semantic Segmentation | Depth | Surface Normals | |||||
| Short | Mid | Short | Mid | Short | Mid | |||
| ALL | MO | ALL | MO | 11.25 ∘ | 11.25 ∘ | |||
| Oracle | 77.1 | 77.3 | 77.1 | 77.3 | 89.6 | 89.6 | 96.3 | 96.3 |
| VISTA ft | 64.9 | 62.1 | 53.9 | 51.0 | 86.4 | 82.8 | 93.0 | 90.0 |
| Dino-Foresight | 71.8 | 71.7 | 59.8 | 57.6 | 88.6 | 85.4 | 94.4 | 91.3 |
| DeltaTok | 72.1 | - | 60.0 | - | 88.5 | 85.6 | - | - |
| Method | Dim | Reconstruction | Prediction (Short-Term) | Prediction (Mid-Term) | ||||||
| Cos Sim | ALL | MO | Cos Sim | ALL | MO | Cos Sim | ALL | MO | ||
| Low Resolution | ||||||||||
| Raw VFM Features | 3072 | 1.0 | 68.12 | 66.81 | 0.968 | 64.02 | 62.78 | 0.933 | 54.22 | 50.77 |
| Two Stage with PCA | 1152 | 0.977 | 65.63 | 61.97 | 0.955 | 61.56 | 57.93 | 0.921 | 52.49 | 47.60 |
| Two Stage with PCA | 256 | 0.952 | 51.19 | 37.18 | 0.940 | 47.85 | 33.63 | 0.914 | 43.10 | 29.46 |
| Two-Stage with AE | 256 | 0.989 | 67.96 | 66.88 | 0.967 | 64.69 | 62.57 | 0.938 | 56.31 | 53.03 |
| Method | Cos Sim | AbsRel | |||||||
| Mid | Long | Longer | Mid | Long | Longer | Mid | Long | Longer | |
| Oracle | - | - | - | 84.0 | 84.0 | 84.0 | .138 | .138 | .138 |
| Dino-Foresight | 0.924 | 0.894 | 0.868 | 76.8 | 73.1 | 69.5 | 0.321 | 0.377 | 0.368 |
| Two-Stage with AE | 0.927 | 0.888 | 0.857 | 75.9 | 72.0 | 68.2 | 0.242 | 0.328 | 0.390 |
| Latent-Foresight | 0.936 | 0.903 | 0.874 | 80.8 | 76.4 | 72.0 | 0.206 | 0.267 | 0.310 |
| Dino-Foresight (zero-shot) | 0.905 | 0.865 | 0.837 | 74.8 | 69.5 | 66.0 | 0.349 | 0.424 | 0.415 |
| 1-step | 8-step | |||||
| Method | BG | ALL | MO | BG | ALL | MO |
| VFMF (1 gen) | 97.71 | 88.19 | 78.66 | 90.34 | 59.53 | 28.71 |
| VFMF*(32 gen) | – | – | – | 91.62 | 61.74 | 31.86 |
| VFMF (32 gen) | 98.09 | 89.99 | 81.88 | 91.95 | 62.07 | 32.19 |
| Latent-Foresight (1 gen) | 98.32 | 91.16 | 83.99 | 91.64 | 63.47 | 35.29 |
| Method | Reconstruction | Prediction (Short-Term) | Prediction (Mid-Term) | |||||||
| Cos Sim | ALL | MO | Cos Sim | ALL | MO | Cos Sim | ALL | MO | ||
| (a) | Latent-Foresight (Ours) | 0.989 | 68.29 | 67.26 | 0.972 | 65.45 | 64.10 | 0.944 | 56.84 | 53.86 |
| (b) | w/o RoPE in AE | 0.989 | 68.22 | 67.26 | 0.971 | 65.42 | 64.37 | 0.944 | 56.60 | 53.42 |
| (c) | w/o Auxiliary Reconstruction ( ) | 0.988 | 68.27 | 67.19 | 0.967 | 65.09 | 63.43 | 0.937 | 56.55 | 53.36 |
| (d) | BatchNorm LayerNorm | 0.989 | 68.27 | 67.22 | 0.970 | 64.70 | 63.53 | 0.942 | 55.88 | 52.66 |
| (e) | BatchNorm KL Regularization | 0.968 | 64.43 | 64.50 | 0.955 | 61.33 | 61.31 | 0.913 | 46.64 | 41.78 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Semantic Segm. | Depth | Surface Normals | |||||||||
| Short | Mid | Short | Mid | Short | Mid | |||||||
| ALL | MO | ALL | MO | AbsR | AbsR | m | 11.25 ∘ | m | 11.25 ∘ | |||
| Oracle | 77.1 | 77.3 | 77.1 | 77.3 | 89.6 | .103 | 89.6 | .103 | 2.88 | 96.3 | 2.88 | 96.3 |
| Copy Last | 54.7 | 52.0 | 40.4 | 32.3 | 84.1 | .154 | 77.8 | .212 | 4.41 | 89.2 | 5.39 | 84.0 |
| VISTA ft | 64.9 | 62.1 | 53.9 | 51.0 | 86.4 | .124 | 82.8 | .153 | 3.75 | 93.0 | 4.30 | 90.0 |
| Dino-Foresight | 71.8 | 71.7 | 59.8 | 57.6 | 88.6 | .114 | 85.4 | .136 | 3.39 | 94.4 | 4.00 | 91.3 |
| Method | Reconstruction | Prediction (Short-Term) | Prediction (Mid-Term) | ||||||
| Cos Sim | ALL | MO | Cos Sim | ALL | MO | Cos Sim | ALL | MO | |
| Cosine Only | 0.889 | 27.50 | 15.78 | 0.895 | 33.22 | 24.23 | 0.881 | 32.46 | 22.92 |
| MSE Only | 0.988 | 68.26 | 67.28 | 0.971 | 65.36 | 64.14 | 0.943 | 56.54 | 53.59 |
| MSE + Cosine | 0.989 | 68.22 | 67.26 | 0.971 | 65.24 | 64.37 | 0.944 | 56.60 | 53.42 |
| Dimensionality | Reconstruction | Prediction (Short-Term) | Prediction (Mid-Term) | ||||||
| Cos Sim | ALL | MO | Cos Sim | ALL | MO | Cos Sim | ALL | MO | |
| 32 | 0.980 | 64.90 | 61.84 | 0.966 | 62.59 | 59.93 | 0.941 | 54.37 | 49.82 |
| 64 | 0.984 | 67.14 | 65.79 | 0.969 | 64.24 | 62.48 | 0.943 | 56.57 | 54.03 |
| 128 | 0.986 | 67.77 | 66.51 | 0.971 | 65.26 | 64.03 | 0.945 | 56.95 | 54.12 |
| 256 | 0.989 | 68.22 | 67.26 | 0.971 | 65.42 | 64.37 | 0.944 | 56.60 | 53.42 |
| 512 | 0.990 | 68.21 | 67.18 | 0.971 | 65.23 | 64.00 | 0.942 | 56.80 | 54.24 |
| Noise Distribution | Reconstruction | Prediction (Short-Term) | Prediction (Mid-Term) | ||||||
| Cos Sim | ALL | MO | Cos Sim | ALL | MO | Cos Sim | ALL | MO | |
| Uniform | 0.990 | 68.17 | 67.17 | 0.969 | 64.34 | 63.11 | 0.937 | 54.60 | 51.00 |
| Logit-Normal (-0.8,0.8) | 0.990 | 68.18 | 67.18 | 0.967 | 63.73 | 62.75 | 0.933 | 53.94 | 50.89 |
| Logit-Normal (-2,0.8) | 0.990 | 68.22 | 67.18 | 0.971 | 65.35 | 64.40 | 0.941 | 56.30 | 53.42 |
| Logit-Normal (-2,1.5) | 0.990 | 68.21 | 67.18 | 0.971 | 65.23 | 64.00 | 0.942 | 56.80 | 54.24 |