提升遥感少样本目标检测的泛化能力,解决小样本下识别不准问题。
Generalization-Enhanced Few-Shot Object Detection in Remote Sensing
- 融合多尺度特征与注意力机制,增强复杂场景下的特征表达。
- 在DIOR和NWPU VHR-10数据集上达到当前最优性能,少样本检测准确率显著提升。
- 适合遥感图像中罕见或新类别目标检测,对标注数据少的场景特别实用。
遥感目标检测因影像高分辨率、多尺度特征及地物类型多样而极具挑战。尽管深度学习已取得显著进展,但通常依赖大量标注数据。获取新类或稀有类的充足标注数据在遥感场景中既困难又耗时,限制了模型泛化能力。为此,少样本学习(FSL)成为可行方案,其目标是从少量标注样本中学习新类别。本文提出通用增强型少样本目标检测(GE-FSOD)模型,针对遥感环境下的泛化瓶颈,引入三项创新:跨层级融合金字塔注意力网络(CFPAN)以增强多尺度特征表示;多阶段精修区域建议网络(MRRPN)提升区域建议精度;广义分类损失(GCL)改善少样本条件下的分类性能。在DIOR与NWPU VHR-10数据集上的大量实验表明,该模型在遥感少样本目标检测任务中达到当前最优水平。
原文摘要 · Abstract (English)
Remote sensing object detection is particularly challenging due to the high resolution, multi-scale features, and diverse ground object characteristics inherent in satellite and UAV imagery. These challenges necessitate more advanced approaches for effective object detection in such environments. While deep learning methods have achieved remarkable success in remote sensing object detection, they typically rely on large amounts of labeled data. Acquiring sufficient labeled data, particularly for novel or rare objects, is both challenging and time-consuming in remote sensing scenarios, limiting the generalization capabilities of existing models. To address these challenges, few-shot learning (FSL) has emerged as a promising approach, aiming to enable models to learn new classes from limited labeled examples. Building on this concept, few-shot object detection (FSOD) specifically targets object detection challenges in data-limited conditions. However, the generalization capability of FSOD models, particularly in remote sensing, is often constrained by the complex and diverse characteristics of the objects present in such environments. In this paper, we propose the Generalization-Enhanced Few-Shot Object Detection (GE-FSOD) model to improve the generalization capability in remote sensing FSOD tasks. Our model introduces three key innovations: the Cross-Level Fusion Pyramid Attention Network (CFPAN) for enhanced multi-scale feature representation, the Multi-Stage Refinement Region Proposal Network (MRRPN) for more accurate region proposals, and the Generalized Classification Loss (GCL) for improved classification performance in few-shot scenarios. Extensive experiments on the DIOR and NWPU VHR-10 datasets show that our model achieves state-of-the-art performance for few-shot object detection in remote sensing.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。