3 个月前

SSL4EO-S12:面向地球观测自监督学习的大规模多模态、多时相数据集

SSL4EO-S12:面向地球观测自监督学习的大规模多模态、多时相数据集

摘要

自监督预训练具有在无需人工标注的情况下生成富有表现力表征的巨大潜力。目前地球观测(Earth Observation, EO)领域的大多数预训练方法均基于ImageNet或中等规模的有标签遥感(Remote Sensing, RS)数据集。为此,我们发布了一个全新的无标签遥感数据集SSL4EO-S12(Self-Supervised Learning for Earth Observation - Sentinel-1/2),旨在整合欧洲航天局(ESA)Sentinel-1与Sentinel-2卫星任务所获取的全球范围、多模态、多季节的大型卫星影像语料库。在地球观测应用中,我们验证了SSL4EO-S12在多种自监督学习方法上的有效性,包括MoCo-v2、DINO、MAE和data2vec。基于该数据集预训练所得模型,在下游任务中的表现接近甚至超越了传统监督学习的精度水平。此外,相较于现有数据集,SSL4EO-S12在预训练效果上展现出显著优势。我们已将该数据集、相关源代码及预训练模型公开发布,访问地址为:https://github.com/zhu-xlab/SSL4EO-S12。

代码仓库

zhu-xlab/softcon
pytorch
GitHub 中提及
zhu-xlab/dino-mm
pytorch
GitHub 中提及
zhu-xlab/ssl4eo-s12
官方
pytorch
GitHub 中提及
zhu-xlab/ssl4eo-review
pytorch
GitHub 中提及

基准测试

基准方法指标
multi-label-image-classification-onMoCo-v3 (ViT-S/16, fine tune)
mAP (micro): 89.9
official split: No
multi-label-image-classification-onMAE (ViT-S/16, fine tune)
mAP (micro): 88.9
official split: No
multi-label-image-classification-onMoCo-v2 (ResNet50, fine tune)
mAP (micro): 91.8
official split: No
multi-label-image-classification-on-2MoCov3 (ViT-S/16)
F1 Score: 80.5
mAP (micro): 89.3
multi-label-image-classification-on-2MoCov2 (ResNet50)
F1 Score: 79.8
mAP (micro): 88.7

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SSL4EO-S12:面向地球观测自监督学习的大规模多模态、多时相数据集 | 论文 | HyperAI超神经