arXiv:2505.22129cs.CV2025-05ICCV被引 15

揭秘Stable Diffusion生成全景图的关键机制,提出高效新方法

What Makes for Text to 360-degree Panorama Generation with Stable Diffusion?

  • 发现注意力模块中值矩阵和输出权重对全景生成更关键
  • 新框架UniPano性能超现有方法,训练更快更省显存
  • 适合研究扩散模型与全景生成的开发者参考

近期文本到图像扩散模型(如Stable Diffusion)的发展推动了其在360度全景图生成中的应用。已有工作表明,通过低秩适配技术可在预训练扩散模型上实现全景图生成。然而,透视图与全景图之间存在显著领域差异,引发对这一成功背后机制的疑问。本文假设并验证:在全景数据微调时,可训练参数表现出不同行为,其适应过程隐含利用预训练模型先验知识的内在机制。分析发现:1)注意力模块中的查询与键矩阵负责透视与全景共有的信息,对全景生成影响较小;2)值矩阵与输出权重矩阵则专门用于将预训练知识适配至全景域,在微调中起更关键作用。基于此,我们提出简单高效的UniPano框架,作为未来研究的优雅基线。该方法不仅性能优于现有方法,且相比双分支方案显著降低内存占用与训练时间,支持更高分辨率端到端全景生成。代码将公开。

原文摘要 · Abstract (English)

Recent prosperity of text-to-image diffusion models, e.g. Stable Diffusion, has stimulated research to adapt them to 360-degree panorama generation. Prior work has demonstrated the feasibility of using conventional low-rank adaptation techniques on pre-trained diffusion models to generate panoramic images. However, the substantial domain gap between perspective and panoramic images raises questions about the underlying mechanisms enabling this empirical success. We hypothesize and examine that the trainable counterparts exhibit distinct behaviors when fine-tuned on panoramic data, and such an adaptation conceals some intrinsic mechanism to leverage the prior knowledge within the pre-trained diffusion models. Our analysis reveals the following: 1) the query and key matrices in the attention modules are responsible for common information that can be shared between the panoramic and perspective domains, thus are less relevant to panorama generation; and 2) the value and output weight matrices specialize in adapting pre-trained knowledge to the panoramic domain, playing a more critical role during fine-tuning for panorama generation. We empirically verify these insights by introducing a simple framework called UniPano, with the objective of establishing an elegant baseline for future research. UniPano not only outperforms existing methods but also significantly reduces memory usage and training time compared to prior dual-branch approaches, making it scalable for end-to-end panorama generation with higher resolution. The code will be released.

全景生成扩散模型微调机制

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