Steering Visual Generation in Unified Multimodal Models with Understanding Supervision

Zeyu Liu1,2,3,4   Zanlin Ni1,2,3   Yang Yue1,2,3   Cheng Da4   Huan Yang4  
Di Zhang4   Kun Gai4   Gao Huang1,2,3  
1Department of Automation, Tsinghua University        
2Beijing Key Laboratory of Embodied Intelligence Systems        
3Institute for Embodied Intelligence and Robotics, Tsinghua University        
4KlingAI Research        
Image Generation Image Editing

Abstract

Unified multimodal models are envisioned to bridge the gap between understanding and generation. Yet, to achieve competitive performance, state-of-the-art models adopt largely decoupled understanding and generation components. This design, while effective for individual tasks, weakens the connection required for mutual enhancement, leaving the potential synergy empirically uncertain.

We propose to explicitly restore this synergy by introducing Understanding-Oriented Post-Training (UNO), a lightweight framework that treats understanding not only as a distinct task, but also a supervisory signal to steer generative representations. By incorporating objectives that encode semantic abstraction (captioning) and structural details (visual regression), we enable effective gradient flow from understanding to generation. Extensive experiments on image generation and editing demonstrate that understanding can serve as an effective catalyst for generation.



Approach

Enhanced Representational Capacity

capacity

Conceptual illustration of training process and backward gradient flow

(a) Generation Training: Current generative training in unified models encode conditions using the understanding expert and transfers information uni-directionally via conditioning to the generative expert, where the outputs are optimized using low-level flow matching objectives. Generation experts receive gradients solely from generative targets, while signals from the understanding pathway remain isolated.


(b) Understanding-Oriented Post-Training: Understanding-oriented post-training jointly supervises the sample with generation and understanding objectives. Specifically, the understanding expert conditions on the noised generative representations, and jointly conducts (i) language supervision through captioning and (ii) visual understanding supervision by using metaquery tokens to predict subsequent understanding tokens. This enables generation blocks to receive gradients from both pathways, enabling strong understanding supervision to directly shape generative representations.



Main Results

pipeline

Quantitative generation results


pipeline

Quantitative editing results



Gallery

Image Generation Image Editing


BibTeX


@article{liu2026steering,
  title={Steering Visual Generation in Unified Multimodal Models with Understanding Supervision},
  author={Liu, Zeyu and Ni, Zanlin and Yue, Yang and Da, Cheng and Yang, Huan and Zhang, Di and Gai, Kun and Huang, Gao},
  journal={arXiv preprint arXiv:2605.05781},
  year={2026}
}

Acknowledgements


Website adapted from the following template.