COMBO: Compositional World Models for Embodied Multi-Agent Cooperation

This repo contains codes for the following paper:

Hongxin Zhang*, Zeyuan Wang*, Qiushi Lyu*, Zheyuan Zhang, Sunli Chen, Tianmin Shu, Yilun Du, Chuang Gan: COMBO: Compositional World Models for Embodied Multi-Agent Cooperation

Project Website: COMBO

Getting started

Installation

We recommend installing dependencies using conda.

conda env create -f environment.yml
conda activate combo

If you're running TDW on a remote Linux server, follow the TDW Installation Document to configure the X server.

Run Experiments

We prepare example scripts to run experiments under the folder scripts.

For example, to run experiments with COMBO agents on TDW-Game, run the following command in folder tdw_mat.

./scripts/run_game.sh

Acknowledgement

This repository is built upon the codebase of AVDC and LLaVA.

Citation

If you find our work useful, please consider citing:

@article{zhang2024combo,
      title={COMBO: Compositional World Models for Embodied Multi-Agent Cooperation}, 
      author={Hongxin Zhang and Zeyuan Wang and Qiushi Lyu and Zheyuan Zhang and Sunli Chen and Tianmin Shu and Yilun Du and Chuang Gan},
      year={2024},
      journal = {arXiv preprint arXiv:2404.10775},
}