Parallel Multimodal Diffusion Language Models for Thinking-Aware Editing and Generation
Ye Tian, Ling Yang, JiongFan Yang, Anran Wang, Yu Tian, Jiani zheng, Haochen Wang, Zhiyang Teng
Abstract
While thinking-aware generation aims to improve performance on complex tasks, we identify a critical failure mode where existing sequential, autoregressive approaches can paradoxically degrade performance due to error propagation. To systematically analyze this issue, we propose ParaBench, a new benchmark designed to evaluate both text and image output modalities. Our analysis using ParaBench reveals that this performance degradation is strongly correlated with poor alignment between the generated reasoning and the final image. To resolve this, we propose a parallel multimodal diffusion framework that enables continuous, bidirectional interaction between text and images throughout the entire denoising trajectory. The model is trained with supervised finetuning and then further optimized by Parallel Reinforcement Learning (ParaRL), a novel strategy that applies semantic rewards along the trajectory to enforce cross-modal consistency. Experiments validate that our approach significantly improves cross-modal alignment and semantic consistency, achieving a 6.9\% improvement in Output Alignment on ParaBench compared to the state-of-the-art model, Bagel, establishing a more robust paradigm for thinking-aware image synthesis.
BibTeX
@inproceedings{
tian2026parallel,
title={Parallel Multimodal Diffusion Language Models for Thinking-Aware Editing and Generation},
author={Ye Tian and Ling Yang and JiongFan Yang and Anran Wang and Yu Tian and Jiani zheng and Haochen Wang and Zhiyang Teng and Zhuochen Wang and Yinjie Wang and Yunhai Tong and Mengdi Wang and Xiangtai Li},
booktitle={The Fourteenth International Conference on Learning Representations},
year={2026},
url={https://openreview.net/forum?id=mkQAd11ovn}
}