提升扩散模型在图像超分辨率中的语义与训练推理一致性。
Improving Consistency in Diffusion Models for Image Super-Resolution
- 用图像嵌入替代文本提示,增强细节引导。
- 新方法在Real-ISR上达到当前最佳性能。
- 适合关注图像超分与扩散模型融合的研究者。
近期方法利用强大的文本到图像(T2I)扩散模型实现真实世界图像超分辨率(Real-ISR),效果优于以往模型。但观察发现扩散模型存在两类不一致问题:一是语义不一致,源于扩散引导机制——T2I生成注重文本提示的语义一致性,而Real-ISR强调从低质量(LQ)图像中像素级重建,需更精细的LQ输入语义引导;二是训练-推理不一致,源于DDPM假设高质量(HQ)潜在表示被高斯噪声污染作为每步去噪输入。为此,本文提出ConsisSR以解决上述两类不一致。一方面,设计混合提示适配器(HPA),使用更强的CLIP图像嵌入而非粗粒度分类文本提示,挖掘颜色与纹理信息;另一方面,引入时间感知潜在增强(TALA),基于概率函数p(t)优化SDSR训练策略。通过将加噪后的LQ潜在表示作为输入,TALA不仅聚焦扩散噪声,还逐步将LQ潜在表示向HQ对应物逼近。本方法在现有扩散模型中表现最优,代码将公开。
原文摘要 · Abstract (English)
Recent methods exploit the powerful text-to-image (T2I) diffusion models for real-world image super-resolution (Real-ISR) and achieve impressive results compared to previous models. However, we observe two kinds of inconsistencies in diffusion-based methods which hinder existing models from fully exploiting diffusion priors. The first is the semantic inconsistency arising from diffusion guidance. T2I generation focuses on semantic-level consistency with text prompts, while Real-ISR emphasizes pixel-level reconstruction from low-quality (LQ) images, necessitating more detailed semantic guidance from LQ inputs. The second is the training-inference inconsistency stemming from the DDPM, which improperly assumes high-quality (HQ) latent corrupted by Gaussian noise as denoising inputs for each timestep. To address these issues, we introduce ConsisSR to handle both semantic and training-inference consistencies. On the one hand, to address the semantic inconsistency, we proposed a Hybrid Prompt Adapter (HPA). Instead of text prompts with coarse-grained classification information, we leverage the more powerful CLIP image embeddings to explore additional color and texture guidance. On the other hand, we introduce Time-Aware Latent Augmentation (TALA) to bridge the training-inference inconsistency. Based on the probability function p(t), we accordingly enhance the SDSR training strategy. With LQ latent with Gaussian noise as inputs, our TALA not only focuses on diffusion noise but also refine the LQ latent towards the HQ counterpart. Our method demonstrates state-of-the-art performance among existing diffusion models. The code will be made publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。