用残差推理加速遥感图像融合,15步完成超分辨率,速度提升90%以上。
ResPanDiff: Diffusion Model for Pansharpening by Inferring Residual Inference
- 构建马尔可夫链直接生成低分辨与高分辨图像的残差
- 仅需15次采样步骤,性能超越现有最先进方法
- 适合需要快速高精度遥感图像融合的研究与应用
基于扩散模型的全色锐化任务主要受限于推理速度慢,源于大量采样步骤。尽管已有加速采样技术,但在多源图像融合时常牺牲性能。为此,我们提出一种新型高效扩散模型ResPanDiff,显著减少采样步骤而不损失性能。ResPanDiff创新性地设计了一个马尔可夫链,从噪声残差过渡到低分辨多光谱(LRMS)与高分辨多光谱(HRMS)图像间的残差,从而减少采样次数并提升效果。同时,我们设计了潜空间以增强编码阶段特征提取能力,引入浅层条件注入(SC-I)获取高维隐藏特征,并设计损失函数以更好指导残差生成。实验在多个全色锐化数据集上表明,该方法优于近期最先进(SOTA)技术,仅需15次采样步骤,相较基准扩散模型减少超过90%的步骤。实验还包括详尽讨论与消融研究,验证方法有效性。
原文摘要 · Abstract (English)
The implementation of diffusion-based pansharpening task is predominantly constrained by its slow inference speed, which results from numerous sampling steps. Despite the existing techniques aiming to accelerate sampling, they often compromise performance when fusing multi-source images. To ease this limitation, we introduce a novel and efficient diffusion model named Diffusion Model for Pansharpening by Inferring Residual Inference (ResPanDiff), which significantly reduces the number of diffusion steps without sacrificing the performance to tackle pansharpening task. In ResPanDiff, we innovatively propose a Markov chain that transits from noisy residuals to the residuals between the LRMS and HRMS images, thereby reducing the number of sampling steps and enhancing performance. Additionally, we design the latent space to help model extract more features at the encoding stage, Shallow Cond-Injection~(SC-I) to help model fetch cond-injected hidden features with higher dimensions, and loss functions to give a better guidance for the residual generation task. enabling the model to achieve superior performance in residual generation. Furthermore, experimental evaluations on pansharpening datasets demonstrate that the proposed method achieves superior outcomes compared to recent state-of-the-art~(SOTA) techniques, requiring only 15 sampling steps, which reduces over $90\%$ step compared with the benchmark diffusion models. Our experiments also include thorough discussions and ablation studies to underscore the effectiveness of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。