跪拜 Guibai
← All articles
Algorithms · Artificial Intelligence · Machine Learning

Nano World Models Puts a Reproducible Baseline Under World-Model Research

By 罗西的思考 ·
Read original on juejin.cn ↗ Google Translate ↗ Alt translation

World-model research has no common baseline across datasets, training recipes, and metrics; NanoWM gives teams a single, reproducible pipeline to isolate which design decisions actually drive planning success. Its ablation on semantic latents, in particular, warns that representation quality does not translate to controllable dynamics under a diffusion objective.

Summary

Nano World Models (NanoWM) is an open-source, minimal codebase for building world models, built around Diffusion Forcing. It unifies multiple predictive objectives (x, epsilon, v, and flow matching), five action-injection mechanisms, three latent encoders, and four model scales into one Hydra-based pipeline. The project's stated goal is to shift world-model research away from inventing new architectures and toward rigorous, reproducible ablation of design choices that actually matter.

Takeaways
— NanoWM supports four model sizes: S at 40M, B at 160M, L at 600M, and XL at 830M parameters.
— The codebase supports diffusion and flow matching, with x, epsilon, and v prediction targets; v-prediction gives the best FID (42.27) on RT-1, epsilon-prediction the worst (48.86).
— Five action-injection methods are implemented: additive, AdaLN, AdaLN-Fuse, FiLM, and cross-attention; FiLM wins on RT-1 (FID 40.62) while additive wins on PushT (FID 23.89).
— Semantic latent spaces (Web-DINO, V-JEPA 2.1) collapse action conditioning: action embeddings reach RMS near 0 (0.002 and 0.001) and MPC planning success drops to 0%.
— SD-VAE is the only latent encoder that supports planning, achieving 25% success on PushT with CEM and a 6x difference in MSE between ground-truth and random actions.
— Long-range rollout error accumulates, but increasing DDIM sampling steps from 50 to 250 reduces LPIPS drift on 50-frame CS:GO generation.
— A single training-evaluation recipe transfers across six datasets, but simple domains (Point Maze FID 9.66) are far easier than complex ones (Granular FID 40.05, RT-1 FID 35.08).
— NanoWM can generate sequences 4x longer than its training context by using Diffusion Forcing's autoregressive sliding window.
Conclusions

The finding that semantic latents (Web-DINO, V-JEPA) drive action embeddings to near zero under diffusion training means latent space choice cannot be separated from the training objective; a representation that is good for recognition is not automatically good for controllable generation.

Cross-attention, the most expressive action-injection mechanism, performed worst on RT-1 because the model could predict adjacent frames without using the action signal, a symptom of overfitting to temporal correlation rather than a failing of the mechanism itself.

The paper's decision to standardize six datasets and a fixed evaluation recipe is a direct response to the fragmentation of world-model research, where every lab uses its own data, training scheme, and metric, making scientific comparison nearly impossible.

Source: juejin.cn ↗ Google Translate ↗ Backup ↗