用预训练基础模型解决高光谱图像少样本跨域分类难题
Cross-Domain Few-Shot Learning for Hyperspectral Image Classification Based on Mixup Foundation Model
- 基于遥感基础模型,冻结主干网络快速适配下游任务
- 混合域适应与平滑标签,跨域差异大时准确率提升14%
- 适合资源有限的遥感图像分类研究者使用
尽管高光谱图像(HSI)分类中的跨域少样本学习(CDFSL)已受广泛关注,但现有方法常依赖外部噪声进行数据增强,过度简化数据稀缺问题,且模型参数量大,易过拟合。据我们所知,尚未有工作探索具备强泛化能力的基础模型在该任务中的潜力。本文提出MIxup FOundation MOdel(MIFOMO),基于大规模遥感任务预训练的遥感基础模型,具备可迁移特征。引入共熔投影(CP)实现冻结主干网络下的快速适配;提出混合域适应(MDM)应对极端域差异;并采用标签平滑缓解伪标签噪声问题。严格实验表明,MIFOMO相较先前方法最高提升14%。代码已开源:https://github.com/Naeem-Paeedeh/MIFOMO。
原文摘要 · Abstract (English)
Although cross-domain few-shot learning (CDFSL) for hyper-spectral image (HSI) classification has attracted significant research interest, existing works often rely on an unrealistic data augmentation procedure in the form of external noise to enlarge the sample size, thus greatly simplifying the issue of data scarcity. They involve a large number of parameters for model updates, being prone to the overfitting problem. To the best of our knowledge, none has explored the strength of the foundation model, having strong generalization power to be quickly adapted to downstream tasks. This paper proposes the MIxup FOundation MOdel (MIFOMO) for CDFSL of HSI classifications. MIFOMO is built upon the concept of a remote sensing (RS) foundation model, pre-trained across a large scale of RS problems, thus featuring generalizable features. The notion of coalescent projection (CP) is introduced to quickly adapt the foundation model to downstream tasks while freezing the backbone network. The concept of mixup domain adaptation (MDM) is proposed to address the extreme domain discrepancy problem. Last but not least, the label smoothing concept is implemented to cope with noisy pseudo-label problems. Our rigorous experiments demonstrate the advantage of MIFOMO, where it beats prior arts with up to 14% margin. The source code of MIFOMO is open-sourced at https://github.com/Naeem-Paeedeh/MIFOMO for reproducibility and convenient further study.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。