1. 大核瓶颈架构的设计动机与核心思想
在目标检测领域,感受野的大小直接决定了模型对目标上下文信息的捕获能力。传统YOLO系列模型主要依赖3×3卷积核进行特征提取,这种设计虽然计算高效,但在处理大尺度目标或需要长距离依赖建模的场景时存在明显局限。我们团队在开发YOLOv26过程中发现,当目标尺寸超过感受野范围时,检测性能会出现显著下降。
大核瓶颈架构的提出源于三个关键观察:
- 现代高分辨率图像中,大尺寸目标占比越来越高(如自动驾驶场景中的车辆、智慧城市中的建筑物)
- 现有模型的感受野增长主要依赖网络深度堆叠,导致计算量呈指数级增长
- 5×5卷积在保持合理计算量的前提下,单层即可提供3×3卷积2.78倍的感受野面积
1.1 感受野的数学本质
感受野(RF)的计算遵循递推公式:
code复制RF_l = RF_{l-1} + (k_l - 1) × ∏_{i=1}^{l-1} s_i
其中k_l是第l层的卷积核大小,s_i是第i层的步长。对于步长为1的卷积:
- 3×3卷积堆叠3层时:RF=7×7
- 5×5卷积堆叠3层时:RF=13×13
我们在ImageNet预训练实验中发现,使用5×5卷积的模型在深层特征图上能保留更多大目标的结构信息。如图1所示,传统3×3卷积在特征图缩小到原图1/16时,有效感受野仅能覆盖目标局部,而5×5卷积仍能保持对整体形状的感知。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 大核瓶颈模块的详细实现
2.1 基础模块设计
大核瓶颈采用"压缩-处理-扩展"的三段式结构,核心代码实现如下:
python复制class LargeKernelBottleneck(nn.Module):
def __init__(self, c1, c2, shortcut=True, g=1, e=0.5):
super().__init__()
c_ = int(c2 * e) # 中间压缩通道数
self.cv1 = Conv(c1, c_, 1) # 1×1压缩
self.cv2 = Conv(c_, c_, 5, g=g) # 5×5大核卷积
self.cv3 = Conv(c_, c2, 1) # 1×1扩展
se
