面向低空防御的反无人机目标检测模型结构化剪枝与微调方法

    Structured pruning and fine-tuning methods for anti-UAV object detection models in low-altitude defense

    • 针对低空防御场景的实时性检测需求,本文提出了一套面向无人机检测的轻量化结构化剪枝框架。该框架首先通过结构解耦变换,将YOLO26中C2、C2f、C3k2等结构耦合度高的复杂特征提取模块等价转换为通道独立的规整算子,为后续剪枝奠定架构基础;进而集成Random(随机剪枝)、L1(L1 范数剪枝)与LAMP(层自适应幅度剪枝)三种重要性评估策略对卷积通道进行量化评分,并在目标加速比的约束下执行渐进式迭代剪枝,以小比例多步压缩的方式逐步移除冗余通道,最后通过微调训练实现精度重建。实验结果表明,剪枝后的模型仅需1.2 M参数量,较主流轻量模型YOLO26参数量减少近50%,推理速度达到1 355.7 FPS(帧每秒),在所有对比模型中最高。精度方面,本模型的mAP50(交并比阈值为 0.5 时的平均精度)为0.965,与YOLO26的0.966仅相差0.001;mAP50-95(交并比阈值在 0.5~0.95 区间内的平均精度)为0.652,同样仅次于YOLO26。与YOLOv5、YOLOv8、YOLOv11及YOLOv12等模型相比,本模型在参数量大幅降低的前提下,精度与速度均具有显著优势,验证了所提剪枝方法的有效性与实用性。

       

      Abstract: The paper proposes a lightweight structured pruning framework tailored for anti-drone scenarios. The framework first employs structural decoupling transformations to equivalently convert highly coupled complex feature extraction modules in YOLO26(i.e., C2, C2f and C3k2) into channel-independent regularization operators, laying the architectural foundation for subsequent pruning; it then integrates three importance evaluation strategies—Random, L1, and LAMP—to assign quantitative scores to convolutional channels, and performs progressive iterative pruning under the constraint of a target acceleration ratio, gradually removing redundant channels through multiple steps of small-scale compression; finally, its accuracy is restored through fine-tuning. Experimental results show that the pruned model requires only 1.2 million parameters, a reduction of nearly 50% compared to the mainstream lightweight model YOLO26, and achieves an inference speed of 1 355.7 FPS, the highest among all compared models. In terms of accuracy, the model's mAP50 is 0.965, differing from YOLO26's 0.966 by only 0.001; its mAP50-95 is 0.652, also second only to YOLO26. Compared to models such as YOLOv5, YOLOv8, YOLOv11, and YOLOv12, this model demonstrates significant advantages in both accuracy and speed despite a substantial reduction in the number of parameters, which verifies the effectiveness and practicality of the proposed pruning method.

       

    /

    返回文章
    返回