针对X光安检中物品检测的长尾分布问题,提出新框架提升小类物品识别效果。
PAD-F: Prior-Aware Debiasing Framework for Long-Tailed X-ray Prohibited Item Detection
- 利用材料吸收率和梯度泊松融合生成难样本,增强尾部类别数据
- 通过隐式共现聚合模块提升模糊物品特征表达,尾部AP50提升17.2%
- 适用于安检、医疗等需精准识别稀有目标的场景
在X光安检图像中检测违禁物品是一项关键且具有挑战性的任务。随着深度学习的发展,目标检测算法已广泛应用于该领域。然而,真实场景中违禁物品类别分布呈现显著的长尾特征。由于X光成像原理特殊,传统长尾检测方法在此领域效果有限。为此,我们提出先验感知去偏框架PAD-F,采用双路径策略,结合材料先验与共现先验。在数据层面,通过基于材料吸收率的放置策略与梯度引导的泊松融合技术,构建大量难样本以增强尾部类别训练。在特征层面,设计可插拔的隐式共现聚合模块(ICA),通过学习图像内统计共现关系,增强模糊物体特征表示。在HiXray与PIDray数据集上的大量实验表明,PAD-F显著提升多个主流检测器性能,尾部类别AP50最高提升17.2%,全面超越现有最优方法。本工作为解决X光安检中的长尾检测难题提供了有效且通用的解决方案。
原文摘要 · Abstract (English)
Detecting prohibited items in X-ray security imagery is a challenging yet crucial task. With the rapid advancement of deep learning, object detection algorithms have been widely applied in this area. However, the distribution of object classes in real-world prohibited item detection scenarios often exhibits a distinct long-tailed distribution. Due to the unique principles of X-ray imaging, conventional methods for long-tailed object detection are often ineffective in this domain. To tackle these challenges, we introduce the Prior-Aware Debiasing Framework (PAD-F), a novel approach that employs a two-pronged strategy leveraging both material and co-occurrence priors. At the data level, our Explicit Material-Aware Augmentation (EMAA) component generates numerous challenging training samples for tail classes. It achieves this through a placement strategy guided by material-specific absorption rates and a gradient-based Poisson blending technique. At the feature level, the Implicit Co-occurrence Aggregator (ICA) acts as a plug-in module that enhances features for ambiguous objects by implicitly learning and aggregating statistical co-occurrence relationships within the image. Extensive experiments on the HiXray and PIDray datasets demonstrate that PAD-F significantly boosts the performance of multiple popular detectors. It achieves an absolute improvement of up to +17.2% in AP50 for tail classes and comprehensively outperforms existing state-of-the-art methods. Our work provides an effective and versatile solution to the critical problem of long-tailed detection in X-ray security.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。