融合小波与线性注意力机制的红外目标检测算法

Infrared Target Detection Algorithm Combining Wavelets and Linear Attention

  • 摘要: 针对当前红外目标检测器骨干网络算法复杂度高以及有效感受野受限问题,本文提出了基于小波变换卷积与线性注意力的混合检测算法WTMLLA-YOLO。首先,构建多尺度小波特征提取模块,在不显著增加模型复杂度的条件下扩展感受野并强化低频结构信息响应,从而提升对轮廓与形状特征的表征能力。其次,设计无递归的双分支线性注意力机制,以可并行的线性复杂度建模全局上下文并促进局部——全局信息交互,降低遮挡与近距离目标的漏检风险,并提升检测精度。最后,通过解耦检测头对跨层融合特征进行重校准,消除多尺度特征的语义冲突。FLIR与MSRS数据集上的实验表明,算法在较基准模型mAP50值提升3.5%的同时计算量FLOPs降低0.3G,显著提升了检测精度和鲁棒性。

     

    Abstract: To address the issues of high complexity and limited effective receptive field in current infrared target detector backbone network algorithms, this paper proposes a hybrid detection algorithm, WTMLLAYOLO, based on wavelet transform convolution and linear attention. First, a multi-scale wavelet feature extraction module is constructed to expand the receptive field and enhance low-frequency structural information response without significantly increasing model complexity, thereby improving the representation ability of contour and shape features. Second, a non-recursive bi-branch linear attention mechanism is designed to model the global context with parallelizable linear complexity and promote local-global information interaction, reducing the risk of missed detections of occluded and near-range targets, and further improving detection accuracy. Finally, the cross-layer fused features are recalibrated by decoupling the detection head to eliminate semantic conflicts between multi-scale features. Experiments on the FLIR and MSRS datasets show that the algorithm achieves a 3.5% improvement in accuracy compared to the benchmark model while reducing computational cost by 0.3G (FLOPs), significantly improving detection accuracy and robustness.

     

/

返回文章
返回