对比预训练与从零训练,发现任务相似时预训练才更有效。
How Effective is Pre-training of Large Masked Autoencoders for Downstream Earth Observation Tasks?
- 用掩码自编码器预训练视觉变换器模型
- 重建任务中预训练显著提升效果,分类/分割则相近或更差
- 适合关注遥感图像重建的科研人员
自监督预训练在计算机视觉中表现优异,尤其在标注数据稀缺时。在遥感领域,基于视觉变换器(ViT)的模型已成功用于迁移学习。然而,预训练模型在何种条件下优于从零训练仍不明确。本研究评估了两种大型ViT-based掩码自编码器(MAE)模型——基础模型Prithvi和SatMAE——在下游遥感任务中的表现,涵盖重建、分割和分类。实验发现:当微调任务与预训练任务相似(如重建)时,预训练优势明显;而对分割或分类任务,通过调整超参数从零训练反而达到相当或更优效果。
原文摘要 · Abstract (English)
Self-supervised pre-training has proven highly effective for many computer vision tasks, particularly when labelled data are scarce. In the context of Earth Observation (EO), foundation models and various other Vision Transformer (ViT)-based approaches have been successfully applied for transfer learning to downstream tasks. However, it remains unclear under which conditions pre-trained models offer significant advantages over training from scratch. In this study, we investigate the effectiveness of pre-training ViT-based Masked Autoencoders (MAE) for downstream EO tasks, focusing on reconstruction, segmentation, and classification. We consider two large ViT-based MAE pre-trained models: a foundation model (Prithvi) and SatMAE. We evaluate Prithvi on reconstruction and segmentation-based downstream tasks, and for SatMAE we assess its performance on a classification downstream task. Our findings suggest that pre-training is particularly beneficial when the fine-tuning task closely resembles the pre-training task, e.g. reconstruction. In contrast, for tasks such as segmentation or classification, training from scratch with specific hyperparameter adjustments proved to be equally or more effective.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。