Skip to main navigation Skip to search Skip to main content

Multilabel-Guided Soft Contrastive Learning for Efficient Earth Observation Pretraining

  • Technical University of Munich
  • Deutsches Zentrum für Luft- und Raumfahrt e.V. (DLR)

Research output: Contribution to journalArticlepeer-review

11 Scopus citations

Abstract

Self-supervised pretraining on large-scale satellite data has raised great interest in building Earth observation (EO) foundation models. However, many important resources beyond pure satellite imagery, such as land-cover-land-use products that provide free global semantic information, as well as vision foundation models that hold strong knowledge of the natural world, are not widely studied. In this work, we show these free additional resources not only help resolve common contrastive learning bottlenecks but also significantly boost the efficiency and effectiveness of EO pretraining. Specifically, we first propose soft contrastive learning (SoftCon) that optimizes cross-scene soft similarity based on land-cover-generated multilabel supervision, naturally solving the issue of multiple positive samples and too strict positive matching in complex scenes. Second, we revisit and explore cross-domain continual pretraining for both multispectral and synthetic aperture radar (SAR) imagery, building efficient EO foundation models from strongest vision models such as DINOv2. Adapting simple weight-initialization and Siamese masking strategies into our SoftCon framework, we demonstrate impressive continual pretraining performance even when the input modalities are not aligned. Without prohibitive training, we produce multispectral and SAR foundation models that achieve significantly better results in 10 out of 11 downstream tasks than most existing SOTA models. For example, our ResNet50/ViT-S achieve 84.8/85.0 linear probing mAP scores on BigEarthNet-10%, which are better than most existing ViT-L models; under the same setting, our ViT-B sets a new record of 86.8 in multispectral, and 82.5 in SAR, the latter even better than many multispectral models.

Original languageEnglish
Article number5644516
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume62
DOIs
StatePublished - 2024

Keywords

  • Continual pretraining
  • Earth observation (EO)
  • contrastive learning
  • foundation model
  • multispectral
  • remote sensing
  • self-supervised learning
  • synthetic aperture radar (SAR)

Fingerprint

Dive into the research topics of 'Multilabel-Guided Soft Contrastive Learning for Efficient Earth Observation Pretraining'. Together they form a unique fingerprint.

Cite this