Contextualized Diffusion Models for Text-Guided Image and Video Generation
Ling Yang, Zhilong Zhang, Zhaochen Yu, Jingwei Liu, Minkai Xu, Stefano Ermon, Bin Cui
TL;DR
ContextDiff addresses the mismatch between forward and reverse conditioning in text-guided diffusion models by injecting cross-modal context into the full diffusion trajectory. It generalizes the contextualized trajectory adapter to DDPMs and DDIMs, aligning forward and reverse processes to better convey textual semantics. Empirically, ContextDiff achieves new state-of-the-art results on text-to-image generation and text-to-video editing, with improvements in semantic alignment and temporal consistency. The work provides code and demonstrates the practical impact of cross-modal trajectory conditioning for multimodal synthesis.
Abstract
Conditional diffusion models have exhibited superior performance in high-fidelity text-guided visual generation and editing. Nevertheless, prevailing text-guided visual diffusion models primarily focus on incorporating text-visual relationships exclusively into the reverse process, often disregarding their relevance in the forward process. This inconsistency between forward and reverse processes may limit the precise conveyance of textual semantics in visual synthesis results. To address this issue, we propose a novel and general contextualized diffusion model (ContextDiff) by incorporating the cross-modal context encompassing interactions and alignments between text condition and visual sample into forward and reverse processes. We propagate this context to all timesteps in the two processes to adapt their trajectories, thereby facilitating cross-modal conditional modeling. We generalize our contextualized diffusion to both DDPMs and DDIMs with theoretical derivations, and demonstrate the effectiveness of our model in evaluations with two challenging tasks: text-to-image generation, and text-to-video editing. In each task, our ContextDiff achieves new state-of-the-art performance, significantly enhancing the semantic alignment between text condition and generated samples, as evidenced by quantitative and qualitative evaluations. Our code is available at https://github.com/YangLing0818/ContextDiff
