arXiv:2509.12918cs.CV2025-09被引 6

压缩YOLOv8模型,实现边缘设备实时航拍目标检测

A Novel Compression Framework for YOLOv8: Achieving Real-Time Aerial Object Detection on Edge Devices via Structured Pruning and Channel-Wise Distillation

  • 三阶段压缩:动态稀疏训练+结构化通道剪枝+通道级知识蒸馏
  • 参数减少73.5%,推理速度提升至68帧/秒,精度损失仅2.7%
  • 适合资源受限场景的实时目标检测应用,尤其航拍无人机

在资源受限设备上高效部署用于航拍目标检测的深度学习模型,需大幅压缩模型且不牺牲性能。本文提出一种针对YOLOv8的三阶段压缩框架,融合稀疏感知训练、结构化通道剪枝与通道级知识蒸馏(CWD)。首先,稀疏感知训练在优化过程中引入动态稀疏性,平衡参数量与检测精度。其次,利用批量归一化缩放因子进行结构化通道剪枝,显著降低模型大小与计算复杂度。最后,为缓解剪枝带来的精度下降,采用可调温度与损失权重的CWD方法,特别优化小中目标检测。在VisDrone数据集上的实验表明,对YOLOv8m,参数从25.85M降至6.85M(减少73.51%),FLOPs由49.6G降至13.3G,MACs由101G降至34.5G,AP50仅下降2.7%。压缩后模型达到47.9 AP50,推理速度从26 FPS提升至45 FPS。进一步使用TensorRT轻量化优化,虽使AP50微降至47.6,但推理速度达68 FPS,验证了该方法在高吞吐、低资源场景下的实用性。

原文摘要 · Abstract (English)

Efficient deployment of deep learning models for aerial object detection on resource-constrained devices requires significant compression without com-promising performance. In this study, we propose a novel three-stage compression pipeline for the YOLOv8 object detection model, integrating sparsity-aware training, structured channel pruning, and Channel-Wise Knowledge Distillation (CWD). First, sparsity-aware training introduces dynamic sparsity during model optimization, effectively balancing parameter reduction and detection accuracy. Second, we apply structured channel pruning by leveraging batch normalization scaling factors to eliminate redundant channels, significantly reducing model size and computational complexity. Finally, to mitigate the accuracy drop caused by pruning, we employ CWD to transfer knowledge from the original model, using an adjustable temperature and loss weighting scheme tailored for small and medium object detection. Extensive experiments on the VisDrone dataset demonstrate the effectiveness of our approach across multiple YOLOv8 variants. For YOLOv8m, our method reduces model parameters from 25.85M to 6.85M (a 73.51% reduction), FLOPs from 49.6G to 13.3G, and MACs from 101G to 34.5G, while reducing AP50 by only 2.7%. The resulting compressed model achieves 47.9 AP50 and boosts inference speed from 26 FPS (YOLOv8m baseline) to 45 FPS, enabling real-time deployment on edge devices. We further apply TensorRT as a lightweight optimization step. While this introduces a minor drop in AP50 (from 47.9 to 47.6), it significantly improves inference speed from 45 to 68 FPS, demonstrating the practicality of our approach for high-throughput, re-source-constrained scenarios.

目标检测模型压缩边缘计算YOLOv8

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