arXiv:2602.14040cs.CV2026-02

用可解释性方法精准剪枝,让目标检测模型更轻更快还更准。

Explainability-Inspired Layer-Wise Pruning of Deep Neural Networks for Efficient Object Detection

  • 基于梯度-激活归因评估每层重要性,比传统剪枝更贴合实际功能贡献。
  • 在ShuffleNetV2上提速10%,而传统方法反而降性能13.7%。
  • 适合追求高效部署且重视模型可解释性的边缘计算研究者。

深度神经网络在目标检测中表现卓越,但其复杂性给资源受限平台部署带来挑战。传统基于权重大小的剪枝方法未必反映网络组件对任务的实际贡献。本文提出一种受可解释性启发的分层剪枝框架,利用类SHAP的梯度-激活归因来估计各层重要性,提供数据驱动的功能贡献代理,而非依赖静态权重大小。我们在ResNet-50、MobileNetV2、ShuffleNetV2、Faster R-CNN、RetinaNet和YOLOv8等多种检测架构上进行实验,评估在Microsoft COCO 2017验证集上的表现。结果表明,该方法相比L1范数剪枝能更准确识别低重要性层,实现更好的精度-效率权衡。例如,在ShuffleNetV2上,本方法实现10%的推理速度提升,而L1剪枝导致性能下降13.7%;在RetinaNet上,本方法保持基线mAP(0.151)几乎不变,而L1剪枝造成1.3% mAP下降,同时仅获得6.2%的速度提升。这些发现凸显数据驱动层重要性评估的重要性,并证明可解释性引导的压缩为边缘与资源受限平台部署提供了更可靠的方向,兼顾性能与可解释性。

原文摘要 · Abstract (English)

Deep neural networks (DNNs) have achieved remarkable success in object detection tasks, but their increasing complexity poses significant challenges for deployment on resource-constrained platforms. While model compression techniques such as pruning have emerged as essential tools, traditional magnitude-based pruning methods do not necessarily align with the true functional contribution of network components to task-specific performance. In this work, we present an explainability-inspired, layer-wise pruning framework tailored for efficient object detection. Our approach leverages a SHAP-inspired gradient--activation attribution to estimate layer importance, providing a data-driven proxy for functional contribution rather than relying solely on static weight magnitudes. We conduct comprehensive experiments across diverse object detection architectures, including ResNet-50, MobileNetV2, ShuffleNetV2, Faster R-CNN, RetinaNet, and YOLOv8, evaluating performance on the Microsoft COCO 2017 validation set. The results show that the proposed attribution-inspired pruning consistently identifies different layers as least important compared to L1-norm-based methods, leading to improved accuracy--efficiency trade-offs. Notably, for ShuffleNetV2, our method yields a 10\% empirical increase in inference speed, whereas L1-pruning degrades performance by 13.7\%. For RetinaNet, the proposed approach preserves the baseline mAP (0.151) with negligible impact on inference speed, while L1-pruning incurs a 1.3\% mAP drop for a 6.2\% speed increase. These findings highlight the importance of data-driven layer importance assessment and demonstrate that explainability-inspired compression offers a principled direction for deploying deep neural networks on edge and resource-constrained platforms while preserving both performance and interpretability.

模型剪枝目标检测可解释性边缘计算

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