导读机车螺旋弹簧是铁路车辆悬挂系统的核心部件其缺陷可能导致脱轨等严重安全事故。传统的人工目视检查面临主观性强、疲劳易漏检、效率低等问题。弹簧缺陷检测的技术难点在于缺陷形态多样裂纹、腐蚀、脱碳、表面划伤、点蚀共五类空间尺度跨度大从微小表面裂纹到大面积腐蚀且弹簧的螺旋几何形状引入了镜面反射和遮挡干扰。MSD-DETRMulti-Scale Deformable Detection Transformer with Structural Re-parameterization针对这些挑战在RT-DETR基础上引入结构重参数化、可变形注意力和跨尺度特征融合三项改进在包含5000张图像、12847个缺陷实例的真实机车弹簧数据集上达到92.4% mAP0.5和98 FPS超过YOLOv8-L4.2% mAP和基准RT-DETR-L3.3% mAP。论文信息标题A Deformable Attention-Based Detection Transformer with Cross-Scale Feature Fusion for Industrial Coil Spring Inspection作者Matteo Rossi, Pony Matt机构Maharaja Agrasen University一、弹簧缺陷检测的三重挑战机车螺旋弹簧在运行中承受复杂的循环载荷长期使用后可能出现五类缺陷表面裂纹SC、腐蚀损伤CD、脱碳DC、表面划伤SS和点蚀PT。自动化视觉检测面临的主要挑战包括尺度变异性弹簧缺陷的空间尺度跨度很大。表面微裂纹可能仅占图像中几个像素而大面积腐蚀区域可能覆盖显著的图像范围。这要求检测模型同时具备细粒度的局部特征提取能力和大范围的上下文理解能力。形态多样性不同类型的缺陷具有显著不同的视觉特征。裂纹表现为细长的线性不连续腐蚀呈现不规则的空间分布脱碳区域则是边界模糊的低对比度区域表面划伤和点蚀则各有其独特的纹理模式。复杂表面几何弹簧的螺旋结构导致拍摄时出现镜面反射、相邻圈遮挡等干扰增加了缺陷和背景区分的难度。二、MSD-DETR的三项关键设计MSD-DETR在RT-DETR框架基础上进行了三项改进。图片来源于原论文结构重参数化模块借鉴RepVGG的设计思想MSD-DETR在骨干网络的卷积层中引入可重参数化的多分支结构训练时使用三个并行分支3×3卷积分支捕捉局部空间模式1×1卷积分支学习通道间关系恒等映射分支保持梯度流通畅推理时三个分支的参数通过数学变换合并为单个3×3卷积这一设计的好处是训练时享受多分支结构的表征多样性和更好的梯度流推理时几乎不增加额外计算开销。合并过程是精确的数学等价无近似误差。可变形注意力机制标准自注意力的计算复杂度为 为空间位置数在处理高分辨率特征图时计算开销过大。可变形注意力通过学习稀疏采样位置将复杂度降至 为每个查询的采样点数采样偏移 和注意力权重 均由查询特征预测使注意力能够自适应地聚焦于缺陷相关区域。多尺度扩展将可变形注意力扩展到同时处理多个分辨率级别的特征图注意力权重在所有尺度和采样点之间归一化实现跨尺度的信息整合。双向跨尺度特征融合在特征金字塔网络的基础上MSD-DETR设计了双向融合路径自顶向下语义增强路径从最高层最低分辨率开始逐层向下传播语义信息自底向上细节增强路径从最低层最高分辨率开始逐层向上传播细节信息融合模块中使用了两个轻量化组件GSConvGhost Shuffle Convolution通过1×1卷积通道投影深度可分离卷积通道混洗降低参数量VoVGSCSPVariational one-shot aggregation GSConv Cross Stage Partial通过密集连接增强多尺度特征的聚合效果三、在真实弹簧数据集上的实验结果整体性能对比论文在真实机车螺旋弹簧数据集上进行实验。与多个检测器对比方法mAP0.5FPSYOLOv8-L88.2%78RT-DETR-L89.1%114MSD-DETR92.4%98MSD-DETR相比YOLOv8-L提升4.2% mAP相比RT-DETR-L基线提升3.3% mAP。在推理速度上MSD-DETR98 FPS快于YOLOv8-L78 FPS略低于RT-DETR-L114 FPS整体在精度和速度之间取得了有利的平衡。图片来源于原论文各项改进的贡献三项技术改进各自的贡献结构重参数化增强了骨干网络的特征提取能力尤其对不同形态的缺陷特征的判别能力可变形注意力使编码器能够自适应聚焦于缺陷区域减少背景干扰对具有不规则空间分布的缺陷尤为有效跨尺度特征融合同时利用浅层的空间细节和深层的语义上下文提升了对不同尺度缺陷的检测一致性四、消融实验各模块的边际贡献论文通过逐步叠加各模块的方式验证了每项设计的有效性。从基准RT-DETR出发加入结构重参数化 → mAP提升多分支训练增强了特征多样性加入可变形注意力 → mAP进一步提升自适应采样提升了对稀疏缺陷的关注加入双向跨尺度融合 → 达到最终92.4%细粒度和语义信息的互补提升了跨尺度检测从消融结果来看可变形注意力DA贡献最大1.7% mAP跨尺度特征融合CSFF次之1.4%结构重参数化Rep贡献1.2%。这表明在弹簧检测场景中缺陷的稀疏空间分布和多尺度特性是两个关键瓶颈。五、总结与思考MSD-DETR通过结构重参数化、可变形注意力和跨尺度特征融合三项改进在机车弹簧缺陷检测任务上达到了92.4% mAP0.5和98 FPS的性能。方法建立在RT-DETR这一成熟框架之上工程实现相对直接。在此基础上有几点值得进一步思考跨域泛化验证论文在弹簧数据集之外还在NEU-DET钢材表面缺陷数据集上进行了零样本迁移实验初步验证了MSD-DETR的跨域泛化能力。论文指出未来将通过知识蒸馏和模型剪枝进一步优化推理效率。数据集未公开论文使用的弹簧数据集未公开其他研究者无法复现结果或进行公平对比。不过论文对数据集有较为详细的描述5000张图像、12847个缺陷实例、5个类别、70%/15%/15%的训练/验证/测试划分、图像分辨率2448×2048。