用高效确定性去噪提升小卫星图像分辨率16倍
Efficient Denoising Method to Improve The Resolution of Satellite Images
- 将扩散模型转为一致性模型,用确定性微分方程加速去噪
- 在DOTA v2.0数据集上实现16倍超分辨率,计算耗时减少20倍
- 适合需要快速处理海量卫星影像的遥感与城市规划场景
卫星广泛用于地表覆盖估计与监测,应对气候变化挑战。高分辨率影像有助于识别更小的地物并分类地表类型。近年来小卫星因成本低而流行,但其空间分辨率较弱。本文提出一种计算高效的引导式去噪扩散模型(DDM),用于提升低质量卫星图像分辨率。基于随机微分方程的去噪通常需数百次迭代,本文采用确定性常微分方程(ODE)方法,通过教师-学生蒸馏将Stable Diffusion中的DDM转换为一致性模型(CM),实现快速去噪。实验使用DOTA v2.0数据集,该数据集用于城市规划和地表覆盖估计。所提方法使卫星图像分辨率提升16倍,计算时间减少20倍;低分辨率图像的FID得分从10.0降至1.9,显著改善生成质量。
原文摘要 · Abstract (English)
Satellites are widely used to estimate and monitor ground cover, providing critical information to address the challenges posed by climate change. High-resolution satellite images help to identify smaller features on the ground and classification of ground cover types. Small satellites have become very popular recently due to their cost-effectiveness. However, smaller satellites have weaker spatial resolution, and preprocessing using recent generative models made it possible to enhance the resolution of these satellite images. The objective of this paper is to propose computationally efficient guided or image-conditioned denoising diffusion models (DDMs) to perform super-resolution on low-quality images. Denoising based on stochastic ordinary differential equations (ODEs) typically takes hundreds of iterations and it can be reduced using deterministic ODEs. I propose Consistency Models (CM) that utilize deterministic ODEs for efficient denoising and perform super resolution on satellite images. The DOTA v2.0 image dataset that is used to develop object detectors needed for urban planning and ground cover estimation, is used in this project. The Stable Diffusion model is used as the base model, and the DDM in Stable Diffusion is converted into a Consistency Model (CM) using Teacher-Student Distillation to apply deterministic denoising. Stable diffusion with modified CM has successfully improved the resolution of satellite images by a factor of 16, and the computational time was reduced by a factor of 20 compared to stochastic denoising methods. The FID score of low-resolution images improved from 10.0 to 1.9 after increasing the image resolution using my algorithm for consistency models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。