COMBAT: Conditional World Models for Behavioral Agent Training
Organizations: Overworld AI · Indian Institute of Science Education and Research Bhopal
Abstract
Recent advances in video generation have spurred the development of world models capable of simulating 3D-consistent environments and interactions with static objects. However, a significant limitation remains in their ability to model dynamic, reactive agents that can intelligently influence and interact with the world. To address this gap, we introduce COMBAT, a real-time, action-controlled world model trained on the complex 1v1 fighting game Tekken 3. Our work demonstrates that diffusion models can successfully simulate a dynamic opponent that reacts to player actions, learning its behavior implicitly. Our approach utilizes a 1.2 billion parameter Diffusion Transformer, conditioned on latent representations from a deep compression autoencoder. We employ state-of-the-art techniques, including causal distillation and diffusion forcing, to achieve real-time inference. Crucially, we observe the emergence of sophisticated agent behavior by training the model solely on single-player inputs, without any explicit supervision for the opponent's policy. Unlike traditional imitation learning methods, which require complete action labels, COMBAT learns effectively from partially observed data to generate responsive behaviors for a controllable Player 1. We present an extensive study and introduce novel evaluation methods to benchmark this emergent agent behavior, establishing a strong foundation for training interactive agents within diffusion-based world models.
Figures & tables
| Model | FID | FVD | LPIPS |
|---|---|---|---|
| COMBAT: Pose | 49.7 | 593.4 | 0.05 |
| COMBAT: Non-Pose | 80.9 | 1156.6 | 0.07 |
| Training Step | TAA | ARC |
|---|---|---|
| Ground Truth | 1.00 | 1.00 |
| Step 500 | 3.87 | 1.04 |
| Step 1000 | 0.88 | 3.90 |
| Step 1500 | 1.90 | 1.79 |
| Step 2000 | 1.79 | 1.47 |
Action-Conditioned World Models
Action-conditioned world models predict how environments change under actions to support planning, control, and simulation.
See all 341 papers