mlsanityData to 7 Oct 2026

Efficient Reinforcement for Visual-Textual Thinking with Discrete Diffusion Model

arXiv 2606.14792 ↗First posted 11 Jun 2026cs.CV cs.AIcited by 0

On the map: Diffusion models → Image generation, text-to-image and editing

Abstract (arXiv)

RL-based post-training has been widely adopted to enable interleaved visual and textual reasoning in unified multimodal models capable of both text and image generation. However, most existing approaches are built upon autoregressive (AR) unified models, which require full image regeneration during visual reasoning. In this work, we demonstrate that multimodal discrete diffusion models are effective alternatives to AR models for reinforcement learning in interleaved reasoning, owing to their ability to perform efficient visual rollouts via localized visual editing rather than full image-token regeneration. This reduces rollout computation during GRPO by 26.9% compared to AR baselines, with minimal performance drop. Despite the improved efficiency, we find that joint reward assignment, which employs a shared reward signal across modalities, introduces cross-modal interference between unrelated image and text token sequences during RL updates. To address this issue, we propose factorized reward assignment, a strategy that assigns rewards independently to text and vision segments. With factorized reward assignment, our RL approach achieves an 11.2% improvement over joint reward assignment and a 38.04% improvement over the base model.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 43 of its 50 references that are in the map. First posted on arXiv 11 Jun 2026.

Builds on

43 of 50 references.

Closest ancestors · 10

Same direction, newest first

  1. 2026LLaDA2.0-Uni: Unifying Multimodal Understanding and Generation with Diffusion Large Language Model22 Apr
  2. 2025MMaDA-Parallel: Multimodal Large Diffusion Language Models for Thinking-Aware Editing and Generation12 Nov
  3. 2025Muddit: Liberating Generation Beyond Text-to-Image with a Unified Discrete Diffusion Model29 May
  4. 2025FUDOKI: Discrete Flow-based Unified Understanding and Generation via Kinetic-Optimal Velocities26 May
  5. 2025MMaDA: Multimodal Large Diffusion Language Models21 May
  6. 2025Unified Multimodal Discrete Diffusion26 Mar
  7. 2024Dual Diffusion for Unified Image Generation and Understanding31 Dec
  8. 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generation12 Nov
  9. 2024Emu3: Next-Token Prediction is All You Need27 Sep
  10. 2024ANOLE: An Open, Autoregressive, Native Large Multimodal Models for Interleaved Image-Text Generation8 Jul

Other references · 33

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  2. 2024Chameleon: Mixed-Modal Early-Fusion Foundation Modelscited by 904
  3. 2025Large Language Diffusion Modelscited by 872
  4. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 734
  5. 2025Vision-R1: Incentivizing Reasoning Capability in Multimodal Large Language Modelscited by 653
  6. 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 633
  7. 2024Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Modelcited by 468
  8. 2025Pixel Reasoner: Incentivizing Pixel-Space Reasoning with Curiosity-Driven Reinforcement Learningcited by 267
  9. 2025Imagine while Reasoning in Space: Multimodal Visualization-of-Thoughtcited by 204
  10. 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 178
  11. 2025Thyme: Think Beyond Imagescited by 160
  12. 2024Multimodal ArXiv: A Dataset for Improving Scientific Comprehension of Large Vision-Language Modelscited by 138
21 more
  1. 2025Machine Mental Imagery: Empower Multimodal Reasoning with Latent Visual Tokenscited by 120
  2. 2025Latent Visual Reasoningcited by 99
  3. 2025Visual Planning: Let's Think Only with Imagescited by 85
  4. 2025Zebra-CoT: A Dataset for Interleaved Vision Language Reasoningcited by 78
  5. 2025Thinking with Generated Imagescited by 71
  6. 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Modelscited by 67
  7. 2025ThinkMorph: Emergent Properties in Multimodal Interleaved Chain-of-Thought Reasoningcited by 55
  8. 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Modelscited by 53
  9. 2025Unified Autoregressive Visual Generation and Understanding with Continuous Tokenscited by 51
  10. 2025OneCAT: Decoder-Only Auto-Regressive Model for Unified Understanding and Generationcited by 50
  11. 2025Reinforced Visual Perception with Toolscited by 38
  12. 2024BAGEL: Bootstrapping Agents by Guiding Exploration with Languagecited by 34
  13. 2025Visual Thoughts: A Unified Perspective of Understanding Multimodal Chain-of-Thoughtcited by 34
  14. 2025CodePlot-CoT: Mathematical Visual Reasoning by Thinking with Code-Driven Imagescited by 22
  15. 2024MMAR: Towards Lossless Multi-Modal Auto-Regressive Probabilistic Modelingcited by 19
  16. 2025UGen: Unified Autoregressive Multimodal Model with Progressive Vocabulary Learningcited by 16
  17. 2025ARM-Thinker: Reinforcing Multimodal Generative Reward Models with Agentic Tool Use and Visual Reasoningcited by 13
  18. 2025Consolidating Reinforcement Learning for Multimodal Discrete Diffusion Modelscited by 9
  19. 2026Omni-R1: Towards the Unified Generative Paradigm for Multimodal Reasoningcited by 7
  20. 2026Think in Strokes, Not Pixels: Process-Driven Image Generation via Interleaved Reasoningcited by 6
  21. 2026Thinking with Images as Continuous Actions: Numerical Visual Chain-of-Thoughtcited by 4

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Learning Only with Images: Visual Reinforcement Learning with Reasoning, Rendering, and Visual Feedback
  2. 2026Disentangling Perception and Reasoning in Multimodal LLMs via Reward Design
  3. 2026Towards Unified Multimodal Interleaved Generation via Group Relative Policy Optimization
  4. 2026Faithful-MR1: Faithful Multimodal Reasoning via Anchoring and Reinforcing Visual Attention
  5. 2025Perception-R1: Advancing Multimodal Reasoning Capabilities of MLLMs via Visual Perception Reward
  6. 2025MMRPT: MultiModal Reinforcement Pre-Training via Masked Vision-Dependent Reasoning
  7. 2026Visually-Guided Policy Optimization for Multimodal Reasoning
  8. 2026Credit Where It is Due: Cross-Modality Connectivity Drives Precise Reinforcement Learning for MLLM Reasoning