arXiv:2501.01733cs.CVcs.AI2025-01中稿 · publication as a r…被引 8

用混合贴图增强数据,提升嘈杂标注下X光违禁品检测准确率。

Augmentation Matters: A Mix-Paste Method for X-Ray Prohibited Item Detection under Noisy Annotations

  • 从数据增强角度设计标签感知的混合贴图方法
  • 在噪声标注下检测精度显著提升,且泛化至MS-COCO
  • 适合处理真实场景中标注混乱的安检图像任务

自动X光违禁品检测对公共安全至关重要。现有深度学习方法均假设训练图像标注正确,但大规模X光图像中物品重叠普遍,获取准确标注极为困难,导致图像常被噪声标注污染,进而损害模型性能。本文从数据增强新视角出发,提出一种有效的标签感知混合贴图增强方法(Mix-Paste)。具体地,对每个物品区域,从不同图像中选取同类别多个物品块进行混合,并替换原图像中的原始块,从而提高生成图像中包含正确违禁品的概率。该混合过程模拟了物品重叠现象,帮助模型学习X光图像特征。此外,设计基于物品的大损失抑制策略(LLS),以压制因混合操作引入额外物品时产生的大损失。在含噪声的X光数据集上验证了方法优越性;同时在噪声版MS-COCO数据集上评估其泛化能力。结果表明,数据增强在应对标注噪声方面具有巨大潜力。源代码已开源:https://github.com/wscds/Mix-Paste。

原文摘要 · Abstract (English)

Automatic X-ray prohibited item detection is vital for public safety. Existing deep learning-based methods all assume that the annotations of training X-ray images are correct. However, obtaining correct annotations is extremely hard if not impossible for large-scale X-ray images, where item overlapping is ubiquitous.As a result, X-ray images are easily contaminated with noisy annotations, leading to performance deterioration of existing methods.In this paper, we address the challenging problem of training a robust prohibited item detector under noisy annotations (including both category noise and bounding box noise) from a novel perspective of data augmentation, and propose an effective label-aware mixed patch paste augmentation method (Mix-Paste). Specifically, for each item patch, we mix several item patches with the same category label from different images and replace the original patch in the image with the mixed patch. In this way, the probability of containing the correct prohibited item within the generated image is increased. Meanwhile, the mixing process mimics item overlapping, enabling the model to learn the characteristics of X-ray images. Moreover, we design an item-based large-loss suppression (LLS) strategy to suppress the large losses corresponding to potentially positive predictions of additional items due to the mixing operation. We show the superiority of our method on X-ray datasets under noisy annotations. In addition, we evaluate our method on the noisy MS-COCO dataset to showcase its generalization ability. These results clearly indicate the great potential of data augmentation to handle noise annotations. The source code is released at https://github.com/wscds/Mix-Paste.

X光检测数据增强噪声标注目标检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。