Star 历史趋势
数据来源: GitHub API · 生成自 Stargazers.cn
README.md
NVIDIA

✨Gamma-World: Generative Multi-Agent World Modeling
Beyond Two Players✨

Fangfu Liu1,2*     Kai He1,3,4*     Tianchang Shen1     Tianshi Cao1     Sanja Fidler1,3,4     Yueqi Duan2     Jun Gao1
Igor Gilitschenski3,4†     Zian Wang1†     Xuanchi Ren1†

1NVIDIA     2Tsinghua University     3University of Toronto     4Vector Institute

              

γ-World teaser

γ-World: We introduce a generative multi-agent world model that rolls out a single shared environment for multiple independently controllable agents. γ-World supports permutation-symmetric agent conditioning with Simplex Rotary Agent Encoding, efficient cross-agent communication with Sparse Hub Attention, real-time 24 FPS streaming with a distilled block-causal student, and zero-shot generalization from two to four players.

📢 News

  • 🚀[06/16/2026] We release the γ-World code and training pipeline.
  • 🚀[05/28/2026] We release γ-World with the project page, paper, videos, qualitative results, and method overview.

🌟 Overview

γ-World interactively generates coherent future frames from multi-agent actions while preserving shared-world consistency, scaling from multiplayer virtual games to real-world multi-robot environments.

https://github.com/user-attachments/assets/11a81855-5b51-4117-bfcd-ef07246e0a4e

📦 User Guide

  • Setup — environment and dependencies
  • Inference — run the bidirectional, causal, and causal-few-step models on your own init frames and actions
  • Training — convert a released checkpoint to DCP and train the teacher / causal / DMD stages

📖 Abstract

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.

We present γ-World, a generative multi-agent world model for interactive simulation. γ-World 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.

🖼️ Gallery

Two-Agent Interaction

Each agent is independently controllable while sharing the same evolving world.

Two-agent interaction

Four-Agent Generalization

Benefiting from the permutation-symmetric simplex agent encoding, γ-World generalizes from two to four players without additional training.

Four-agent generalization

Real-World Robotics Coordination

γ-World extends to real-world multi-robot coordination scenarios, demonstrating applicability beyond virtual environments.

Real-world robotics coordination

🧠 Method

Method overview

γ-World takes synchronized observations and actions from multiple agents as input, tokenizes each agent stream with shared visual and action encoders, and generates future multi-agent rollouts with a causal multi-agent DiT. The model formulates the input with an explicit synchronized agent axis, encodes exchangeable agent identity using Simplex Rotary Agent Encoding, and routes cross-agent information through Sparse Hub Attention. During streaming inference, the causal student uses KV caches for past visual tokens and hub states to preserve block-causal generation while scaling efficiently with the number of agents.

Simplex Rotary Agent Encoding

Simplex Rotary Agent Encoding is a parameter-free extension of 3D RoPE. Instead of assigning agents scalar indices or learned identity vectors, γ-World places them at the vertices of a regular simplex in rotary angle space. All agents have equal pairwise distances, so every pair is permutation-equivalent while each agent retains a distinct rotary phase.

Sparse Hub Attention

Sparse Hub Attention routes cross-agent communication through a small set of learnable hub tokens. Agent tokens attend to their own stream and to the hubs; the hubs aggregate information across agents and broadcast it back. This preserves a shared communication pathway without dense pairwise interaction, reducing cross-agent attention from O(N²) to O(N).

Efficiency

Sparse Hub Attention timing

Sparse Hub Attention achieves significantly lower latency and FLOPs as the number of agents increases, making γ-World more scalable beyond two players.

📚 Citation

If you find γ-World useful for your research or applications, please cite our paper:

@article{gammaworld2026,
    title={Gamma-World: Generative Multi-Agent World Modeling Beyond Two Players},
    author={Fangfu Liu and Kai He and Tianchang Shen and Tianshi Cao and
            Sanja Fidler and Yueqi Duan and Jun Gao and Igor Gilitschenski and
            Zian Wang and Xuanchi Ren},
    journal={arXiv preprint arXiv:2605.28816},
    year={2026}
}

Acknowledgements

The authors would like to thank Product Managers Aditya Mahajan and Matt Cragun for their valuable support and guidance, Jingnan Gao for proof discussion, Yixin Hong for demo creation, and Jiawei Chi and Weiliang Chen for training support.

License

Gamma-World codebase is licensed under the Apache License 2.0.

关于 About

Implementation of Gamma-World: Generative Multi-Agent World Modeling Beyond Two Players
aigcmulti-agentroboticsvideo-gameworldmodel

语言 Languages

Python99.6%
Shell0.2%
Dockerfile0.2%
Just0.1%

提交活跃度 Commit Activity

代码提交热力图
过去 52 周的开发活跃度
7
Total Commits
峰值: 4次/周
Less
More

核心贡献者 Contributors