3 个月前

DDPM-CD:去噪扩散概率模型作为变化检测的特征提取器

DDPM-CD:去噪扩散概率模型作为变化检测的特征提取器

摘要

遥感变化检测对于理解地球表面动态变化至关重要,有助于环境变化监测、人类活动影响评估、未来趋势预测以及决策支持。本文提出一种新颖的变化检测方法,该方法在训练过程中可利用现成的、未标注的遥感图像,通过预训练一种去噪扩散概率模型(Denoising Diffusion Probabilistic Model, DDPM)来实现。DDPM是一类用于图像合成的生成模型,其通过马尔可夫链逐步将训练图像转换为高斯分布,从而学习训练数据的分布特性。在推理阶段(即采样过程)中,该模型可从高斯噪声出发,生成一组与训练分布更为接近的多样化图像样本,从而实现当前最先进的图像合成效果。然而,在本研究中,我们的关注点并非图像合成本身,而是将预训练的DDPM作为下游任务——变化检测的特征提取器。具体而言,我们利用预训练DDPM所生成的特征表示,并结合变化标签,对一个轻量级变化分类器进行微调。在LEVIR-CD、WHU-CD、DSIFN-CD和CDD等多个公开数据集上的实验结果表明,所提出的DDPM-CD方法在F1分数、交并比(IoU)和总体准确率等关键指标上均显著优于现有的最先进变化检测方法,充分凸显了预训练DDPM作为下游应用特征提取器的重要价值。相关代码与预训练模型已开源,可访问 https://github.com/wgcban/ddpm-cd 获取。

代码仓库

wgcban/ddpm-cd
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
change-detection-on-cdd-dataset-season-1DDPM-CD
F1-Score: 95.62
Overall Accuracy: 98.98
change-detection-on-dsifn-cdDDPM-CD
F1: 96.65
Overall Accuracy: 97.09
change-detection-on-levir-cdDDPM-CD
F1: 90.91
Overall Accuracy: 99.09
change-detection-on-whu-cdDDPM-CD
F1: 92.65
Overall Accuracy: 99.42

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DDPM-CD:去噪扩散概率模型作为变化检测的特征提取器 | 论文 | HyperAI超神经