Get the latest tech news

Diffusion for World Modeling

Diffusion for World Modeling: Visual Details Matter in Atari (DIAMOND) 💎 Webpage

The diffusion model takes into account the agent’s action and the previous frames to simulate the environment response. We find that diffusion-based DIAMOND provides better modeling of important visual details than the discrete token-based IRIS. DIAMOND's world model is able to better capture important visual details than the discrete token-based IRIS.

Get the Android app