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

基于动态损失衰减的带噪标签遥感图像鲁棒旋转目标检测

Dynamic Loss Decay based Robust Oriented Object Detection on Remote Sensing Images with Noisy Labels

Piesat Information Technology(航天宏图信息技术股份有限公司)

Guozhang Liu, Ting Liu, Mengke Yuan, Tao Pang, Guangxing Yang, Hao Fu, Tao Wang, Tongkui Liao

AI总结:

针对遥感图像检测数据集的类别标签噪声问题,提出基于动态损失衰减的鲁棒旋转目标检测方法,在多个公开数据集和相关赛事中均表现优异。

摘要:

遥感图像中目标的外观模糊、尺度微小且类别细粒度的特性,不可避免地会导致检测数据集的类别标签中存在带噪标注。然而,在当前的旋转遥感目标检测器中,标签噪声的影响及应对方法尚未得到充分研究。为解决这一问题,受深度神经网络在干净样本和带噪样本上呈现的“早期学习”与“记忆化”两阶段学习动态启发,我们提出了一种通过动态损失衰减(DLD)机制实现的鲁棒旋转遥感目标检测方法。具体而言,我们首先观测到被称为EL的早期学习阶段终点,在该节点之后模型开始记忆错误标签,这会显著降低检测精度。其次,在该训练指标的指导下,我们将每个样本的损失按降序排列,并在后续训练轮次中自适应衰减排名前K的最大损失样本(坏样本)的损失,因为这些大损失样本有很高置信度是由错误标签计算产生的。实验结果表明,在带有合成类别标签噪声的HRSC2016、DOTA-v1.0/v2.0等多个公开数据集上测试,该方法取得了优异的抗噪性能。我们的方案还在2023年全国大数据与计算智能挑战赛“基于亚米级遥感影像的细粒度目标检测”带噪标签赛道中获得了第二名。

英文摘要:

The ambiguous appearance, tiny scale, and fine-grained classes of objects in remote sensing imagery inevitably lead to the noisy annotations in category labels of detection dataset. However, the effects and treatments of the label noises are underexplored in modern oriented remote sensing object detectors. To address this issue, we propose a robust oriented remote sensing object detection method through dynamic loss decay (DLD) mechanism, inspired by the two phase ``early-learning'' and ``memorization'' learning dynamics of deep neural networks on clean and noisy samples. To be specific, we first observe the end point of early learning phase termed as EL, after which the models begin to memorize the false labels that significantly degrade the detection accuracy. Secondly, under the guidance of the training indicator, the losses of each sample are ranked in descending order, and we adaptively decay the losses of the top K largest ones (bad samples) in the following epochs. Because these large losses are of high confidence to be calculated with wrong labels. Experimental results show that the method achieves excellent noise resistance performance tested on multiple public datasets such as HRSC2016 and DOTA-v1.0/v2.0 with synthetic category label noise. Our solution also has won the 2st place in the "fine-grained object detection based on sub-meter remote sensing imagery" track with noisy labels of 2023 National Big Data and Computing Intelligence Challenge.

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

↑