多尺度注意力网络提升红外小目标检测精度
MSCA-Net:Multi-Scale Context Aggregation Network for Infrared Small Target Detection
- 设计多尺度注意力模块,融合不同层级特征增强表达
- 在三个数据集上达到最高78.43%的mIoU,性能领先
- 适合复杂背景下红外小目标检测任务应用
在复杂环境下,红外图像中微小目标检测因对比度低、噪声高而困难,导致特征提取时关键细节丢失。现有方法难以有效融合全局与局部信息,制约了检测效率与准确率。为此,本文提出MSCA-Net,集成三种核心组件:多尺度增强空洞注意力机制(MSEDA)、位置卷积块注意力模块(PCBAM)和通道聚合特征融合块(CAB)。MSEDA通过多尺度特征融合注意力机制自适应聚合多尺度信息,丰富特征表示;PCBAM基于相关矩阵策略捕捉全局与局部特征间关联,实现深层特征交互;CAB通过加权关键特征,融合低层与高层信息,提升模型在复杂背景下的检测性能。实验表明,MSCA-Net在复杂背景下表现优异,在NUAA-SIRST、NUDT-SIRST和IRTS-1K数据集上分别取得78.43%、94.56%和67.08%的mIoU,验证其有效性与实际应用潜力。
原文摘要 · Abstract (English)
In complex environments, detecting tiny infrared targets has always been challenging because of the low contrast and high noise levels inherent in infrared images. These factors often lead to the loss of crucial details during feature extraction. Moreover, existing detection methods have limitations in adequately integrating global and local information, which constrains the efficiency and accuracy of infrared small target detection. To address these challenges, this paper proposes a network architecture named MSCA-Net, which integrates three key components: Multi-Scale Enhanced Dilated Attention mechanism (MSEDA), Positional Convolutional Block Attention Module (PCBAM), and Channel Aggregation Feature Fusion Block (CAB). Specifically, MSEDA employs a multi-scale feature fusion attention mechanism to adaptively aggregate information across different scales, enriching feature representation. PCBAM captures the correlation between global and local features through a correlation matrix-based strategy, enabling deep feature interaction. Moreover, CAB enhances the representation of critical features by assigning greater weights to them, integrating both low-level and high-level information, and thereby improving the models detection performance in complex backgrounds. The experimental results demonstrate that MSCA-Net achieves strong small target detection performance in complex backgrounds. Specifically, it attains mIoU scores of 78.43%, 94.56%, and 67.08% on the NUAA-SIRST, NUDT-SIRST, and IRTSD-1K datasets, respectively, underscoring its effectiveness and strong potential for real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。