arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.08056cs.CV

标签噪声下航空图像中的鲁棒微小目标检测

Robust Tiny Object Detection in Aerial Images amidst Label Noise

Wuhan University(武汉大学)

Haoran Zhu, Chang Xu, Wen Yang, Ruixiang Zhang, Yan Zhang, Gui-Song Xia

AI总结:

针对航空图像微小目标检测中标签噪声导致性能下降的问题,本文提出DN-TOD,通过类别感知标签校正和趋势引导学习策略分别处理类别偏移与边界框噪声,并在多个含噪数据集上验证了其鲁棒性和有效性。

摘要:

遥感图像中微小目标的精确检测仍然是一个重大挑战,原因在于其视觉信息有限且经常出现在复杂场景中。这一挑战进一步被人工标注的实际负担和固有错误所加剧:标注微小目标既费力又容易出错(即标签噪声)。使用含噪声标签训练此类目标的检测器通常会导致性能欠佳,网络往往会过拟合噪声标签。在本研究中,我们解决了噪声标签监督下微小目标检测这一复杂问题。我们系统地研究了不同类型噪声对网络训练的影响,揭示了目标检测器对类别偏移和微小目标不准确边界框的脆弱性。为缓解这些挑战,我们提出了一种去噪微小目标检测器(DN-TOD),其中包含类别感知标签校正(CLC)方案以解决类别偏移问题,以及趋势引导学习策略(TLS)以处理边界框噪声。CLC通过识别并过滤类别偏移的正样本,缓解不准确的类别监督;TLS则通过样本重加权和边界框再生成,减少由噪声框引起的错误监督。此外,我们的方法可以无缝集成到一阶段和两阶段目标检测流程中。在合成数据集(即含噪AI-TOD-v2.0和DOTA-v2.0)和真实世界含噪数据集(即AI-TOD)上进行的综合实验证明了DN-TOD在各种类型标签噪声下的鲁棒性。值得注意的是,当应用于强基线RFLA时,DN-TOD在40%混合噪声下表现出4.9个点的显著性能提升。数据集、代码和模型将公开发布。

英文摘要:

Precise detection of tiny objects in remote sensing imagery remains a significant challenge due to their limited visual information and frequent occurrence within scenes. This challenge is further exacerbated by the practical burden and inherent errors associated with manual annotation: annotating tiny objects is laborious and prone to errors (i.e., label noise). Training detectors for such objects using noisy labels often leads to suboptimal performance, with networks tending to overfit on noisy labels. In this study, we address the intricate issue of tiny object detection under noisy label supervision. We systematically investigate the impact of various types of noise on network training, revealing the vulnerability of object detectors to class shifts and inaccurate bounding boxes for tiny objects. To mitigate these challenges, we propose a DeNoising Tiny Object Detector (DN-TOD), which incorporates a Class-aware Label Correction (CLC) scheme to address class shifts and a Trend-guided Learning Strategy (TLS) to handle bounding box noise. CLC mitigates inaccurate class supervision by identifying and filtering out class-shifted positive samples, while TLS reduces noisy box-induced erroneous supervision through sample reweighting and bounding box regeneration. Additionally, Our method can be seamlessly integrated into both one-stage and two-stage object detection pipelines. Comprehensive experiments conducted on synthetic (i.e., noisy AI-TOD-v2.0 and DOTA-v2.0) and real-world (i.e., AI-TOD) noisy datasets demonstrate the robustness of DN-TOD under various types of label noise. Notably, when applied to the strong baseline RFLA, DN-TOD exhibits a noteworthy performance improvement of 4.9 points under 40% mixed noise. Datasets, codes, and models will be made publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑