CoDi: Any-to-Any Generation via Composable Diffusion

Zineng Tang^1*, Ziyi Yang^2†, Chenguang Zhu², Michael Zeng², Mohit Bansal^1†

¹University of North Carolina at Chapel Hill, ²Microsoft Azure Cognitive Services Research ^* Work done at Microsoft internship and UNC. ^†Corresponding Authors

Paper link: https://arxiv.org/abs/2305.11846

Open Source Checklist:

Release Code Structure
Release Multi-Condition & Single-Condition Scripts and Checkpoints
Release Joint-Generation Scripts and Checkpoints
Release Web Demo

Introduction

We present Composable Diffusion (CoDi), a novel generative model capable of generating any combination of output modalities, such as language, image, video, or audio, from any combination of input modalities. Unlike existing generative AI systems, CoDi can generate multiple modalities in parallel and its input is not limited to a subset of modalities like text or image. Despite the absence of training datasets for many combinations of modalities, we propose to align modalities in both the input and output space. This allows CoDi to freely condition on any input combination and generate any group of modalities, even if they are not present in the training data. CoDi employs a novel composable generation strategy which involves building a shared multimodal space by bridging alignment in the diffusion process, enabling the synchronized generation of intertwined modalities, such as temporally aligned video and audio. Highly customizable and flexible, CoDi achieves strong joint-modality generation quality, and outperforms or is on par with the unimodal state-of-the-art for single-modality synthesis.

Citation

If you find our work useful, please consider citing:

@article{tang2023any,
  title={Any-to-Any Generation via Composable Diffusion},
  author={Tang, Zineng and Yang, Ziyi and Zhu, Chenguang and Zeng, Michael and Bansal, Mohit},
  journal={arXiv preprint arXiv:2305.11846},
  year={2023}
}

Contact

Zineng Tang ([email protected])