← Back to Benchmarks
simmediumatarimetric · varies

Diffusion for World Modeling: Visual Details Matter in Atari

Description

World models constitute a promising approach for training reinforcement learning agents in a safe and sample-efficient manner. Recent world models predominantly operate on sequences of discrete latent variables to model environment dynamics. However, this compression into a compact discrete representation may ignore visual details that are important for reinforcement learning. Concurrently, diffusion models have become a dominant approach for image generation, challenging well-established method

Source

http://arxiv.org/abs/2405.12399v2