arXiv:2510.02253cs.CVcs.AI2025-10中稿 · ICLR被引 32

用区域监督提升扩散模型拖拽编辑效果,解决图像扭曲问题。

DragFlow: Unleashing DiT Priors with Region Based Supervision for Drag Editing

  • 改用区域而非点的拖拽方式,增强特征指导能力。
  • 在新基准上超越现有方法,显著减少目标区域失真。
  • 适合需要精准图像编辑的研究者与设计师使用。

基于拖拽的图像编辑长期受目标区域失真困扰,主要因早期模型(如Stable Diffusion)先验能力不足,难以将优化后的潜在表示映射回自然图像流形。随着从UNet-based DDPM转向更可扩展的DiT与流匹配架构(如SD3.5、FLUX),生成先验显著增强,推动了各类编辑任务进步。然而,拖拽编辑尚未受益于此。本文提出首个有效利用FLUX强先验的拖拽编辑框架DragFlow,实现显著性能提升。我们发现,直接将点级拖拽应用于DiT表现不佳:与UNet高度压缩的特征不同,DiT特征结构不足,无法提供可靠的点级运动引导。为此,DragFlow引入区域级编辑范式,通过仿射变换实现更丰富一致的特征监督。同时,集成预训练开放域个性化适配器(如IP-Adapter)提升主体一致性,并通过梯度掩码硬约束保持背景保真度。进一步采用多模态大语言模型(MLLMs)解决任务歧义。为评估,我们构建新型区域级拖拽基准ReD Bench,包含区域级指令。在DragBench-DR和ReD Bench上的大量实验表明,DragFlow超越点级与区域级基线,达到拖拽编辑新SOTA。代码与数据集已开源。

原文摘要 · Abstract (English)

Drag-based image editing has long suffered from distortions in the target region, largely because the priors of earlier base models, Stable Diffusion, are insufficient to project optimized latents back onto the natural image manifold. With the shift from UNet-based DDPMs to more scalable DiT with flow matching (e.g., SD3.5, FLUX), generative priors have become significantly stronger, enabling advances across diverse editing tasks. However, drag-based editing has yet to benefit from these stronger priors. This work proposes the first framework to effectively harness FLUX's rich prior for drag-based editing, dubbed DragFlow, achieving substantial gains over baselines. We first show that directly applying point-based drag editing to DiTs performs poorly: unlike the highly compressed features of UNets, DiT features are insufficiently structured to provide reliable guidance for point-wise motion supervision. To overcome this limitation, DragFlow introduces a region-based editing paradigm, where affine transformations enable richer and more consistent feature supervision. Additionally, we integrate pretrained open-domain personalization adapters (e.g., IP-Adapter) to enhance subject consistency, while preserving background fidelity through gradient mask-based hard constraints. Multimodal large language models (MLLMs) are further employed to resolve task ambiguities. For evaluation, we curate a novel Region-based Dragging benchmark (ReD Bench) featuring region-level dragging instructions. Extensive experiments on DragBench-DR and ReD Bench show that DragFlow surpasses both point-based and region-based baselines, setting a new state-of-the-art in drag-based image editing. Code and dataset are available at https://github.com/Edennnnnnnnnn/DragFlow.

图像编辑扩散模型拖拽交互区域监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。