Description
The paper LeWorldModel introduces a new way to train AI “world models” directly from raw pixels in a simple and stable manner, addressing a key problem in prior approaches where models collapse or require complex tricks to work. It proposes a streamlined architecture using just two loss terms—one to predict future states in a latent space and another to keep those representations well-structured—eliminating the need for heavy tuning or pre-trained components. Despite being lightweight (about 15M parameters and trainable on a single GPU), the model achieves competitive performance across various control tasks while planning much faster than larger foundation-model-based systems, and it also learns meaningful physical structure, allowing it to detect unrealistic or physically implausible scenarios.
