3 个月前

在遥感场景分类中,我们是否仍需依赖ImageNet预训练?

在遥感场景分类中,我们是否仍需依赖ImageNet预训练?

摘要

由于标注数据的稀缺性,使用在ImageNet上预训练的监督模型已成为遥感场景分类领域的事实标准。然而,近年来高分辨率遥感(HRRS)图像数据集的规模不断扩大,自监督学习技术也取得了显著进展,这引发了两个关键问题:在遥感场景分类任务中,是否仍有必要依赖ImageNet上的监督预训练?相较于在ImageNet上进行监督预训练,使用HRRS图像数据集进行监督预训练,或在ImageNet上进行自监督预训练,哪种方式能在目标遥感场景分类任务上取得更优性能?为回答上述问题,本文在多个HRRS图像数据集上,系统地进行了从零开始训练模型、微调在ImageNet上预训练的监督模型以及微调自监督学习模型的实验。同时,我们评估了所学表征在HRRS场景分类任务中的迁移能力,结果表明:自监督预训练的性能优于监督预训练,而HRRS数据集上的监督预训练表现与自监督预训练相当,或略低。最终,本文提出一种“领域自适应预训练”策略,即先使用ImageNet预训练模型,再在领域内HRRS图像上进行第二轮预训练。实验结果表明,该方法所得到的模型在HRRS场景分类基准测试中达到了当前最优水平。相关源代码及预训练模型已公开,可访问:\url{https://github.com/risojevicv/RSSC-transfer}。

代码仓库

risojevicv/rssc-transfer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
multi-label-classification-on-mlrsnetResNet50 (scratch)
F1-score: 91.83
multi-label-classification-on-mlrsnetResNet50 (fine-tuning)
F1-score: 92.41

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
在遥感场景分类中,我们是否仍需依赖ImageNet预训练? | 论文 | HyperAI超神经