arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.19254cs.CV

RegionRoute: 区域风格迁移与扩散模型

RegionRoute: Regional Style Transfer with Diffusion Model

The University of Texas at Austin(德克萨斯大学奥斯汀分校);Dolby Laboratories, Inc.(杜比实验室)

Bowen Chen, Jake Zuena, Alan C. Bovik, Divya Kothandaraman

AI总结:

RegionRoute提出一种注意力监督的扩散框架,通过训练对齐风格标记的注意力分数与对象掩码,实现精确的区域风格迁移,提升风格应用的局部化能力。

摘要:

精确的空间控制在基于扩散的风格迁移中仍然具有挑战性。这一挑战源于扩散模型将风格视为全局特征,并缺乏对风格表示的显式空间定位,使得难以将风格应用限制在特定对象或区域。据我们所知,现有的扩散模型无法实现真正的局部化风格迁移,通常依赖于手工制作的掩码或多阶段后处理,这会引入边界伪影并限制泛化能力。为了解决这一问题,我们提出了一种注意力监督的扩散框架,通过在训练过程中对齐风格标记的注意力分数与对象掩码,显式地教导模型在哪里应用给定的风格。两个互补的目标,基于KL散度的聚焦损失和使用二元交叉熵的覆盖损失,共同促进准确的定位和密集的覆盖。模块化的LoRA-MoE设计进一步使高效且可扩展的多风格适应成为可能。为了评估局部化的风格化,我们引入了区域风格编辑分数,该分数通过CLIP相似性在目标区域内测量区域风格匹配,并通过掩码LPIPS和像素级一致性在未编辑区域中测量身份保持。实验表明,我们的方法在推理时实现了无掩码、单对象的风格迁移,产生区域准确且视觉一致的结果,优于现有的基于扩散的编辑方法。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑