038、YOLOv11检测头轻量化——通道剪枝设计在解耦头中的即插即用实现与推理加速

038、YOLOv11检测头轻量化——通道剪枝设计在解耦头中的即插即用实现与推理加速

一、从一次线上事故说起

上个月部署YOLOv11到边缘设备，模型在GPU上跑得挺欢，一上RK3588直接卡成PPT。排查发现检测头占了整个模型推理时间的37%，比backbone还离谱。解耦头里那两层3x3卷积，每个通道数都是256，对边缘设备来说简直是灾难。当时就想，能不能在保持精度的前提下，把检测头里的冗余通道砍掉？于是就有了今天要聊的这个通道剪枝方案。

二、解耦头为什么需要瘦身

YOLOv11的解耦头结构大家应该不陌生——分类分支和回归分支各走各的路，每个分支先过两层3x3卷积，再接1x1卷积输出。问题就出在这两层3x3卷积上。原始设计里通道数统一设为256，但实际训练中发现，很多通道的激活值接近零，这些通道对最终检测结果几乎没有贡献。

我做过一个实验：把检测头里所有通道的激活值统计出来，画了个分布图。结果发现大约40%的通道平均激活值不到最大值的5%。这些通道就是典型的“冗余通道”，留着它们只会增加计算量，不会提升精度。

三、通道剪枝的核心思路

剪枝的思路其实很朴素——给每个通道学一个重要性权重，训练过程中让不重要的通道权重趋近于零，然后直接砍掉。具体到YOLOv11的解耦头，我们只对那两层3x3卷积做剪枝，1x1输出层不动，因为输出层的通道数由类别数和anchor数决定，没法剪。

这里有个关键点：剪枝要放在BN层前面。YOLOv11的解耦头里每个3x3卷积后面都跟着BN和SiLU激活，我们利用