English

DiffusionSat: A Generative Foundation Model for Satellite Imagery

Computer Vision and Pattern Recognition 2024-05-28 v2 Artificial Intelligence Machine Learning

Abstract

Diffusion models have achieved state-of-the-art results on many modalities including images, speech, and video. However, existing models are not tailored to support remote sensing data, which is widely used in important applications including environmental monitoring and crop-yield prediction. Satellite images are significantly different from natural images -- they can be multi-spectral, irregularly sampled across time -- and existing diffusion models trained on images from the Web do not support them. Furthermore, remote sensing data is inherently spatio-temporal, requiring conditional generation tasks not supported by traditional methods based on captions or images. In this paper, we present DiffusionSat, to date the largest generative foundation model trained on a collection of publicly available large, high-resolution remote sensing datasets. As text-based captions are sparsely available for satellite images, we incorporate the associated metadata such as geolocation as conditioning information. Our method produces realistic samples and can be used to solve multiple generative tasks including temporal generation, superresolution given multi-spectral inputs and in-painting. Our method outperforms previous state-of-the-art methods for satellite image generation and is the first large-scale generative foundation model for satellite imagery. The project website can be found here: https://samar-khanna.github.io/DiffusionSat/

Keywords

Cite

@article{arxiv.2312.03606,
  title  = {DiffusionSat: A Generative Foundation Model for Satellite Imagery},
  author = {Samar Khanna and Patrick Liu and Linqi Zhou and Chenlin Meng and Robin Rombach and Marshall Burke and David Lobell and Stefano Ermon},
  journal= {arXiv preprint arXiv:2312.03606},
  year   = {2024}
}

Comments

Published at ICLR 2024

R2 v1 2026-06-28T13:42:59.354Z