轻量级框架GSA-YOLO实现毫秒级安检检测,兼顾精度与速度。
GSA-YOLO: A High-Efficiency Framework via Structured Sparsity and Adaptive Knowledge Distillation for Real-Time X-ray Security Inspection

- 融合结构化稀疏与自适应知识蒸馏,优化特征提取与模型压缩。
- 在HiXray和PIDray上达53.1%和67.9%的mAP50:95,速度189.62 FPS。
- 适合高实时性、低算力场景下的工业级安检应用。
X射线安检需在严重遮挡、复杂杂乱环境下实现禁限物品的精准实时检测,现有模型难以兼顾效率与精度。本文提出GSA-YOLO,基于YOLOv8n架构构建轻量级框架,通过三大核心组件提升检测鲁棒性与推理效率:在网络颈部引入组套索(GL)增强特征提取;在检测头采用稀疏结构选择(SSS)大幅剪枝;并设计自适应知识蒸馏(Ada-KD)机制恢复精度。该方法协同优化表示能力与冗余通道剪除,在HiXray与PIDray数据集上分别取得0.531与0.679的mAP50:95,较基线提升2.4%和1.8%,同时推理速度达189.62 FPS,计算量由8.7G降至8.0G,兼具高效与高精度,适用于实际安检场景。
原文摘要 · Abstract (English)
X-ray security inspection requires accurate real-time detection of prohibited items, but existing models often struggle to balance the challenges of severe occlusion, complex clutter, and strict speed requirements. To overcome these challenges, this paper proposes GSA-YOLO, a novel lightweight framework built upon the YOLOv8n architecture, specifically engineered to enhance detection robustness and inference efficiency. GSA-YOLO strategically integrates structured sparsity and adaptive knowledge transfer through three core components: Group Lasso (GL) applied to the network neck for robust feature extraction; Sparse Structure Selection (SSS) applied to the detection head for significant model slimming; and an Adaptive Knowledge Distillation (Ada-KD) mechanism for comprehensive accuracy recovery. This integrated approach synergistically enhances feature representation while pruning redundant channels, maximizing model efficiency without sacrificing performance. Rigorous evaluations on the HiXray and PIDray datasets confirm GSA-YOLO's comprehensive capability, achieving a leading inference speed of 189.62 FPS, accompanied by a reduction in computational cost from 8.7G to 8.0G. Crucially, GSA-YOLO secures mAP50:95 results of 0.531 and 0.679 on HiXray and PIDray, demonstrating 2.4% and 1.8% improvements over the baseline, respectively. Compared to other models, GSA-YOLO exhibits enhanced accuracy while maintaining computational efficiency, making it a promising solution for practical X-ray security inspection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。