Search papers, labs, and topics across Lattice.
This paper introduces Gamma-World, a generative multi-agent world model for interactive simulation that addresses the challenges of scaling world models to multi-agent settings. They propose Simplex Rotary Agent Encoding, a parameter-free extension of 3D RoPE, to represent agents as vertices of a regular simplex, ensuring independent controllability and permutation-symmetry. Additionally, Sparse Hub Attention is introduced to reduce cross-agent attention cost, and a full-context diffusion teacher is distilled into a causal student for real-time rollout at 24 FPS.
Forget fixed agent slots and quadratic attention: Gamma-World uses simplex embeddings and sparse hubs to generate interactive multi-agent environments with better fidelity and control, even generalizing from 2 to 4 players without retraining.
World models for interactive video generation have largely focused on single-agent settings, where future observations are generated from a single control signal. However, many generated environments require multi-agent interaction: multiple players, robots, or embodied agents act simultaneously within a shared space. Scaling world models to such settings requires a principled multi-agent design: agents should remain independently controllable, permutation-symmetric, and support efficient inference while maintaining consistency across time and perspectives. In this paper, we present our generative multi-agent world model for interactive simulation. It introduces Simplex Rotary Agent Encoding, a parameter-free extension of 3D RoPE that represents agents as vertices of a regular simplex in rotary angle space. This gives each agent a distinct phase while making all agents permutation-equivalent, enabling scalable agent identity without learned per-slot identities or a fixed agent ordering. To avoid dense all-to-all attention across agents, we further propose Sparse Hub Attention, where learnable hub tokens mediate token interaction across agents, reducing cross-agent attention cost from quadratic to linear in the number of agents. For real-time rollout, we distill a full-context diffusion teacher into a causal student that generates temporal blocks sequentially with KV caching, enabling action-responsive generation at 24 FPS. Experiments in multiplayer virtual environments show that our model improves video fidelity, action controllability, and inter-agent consistency over slot-based and dense-attention baselines, while generalizing from two to four players without additional training.
Tsinghua AI