无需训练,通过移植噪声块实现图像生成的精准空间定位
GrounDiT: Grounding Diffusion Transformers via Noisy Patch Transplantation
- 利用DiT的语义共享特性,在每步去噪时生成对应框区域的噪声块
- 在HRS和DrawBench上性能超越现有无训练方法,定位更精确
- 适合需要精细区域控制但无法微调模型的用户
我们提出GrounDiT,一种针对扩散Transformer(DiT)的新型无训练空间定位技术,用于文本到图像生成。空间定位通过边界框实现,具有简单且通用的优点,可增强用户对生成图像的控制。然而,以往无训练方法依赖反向传播更新噪声图像,常难以精准控制单个边界框。本文利用Transformer架构的灵活性,证明DiT可生成与每个边界框对应的噪声块,完整编码目标物体,实现细粒度区域控制。该方法基于DiT的一个关键特性——语义共享:当小块与可生成尺寸图像共同去噪时,两者会成为语义克隆。每个块在独立分支中去噪后,于每一步时间戳被移植回原噪声图像对应区域,从而实现稳健的空间定位。在HRS和DrawBench基准测试中,相比先前无训练方法,GrounDiT达到最优性能。
原文摘要 · Abstract (English)
We introduce GrounDiT, a novel training-free spatial grounding technique for text-to-image generation using Diffusion Transformers (DiT). Spatial grounding with bounding boxes has gained attention for its simplicity and versatility, allowing for enhanced user control in image generation. However, prior training-free approaches often rely on updating the noisy image during the reverse diffusion process via backpropagation from custom loss functions, which frequently struggle to provide precise control over individual bounding boxes. In this work, we leverage the flexibility of the Transformer architecture, demonstrating that DiT can generate noisy patches corresponding to each bounding box, fully encoding the target object and allowing for fine-grained control over each region. Our approach builds on an intriguing property of DiT, which we refer to as semantic sharing. Due to semantic sharing, when a smaller patch is jointly denoised alongside a generatable-size image, the two become semantic clones. Each patch is denoised in its own branch of the generation process and then transplanted into the corresponding region of the original noisy image at each timestep, resulting in robust spatial grounding for each bounding box. In our experiments on the HRS and DrawBench benchmarks, we achieve state-of-the-art performance compared to previous training-free approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。