投审稿入口

基于多感受野特征提取与轻量交叉注意力的改进YOLOv11带钢表面缺陷检测算法

Improved YOLOv11based on multi-receptive field feature extraction and lightweight cross attention fusion for strip steel surface defect detection

  • 摘要: 针对带钢表面缺陷大小不一、种类多样,光照不足等复杂背景干扰导致目标缺陷难以检测的问题,本文提出了一种基于多感受野特征提取(MRFFE)与轻量交叉注意力模块(LCA)的改进YOLOv11(MRFFELCA-YOLO)算法。首先,针对YOLOv11特征提取阶段对多尺度缺陷适应性不足的问题,设计了MRFFE模块,通过多分支并行卷积结合残差流,增强算法对不同尺度缺陷的特征识别能力。其次,针对YOLOv11特征融合阶段简单拼接导致的信息稀释与噪声引入问题,引入LCA模块,通过构建交叉注意力与特征增强层,实现自适应特征交互并抑制背景干扰,以提升缺陷特征的识别精度。第三,针对基于交并比(IoU)的损失函数对小目标敏感、定位精度低的问题,采用具有尺度不变性与联合优化机制的高斯组合距离(GCD)作为损失函数,以提升算法对小目标缺陷的检测能力。最后,在东北大学带钢缺陷数据集(NEU-DET)上进行测试,结果表明所提算法均值平均精度(PmAP)达到78.61%,较基准YOLOv11提升3.71%,验证了所提算法的有效性。

     

    Abstract: An improved YOLOv11 (MRFFELCA-YOLO) algorithm based on Multi receptive Field Feature Extraction (MRFFE) and Lightweight Cross Attention Module (LCA) was proposed in order to address the problem of difficult detection of target defects caused by complex background interference such as varying sizes, diverse types, and insufficient lighting on the surface of strip steel. Firstly, in response to the insufficient adaptability of YOLOv11 feature extraction stage to multi-scale defects, an MRFFE module was designed to enhance the model's feature recognition ability for defects of different scales through multi branch parallel convolution combined with residual flow. Secondly, in order to address the issues of information dilution and noise introduction caused by simple concatenation during the YOLOv11 feature fusion stage, an LCA module was introduced to construct cross attention and feature enhancement layers to achieve adaptive feature interaction and suppress background interference, thereby improving the recognition accuracy of defect features. Thirdly, to address the issue of sensitivity to small targets and low localization accuracy in Intersection over Union (IoU) based loss functions, Gaussian Combination Distance (GCD) with scale invariance and joint optimization mechanism is adopted as the loss function to enhance the algorithm's ability to detect defects in small targets. Finally, testing was conducted on NEU-DET (Northeastern University Strip Defect Dataset), and the results showed that the mean Average Precision (PmAP) of the proposed algorithm achieves 78.61%, representing a 3.71% improvement over the baseline YOLOv11.

     

/

返回文章
返回