@inproceedings{
kim2024a,
title={A Versatile Diffusion Transformer with Mixture of Noise Levels for Audiovisual Generation},
author={Gwanghyun Kim and Alonso Martinez and Yu-Chuan Su and Brendan Jou and Jose Lezama and Agrim Gupta and Lijun Yu and Lu Jiang and Aren Jansen and Jacob C Walker and Krishna Somandepalli},
booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
year={2024},
url={https://openreview.net/forum?id=cs1HISJkLU}
}
Training diffusion models for audiovisual sequences allows for a range of generation tasks by learning conditional distributions of various input-output combinations of the two modalities. Nevertheless, this strategy often requires training a separate model for each task which is expensive. Here, we propose a novel training approach to effectively learn arbitrary conditional distributions in the audiovisual space. Our key contribution lies in how we parameterize the diffusion timestep in the forward diffusion process. Instead of the standard fixed diffusion timestep, we propose applying variable diffusion timesteps across the temporal dimension and across modalities of the inputs. This formulation offers flexibility to introduce variable noise levels for various portions of the input, hence the term mixture of noise levels. We propose a transformer-based audiovisual latent diffusion model and show that it can be trained in a task-agnostic fashion using our approach to enable a variety of audiovisual generation tasks at inference time. Experiments demonstrate the versatility of our method in tackling cross-modal and multimodal interpolation tasks in the audiovisual space. Notably, our proposed approach surpasses baselines in generating temporally and perceptually consistent samples conditioned on the input. Project page: neurips13025.github.io
A Versatile Diffusion Transformer with Mixture of Noise Levels for Audiovisual Generation
Gwanghyun Kim Alonso Martinez Yu-Chuan Su Brendan Jou Jose Lezama Agrim Gupta Lijun Yu Lu Jiang Aren Jansen Jacob C Walker Krishna Somandepalli