基于感受野增强与多模块协同优化的目标检测算法OA
Object Detection Algorithm Based on Receptive Field Enhancement and Multi-module Collaborative Optimization
在目标检测任务中,YOLO系列算法在速度与精度之间取得了卓越平衡.然而,其在感受野的设计方面仍存在进一步优化的空间.为了扩大感受野,传统方法通常通过增大卷积核尺寸来实现,但这会显著增加模型的计算复杂度;同时,传统的自注意力机制多依赖于局部或全局窗口来计算注意力权重,容易受到无关区域的干扰,进而影响检测精度.为解决上述问题,提出一种基于感受野增强与多模块协同优化的目标检测算法(RFE-YOLO).RFE-YOLO在主干网设计感受野增强模块(C3K2_WT),利用小波变换可以在不受过度参数化影响的情况下获得更大的感受野;设计的跨阶段部分与可变形自注意力机制模块(C2DAT)采用了可变形自注意力机制聚焦于相关区域并捕获更重要的信息特征;在颈部引入特征融合模块(SBA)通过融合底层特征的边界细节与高层特征的语义信息,实现对目标边界的精准保留和定位的精确重标定;引入Down模块执行下采样操作以降低模型复杂度.在基准数据集MS COCO2017上的实验结果表明,RFE-YOLO相较于YOLO系列其他模型,mAP指标均有提升(最高提升11.6%,最低提升1.1%),且与SOTA算法RT-DETR系列相比,其mAP值也具有可比性,而且RFE-YOLO的模型参数量仅为RT-DETR-L的7.1%.消融实验的结果也表明了各个模块的有效性.
In object detection tasks,the YOLO series algorithms have achieved an outstanding balance between speed and accuracy.However,there remains room for improvement in their receptive field design.Conventional approaches typically enlarge the receptive field by increasing convolutional kernel sizes,which substantially raises computational complexity.Meanwhile,traditional self-attention mechanisms often rely on local or global windows to compute attention weights,making them susceptible to interference from irrelevant regions and consequently degrading detection accuracy.To address these limitations,this paper proposes YOLO with receptive field enhancement(RFE-YOLO),a novel object detec-tion algorithm based on receptive field enhancement and multi-module collaborative optimization.RFE-YOLO introduces a C3K2_WT(cross stage partial with kernel size 2 and wavelet transform convolution)module in the backbone network,which employs wavelet transformation to achieve an expanded receptive field without excessive parameterization.C2DAT(cross stage partial with deformable self-attention transformer)module that utilizes deformable self-attention to focus on relevant regions and capture more critical feature information is designed.The neck network incorporates a selective boundary aggregation(SBA)module to fuse low-level boundary details with high-level semantic features,enabling pre-cise boundary preservation and accurate localization recalibration.A Down module is introduced to perform downsam-pling operations,effectively reducing model complexity.Experimental results on the benchmark MSCOCO 2017 dataset demonstrate that RFE-YOLO consistently outperforms other YOLO variants,with mAP improvements ranging from 1.1%to 11.6%.Moreover,its mAP is on par with that of the state-of-the-art RT-DETR(real-time detection transformer)family,while the parameter count of RFE-YOLO is only 7.1%of RT-DETR-L.Ablation studies further validate the effectiveness of each proposed module.
朱明;胡树煜;冷强奎;马鹏飞
锦州医科大学 智能医学学院,辽宁 锦州 121001锦州医科大学 智能医学学院,辽宁 锦州 121001辽宁工程技术大学 电子与信息工程学院,辽宁 葫芦岛 125105锦州医科大学 智能医学学院,辽宁 锦州 121001
信息技术与安全科学
目标检测感受野多尺度特征融合卷积神经网络可变形自注意力机制
object detectionreceptive fieldmulti-scale feature fusionconvolutional neural networkdeformable self-attention mechanism
《计算机科学与探索》 2026 (7)
2130-2144,15
辽宁省教育厅服务地方项目(JYTFW2020009)辽宁省高校基本科研业务费项目(LJ212410160087). This work was supported by the Local Service Project of Liaoning Provincial Department of Education(JYTFW2020009),and the Fun-damental Research Funds Project of Higher Education Institutions in Liaoning Province(LJ212410160087).
评论