iScience. 2024 Dec 17;28(1):111618. doi: 10.1016/j.isci.2024.111618. eCollection 2025 Jan 17.
ABSTRACT
Achieving lightweight real-time object detection necessitates balancing model compression with detection accuracy, a difficulty exacerbated by low redundancy and uneven contributions from convolutional layers. As an alternative to traditional methods, we propose Rigorous Gradation Pruning (RGP), which uses a desensitized first-order Taylor approximation to assess filter importance, enabling precise pruning of redundant kernels. This approach includes the iterative reassessment of layer significance to protect essential layers, ensuring effective detection performance. We applied RGP to YOLOv8 detectors and tested it on GTSDB, Seaships, and COCO datasets. On GTSDB, RGP achieved 80% compression of YOLOv8n with only a 0.11% drop in mAP0.5, while increasing frames per second (FPS) by 43.84%. For YOLOv8x, RGP achieved 90% compression, a 1.26% mAP0.5:0.95 increase, and a 112.66% FPS boost. Significant compression was also achieved on Seaships and COCO datasets, demonstrating RGP's robustness across diverse object detection tasks and its potential for advancing efficient, high-speed detection models.
PMID:39834872 | PMC:PMC11743878 | DOI:10.1016/j.isci.2024.111618