Probing the Impact of Scale on Data-Efficient, Generalist Transformer World Models for Atari
TLDR
Analyzes scaling behaviors of minimalist transformer world models on Atari 100k, revealing distinct scaling regimes and benefits of joint training.
Reasoning
Strengths include systematic scaling analysis and clear demonstration of joint training benefits; weaknesses are limited scope to Atari and offline expert data, lacking real-world validation.
Read-first score
Read-first score 65.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 48.
Field roles
Rank sensitivity
Stability: volatile; rank range: 387.
Keyword Scores
Deep Analysis
Innovations
- Isolates the independent impact of model scale from architectural mechanisms in world models
- Identifies distinct scaling regimes across Atari environments, with some passing the interpolation threshold and others remaining in the classical regime
- Shows that joint training on multiple environments stabilizes scaling dynamics, ensuring monotonic gains across all environments regardless of their inherent scaling regime
- Demonstrates that improved world model fidelity directly translates to downstream control performance
Methodology
The study uses a minimalist transformer world model trained on fixed offline datasets derived from a presupposed expert policy on the Atari 100k benchmark. Scaling behaviors are analyzed both for individual tasks and a unified setting where a single transformer is trained on a suite of 26 Atari environments. Performance is evaluated via downstream control using policies learned entirely within the simulated dynamics, with results reported as median expert-random-normalized scores.
Key Results
Individual environments exhibit distinct scaling regimes: some show monotonic improvements in the overparameterized regime while others degrade with larger models. Joint training across 26 environments stabilizes scaling, yielding monotonic gains for all environments. The best downstream policy achieves a median expert-random-normalized score of 0.770.
Limitations
- Relies on fixed offline datasets derived from a presupposed expert policy, limiting generalization to other data sources
- Evaluation is confined to the Atari 100k benchmark, which may not reflect broader environments
- Focuses solely on transformer world models and scale, without exploring architectural innovations or other model families