提出新模型OSAD,让雷达图像检测器能识别未知飞机并保持原有性能。
OSAD: Open-Set Aircraft Detection in SAR Images
- 通过全局上下文建模、位置驱动伪标签和原型对比学习三组件提升泛化能力。
- 在未知目标检测上平均精度最高提升18.36%,闭集性能不受影响。
- 适合需要应对未知目标的军事侦察、遥感监测等开放环境应用。
当前主流合成孔径雷达(SAR)图像目标检测方法在开放环境中对未知物体的鲁棒性不足。开集检测旨在使在封闭集上训练的检测器既能识别所有已知目标,又能发现开放环境中的未知目标。核心挑战在于如何提升对潜在未知目标的泛化能力,同时降低在强监督下已知类别的经验分类风险。为此,本文提出一种面向SAR图像的开集飞机检测新模型OSAD,包含三个专用组件:全局上下文建模(GCM)、位置质量驱动伪标签生成(LPG)和原型对比学习(PCL)。GCM通过捕捉长序列位置关系生成注意力图,增强网络对物体的表征能力;LPG利用目标位置与形状线索优化定位质量,避免过度拟合已知类别信息,提升对潜在未知目标的泛化能力;PCL采用基于原型的对比编码损失,促进实例级类内紧凑性和类间差异性,以最小化已知与未知分布的重叠,降低已知类别的经验分类风险。大量实验表明,所提方法能有效检测未知目标,在不牺牲闭集性能的前提下实现竞争力表现,未知目标平均精度最高提升18.36%。
原文摘要 · Abstract (English)
Current mainstream SAR image object detection methods still lack robustness when dealing with unknown objects in open environments. Open-set detection aims to enable detectors trained on a closed set to detect all known objects and identify unknown objects in open-set environments. The key challenges are how to improve the generalization to potential unknown objects and reduce the empirical classification risk of known categories under strong supervision. To address these challenges, a novel open-set aircraft detector for SAR images is proposed, named Open-Set Aircraft Detection (OSAD), which is equipped with three dedicated components: global context modeling (GCM), location quality-driven pseudo labeling generation (LPG), and prototype contrastive learning (PCL). GCM effectively enhances the network's representation of objects by attention maps which is formed through the capture of long sequential positional relationships. LPG leverages clues about object positions and shapes to optimize localization quality, avoiding overfitting to known category information and enhancing generalization to potential unknown objects. PCL employs prototype-based contrastive encoding loss to promote instance-level intra-class compactness and inter-class variance, aiming to minimize the overlap between known and unknown distributions and reduce the empirical classification risk of known categories. Extensive experiments have demonstrated that the proposed method can effectively detect unknown objects and exhibit competitive performance without compromising closed-set performance. The highest absolute gain which ranges from 0 to 18.36% can be achieved on the average precision of unknown objects.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。