无需成对图像数据,用视觉语言模型直接训练图像编辑模型
Learning an Image Editing Model without Image Editing Pairs
- 用视觉语言模型提供编辑反馈,直接优化扩散模型
- 零配对数据下性能媲美大规模监督训练模型
- 适合想减少标注依赖的图像编辑研究者
近期图像编辑模型在自然语言指令下表现优异,但依赖大量输入-目标图像对进行监督微调,而这类真实配对数据难以规模化获取。现有方法使用预训练模型生成合成数据对,但会放大原始模型的缺陷。本文提出全新训练范式,完全摒弃配对数据。通过反向展开扩散模型,并利用视觉语言模型(VLM)评估每一步编辑是否符合指令且保留未修改内容,提供端到端优化梯度。为保证图像质量,引入分布匹配损失(DMD),约束生成图像保持在预训练模型学习的图像流形内。在标准基准测试中,该方法在少步设置下性能与依赖大量有监督配对数据的扩散模型相当。在相同VLM作为奖励模型的前提下,还优于基于强化学习的方法如Flow-GRPO。
原文摘要 · Abstract (English)
Recent image editing models have achieved impressive results while following natural language editing instructions, but they rely on supervised fine-tuning with large datasets of input-target pairs. This is a critical bottleneck, as such naturally occurring pairs are hard to curate at scale. Current workarounds use synthetic training pairs that leverage the zero-shot capabilities of existing models. However, this can propagate and magnify the artifacts of the pretrained model into the final trained model. In this work, we present a new training paradigm that eliminates the need for paired data entirely. Our approach directly optimizes a few-step diffusion model by unrolling it during training and leveraging feedback from vision-language models (VLMs). For each input and editing instruction, the VLM evaluates if an edit follows the instruction and preserves unchanged content, providing direct gradients for end-to-end optimization. To ensure visual fidelity, we incorporate distribution matching loss (DMD), which constrains generated images to remain within the image manifold learned by pretrained models. We evaluate our method on standard benchmarks and include an extensive ablation study. Without any paired data, our method performs on par with various image editing diffusion models trained on extensive supervised paired data, under the few-step setting. Given the same VLM as the reward model, we also outperform RL-based techniques like Flow-GRPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。