Janus-Pro-R1: Advancing Collaborative Visual Comprehension and Generation via Reinforcement Learning

Fei Wu · Siliang Tang (Zhejiang University) · Juncheng Li (Zhejiang University) · Yang Wu (Ant Group) · Jun Xiao (Zhejiang University) · Yueting Zhuang (Zhejiang University) · Yunfei Li (Tsinghua University) · Kaihang Pan (Zhejiang University) · Wendong Bu (Zhejiang University) · Shen Kai (ByteDance) · Yingting Wang (Ant Group) · ZhaoHang (Alibaba Group)
collaborative co-evolutionexploration-exploitation trade-offgenuine cotimage editingimage semantic evaluatoriterative introspective processmultimodal large language modelsreasoning mechanismsreinforcement learningsupervised fine-tuningtext-to-image generationtwo-stage training approachunified image generationvisual comprehensionvisual generation

Recent endeavors in Multimodal Large Language Models (MLLMs) aim to unify visual comprehension and generation. However, these two capabilities remain largely independent, as if they are two separate functions encapsulated within the same model. Consequently, visual comprehension does not enhance visual generation, and the reasoning mechanisms of LLMs have not been fully integrated to revolutionize image generation. In this paper, we propose to enable the collaborative co-evolution of visual comprehension and generation, advancing image generation into an iterative introspective process. We introduce a two-stage training approach: supervised fine-tuning teaches the MLLM with the foundational ability to generate genuine CoT for visual generation, while reinforcement learning activates its full potential via an exploration-exploitation trade-off. Ultimately, we unlock the Aha moment in visual generation, advancing MLLMs from text-to-image tasks to unified image generation. Extensive experiments demonstrate that our model not only excels in text-to-image generation and image editing, but also functions as a superior image semantic evaluator with enhanced visual comprehension capabilities. Project Page: \url{https://janus-pro-r1.github.io}.