用注意力监督实现无需掩码的精准局部风格迁移。
RegionRoute: Regional Style Transfer with Diffusion Model
- 通过对齐风格令牌注意力与物体掩码,显式指导风格应用位置。
- 在目标区域实现高精度风格匹配,未编辑区域保持身份一致。
- 支持多风格高效适配,适合需要精细控制的图像编辑场景。
基于扩散模型的风格迁移在空间控制上仍具挑战,因模型将风格视为全局特征,缺乏对风格表示的空间定位,难以将风格限定于特定对象或区域。现有方法通常依赖手工掩码或多阶段后处理,引入边界伪影且泛化能力差。为此,我们提出一种注意力监督的扩散框架,在训练中通过将风格令牌注意力与物体掩码对齐,显式指导风格应用位置。设计了基于KL散度的聚焦损失和基于二值交叉熵的覆盖损失,联合优化定位准确性和覆盖密度。采用模块化LoRA-MoE结构,实现高效可扩展的多风格适应。为评估局部风格化效果,引入区域风格编辑得分(Regional Style Editing Score),通过CLIP相似性衡量目标区域风格匹配度,以掩码LPIPS和像素级一致性评估未编辑区域身份保留。实验表明,该方法可在推理时实现免掩码、单对象风格迁移,生成区域精确且视觉连贯的结果,显著优于现有扩散模型风格编辑方法。
原文摘要 · Abstract (English)
Precise spatial control in diffusion-based style transfer remains challenging. This challenge arises because diffusion models treat style as a global feature and lack explicit spatial grounding of style representations, making it difficult to restrict style application to specific objects or regions. To our knowledge, existing diffusion models are unable to perform true localized style transfer, typically relying on handcrafted masks or multi-stage post-processing that introduce boundary artifacts and limit generalization. To address this, we propose an attention-supervised diffusion framework that explicitly teaches the model where to apply a given style by aligning the attention scores of style tokens with object masks during training. Two complementary objectives, a Focus loss based on KL divergence and a Cover loss using binary cross-entropy, jointly encourage accurate localization and dense coverage. A modular LoRA-MoE design further enables efficient and scalable multi-style adaptation. To evaluate localized stylization, we introduce the Regional Style Editing Score, which measures Regional Style Matching through CLIP-based similarity within the target region and Identity Preservation via masked LPIPS and pixel-level consistency on unedited areas. Experiments show that our method achieves mask-free, single-object style transfer at inference, producing regionally accurate and visually coherent results that outperform existing diffusion-based editing approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。