Visual-Aware CoT: Achieving High-Fidelity Visual Consistency in Unified Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ye, Zixuan, Liu, Quande, Wei, Cong, Zhang, Yuanxing, Wang, Xintao, Wan, Pengfei, Gai, Kun, Luo, Wenhan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909973758345216
author Ye, Zixuan
Liu, Quande
Wei, Cong
Zhang, Yuanxing
Wang, Xintao
Wan, Pengfei
Gai, Kun
Luo, Wenhan
author_facet Ye, Zixuan
Liu, Quande
Wei, Cong
Zhang, Yuanxing
Wang, Xintao
Wan, Pengfei
Gai, Kun
Luo, Wenhan
contents Recently, the introduction of Chain-of-Thought (CoT) has largely improved the generation ability of unified models. However, it is observed that the current thinking process during generation mainly focuses on the text consistency with the text prompt, ignoring the \textbf{visual context consistency} with the visual reference images during the multi-modal generation, e.g., multi-reference generation. The lack of such consistency results in the failure in maintaining key visual features (like human ID, object attribute, style). To this end, we integrate the visual context consistency into the reasoning of unified models, explicitly motivating the model to sustain such consistency by 1) Adaptive Visual Planning: generating structured visual check list to figure out the visual element of needed consistency keeping, and 2) Iterative Visual Correction: performing self-reflection with the guidance of check lists and refining the generated result in an iterative manner. To achieve this, we use supervised finetuning to teach the model how to plan the visual checking, conduct self-reflection and self-refinement, and use flow-GRPO to further enhance the visual consistency through a customized visual checking reward. The experiments show that our method outperforms both zero-shot unified models and those with text CoTs in multi-modal generation, demonstrating higher visual context consistency.
format Preprint
id arxiv_https___arxiv_org_abs_2512_19686
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Visual-Aware CoT: Achieving High-Fidelity Visual Consistency in Unified Models
Ye, Zixuan
Liu, Quande
Wei, Cong
Zhang, Yuanxing
Wang, Xintao
Wan, Pengfei
Gai, Kun
Luo, Wenhan
Computer Vision and Pattern Recognition
Recently, the introduction of Chain-of-Thought (CoT) has largely improved the generation ability of unified models. However, it is observed that the current thinking process during generation mainly focuses on the text consistency with the text prompt, ignoring the \textbf{visual context consistency} with the visual reference images during the multi-modal generation, e.g., multi-reference generation. The lack of such consistency results in the failure in maintaining key visual features (like human ID, object attribute, style). To this end, we integrate the visual context consistency into the reasoning of unified models, explicitly motivating the model to sustain such consistency by 1) Adaptive Visual Planning: generating structured visual check list to figure out the visual element of needed consistency keeping, and 2) Iterative Visual Correction: performing self-reflection with the guidance of check lists and refining the generated result in an iterative manner. To achieve this, we use supervised finetuning to teach the model how to plan the visual checking, conduct self-reflection and self-refinement, and use flow-GRPO to further enhance the visual consistency through a customized visual checking reward. The experiments show that our method outperforms both zero-shot unified models and those with text CoTs in multi-modal generation, demonstrating higher visual context consistency.
title Visual-Aware CoT: Achieving High-Fidelity Visual Consistency in Unified Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.19686