HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

DDPM-CD: Denoising Diffusion Probabilistic Models as Feature Extractors for Change Detection

Wele Gedara Chaminda Bandara Nithin Gopalakrishnan Nair Vishal M. Patel

DDPM-CD: Denoising Diffusion Probabilistic Models as Feature Extractors for Change Detection

Abstract

Remote sensing change detection is crucial for understanding the dynamics of our planet's surface, facilitating the monitoring of environmental changes, evaluating human impact, predicting future trends, and supporting decision-making. In this work, we introduce a novel approach for change detection that can leverage off-the-shelf, unlabeled remote sensing images in the training process by pre-training a Denoising Diffusion Probabilistic Model (DDPM) - a class of generative models used in image synthesis. DDPMs learn the training data distribution by gradually converting training images into a Gaussian distribution using a Markov chain. During inference (i.e., sampling), they can generate a diverse set of samples closer to the training distribution, starting from Gaussian noise, achieving state-of-the-art image synthesis results. However, in this work, our focus is not on image synthesis but on utilizing it as a pre-trained feature extractor for the downstream application of change detection. Specifically, we fine-tune a lightweight change classifier utilizing the feature representations produced by the pre-trained DDPM alongside change labels. Experiments conducted on the LEVIR-CD, WHU-CD, DSIFN-CD, and CDD datasets demonstrate that the proposed DDPM-CD method significantly outperforms the existing state-of-the-art change detection methods in terms of F1 score, IoU, and overall accuracy, highlighting the pivotal role of pre-trained DDPM as a feature extractor for downstream applications. We have made both the code and pre-trained models available at https://github.com/wgcban/ddpm-cd

Code Repositories

wgcban/ddpm-cd
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
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

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp