diffusion world model

A neural net learns to dream Pokémon Red from button presses

A neural net learned to dream Pokémon Red from button presses.

4 min readMachine Learning
A neural net learns to dream Pokémon Red from button presses
I trained a playable diffusion world model of Pokémon Red, plus a game agent harness on the same System 1/2 split as Krafton's PUBG Ally. Can this be ICLR-workshop worthy? [P]

The gap between what AI can simulate and what it can actually do in the real world is narrowing faster than most people realize, and this project to dream Pokémon Red from button presses is a perfect example of why that matters. A developer trained a diffusion model, the same kind that powers image generators, on recorded gameplay, and the result is a neural net that can walk, fight, and navigate an entire game world it has never truly seen. It learns the rules of the environment, then executes within them, all while running on a single home GPU. That is not a toy. That is a working prototype of how agents will learn to operate in any complex, rule-based system, from spreadsheets to supply chains.

What makes this project stand out is not the novelty of the technique but the discipline of its architecture. The developer split the agent into two layers: a slow thinking planner that reads the game's RAM and sets goals, and a fast reactive executor trained entirely inside the model's imagination. Ten small walking policies, each with 1.58 million parameters, learned to move through a dream version of the game with zero real-world steps, then dropped into the actual emulator and won the first gym badge in 30 out of 30 runs. The planner, whether driven by hand-written rules or a local LLM, consistently outperformed human-designed logic. This mirrors a shift we have seen in other domains, like the system described in When Peer Review Becomes a Masterclass in Withering Critique, where rigorous evaluation separates signal from noise. Here, the developer measured everything, 50 GPU-hours, 16.7 million parameters, 99.27% color accuracy on held-out frames, and published the numbers alongside the code. That transparency is the standard the field needs.

The practical implications for anyone working with data or automation are immediate. If a neural net can learn to navigate a 1996 Game Boy title from button presses alone, the same approach can be applied to any interface with a fixed set of inputs and outputs. Think of enterprise software, legacy databases, or even modern spreadsheet tools that require repetitive navigation. The cost is low enough that a single developer can experiment at home, and the architecture is modular enough to swap in different planners or executors. This is not about replacing human judgment; it is about offloading the grunt work of exploration and execution. The developer who built this works in medical agents and full stack, and calls it a learning project. That should give every team pause about what their own engineers could build with 50 hours of GPU time and a clear problem to solve.

The one unresolved piece is worth watching closely. Battles remain scripted because the dream model ends them early, meaning the agent cannot yet handle adversarial, state-dependent interactions. That is the next frontier, and it mirrors a challenge in larger-scale AI systems: how to keep a world model stable when the environment fights back. For now, the takeaway is concrete: a single developer, working alone, built an agent that learns from imagination and succeeds in reality. The tools to do this are accessible, the metrics are public, and the architecture is replicable. The question is not whether your workflow could benefit from this approach, but whether you are ready to start exploring it.

From Machine Learning

A diffusion model (same family as image generators) that learned Pokémon Red frame by frame from recorded play. Give it the last 4 screens and a button press, and it draws the next screen. 16.7M parameters, 72×80 greyscale, DIAMOND-style EDM U-Net. Left is the real emulator, right is the model dreaming the same moment from the same button presses. Every 10 steps it re-syncs on 4 real frames, otherwise it runs on its own. You can walk around inside the dream with a keyboard.

Same split Krafton used for PUBG Ally: a fast reactive layer and a slow thinking layer.

Read the original at Machine Learning