Saved in:
Bibliographic Details
Main Authors: Liu, Yaoli, Ouyang, Ziheng, Lou, Shengtao, Song, Yiren
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.19990
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908673858600960
author Liu, Yaoli
Ouyang, Ziheng
Lou, Shengtao
Song, Yiren
author_facet Liu, Yaoli
Ouyang, Ziheng
Lou, Shengtao
Song, Yiren
contents Reference-guided image generation has progressed rapidly, yet current diffusion models still struggle to preserve fine-grained visual details when refining a generated image using a reference. This limitation arises because VAE-based latent compression inherently discards subtle texture information, causing identity- and attribute-specific cues to vanish. Moreover, post-editing approaches that amplify local details based on existing methods often produce results inconsistent with the original image in terms of lighting, texture, or shape. To address this, we introduce \ourMthd{}, a detail-aware refinement framework that performs two consecutive stages of reference-driven correction to enhance pixel-level consistency. We first adapt a single-image diffusion editor by fine-tuning it to jointly ingest the draft image and the reference image, enabling globally coherent refinement while maintaining structural fidelity. We then apply reinforcement learning to further strengthen localized editing capability, explicitly optimizing for detail accuracy and semantic consistency. Extensive experiments demonstrate that \ourMthd{} significantly improves reference alignment and fine-grained detail preservation, producing faithful and visually coherent edits that surpass both open-source and commercial models on challenging reference-guided restoration benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2511_19990
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle OmniRefiner: Reinforcement-Guided Local Diffusion Refinement
Liu, Yaoli
Ouyang, Ziheng
Lou, Shengtao
Song, Yiren
Computer Vision and Pattern Recognition
Reference-guided image generation has progressed rapidly, yet current diffusion models still struggle to preserve fine-grained visual details when refining a generated image using a reference. This limitation arises because VAE-based latent compression inherently discards subtle texture information, causing identity- and attribute-specific cues to vanish. Moreover, post-editing approaches that amplify local details based on existing methods often produce results inconsistent with the original image in terms of lighting, texture, or shape. To address this, we introduce \ourMthd{}, a detail-aware refinement framework that performs two consecutive stages of reference-driven correction to enhance pixel-level consistency. We first adapt a single-image diffusion editor by fine-tuning it to jointly ingest the draft image and the reference image, enabling globally coherent refinement while maintaining structural fidelity. We then apply reinforcement learning to further strengthen localized editing capability, explicitly optimizing for detail accuracy and semantic consistency. Extensive experiments demonstrate that \ourMthd{} significantly improves reference alignment and fine-grained detail preservation, producing faithful and visually coherent edits that surpass both open-source and commercial models on challenging reference-guided restoration benchmarks.
title OmniRefiner: Reinforcement-Guided Local Diffusion Refinement
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2511.19990