arXiv:2506.02393cs.CV2025-06被引 21

提出轻量级网络,提升红外小目标检测精度与效率。

RRCANet: Recurrent Reusable-Convolution Attention Network for Infrared Small Target Detection

  • 采用循环可复用卷积块,无额外参数实现深层特征增强。
  • 在三个数据集上达到顶尖性能,参数量显著低于主流方法。
  • 适合需要高效部署的红外目标检测场景,可无缝接入现有模型。

红外小目标检测因目标尺寸小、亮度低、无固定形状且动态变化而极具挑战。近期基于CNN的方法虽取得良好效果,但依赖复杂的特征提取与融合模块。为此,本文提出一种循环可复用卷积注意力网络(RRCA-Net),通过循环嵌套的可复用卷积块(RuCB)在不增加参数的前提下,有效保持并细化深层特征中的小目标信息。同时引入双交互注意力聚合模块(DIAAM),促进相邻层间上下文信息的相互增强与融合。为保障训练稳定收敛,设计了融合物理与数学约束的目标特性损失函数(DpT-k loss)。在NUAA-SIRST、IRSTD-1k、DenseSIRST三个基准数据集上的实验表明,该方法性能媲美当前最优模型,同时参数量更少,且可作为即插即用模块,为多个主流红外小目标检测方法带来一致性能提升。

原文摘要 · Abstract (English)

Infrared small target detection is a challenging task due to its unique characteristics (e.g., small, dim, shapeless and changeable). Recently published CNN-based methods have achieved promising performance with heavy feature extraction and fusion modules. To achieve efficient and effective detection, we propose a recurrent reusable-convolution attention network (RRCA-Net) for infrared small target detection. Specifically, RRCA-Net incorporates reusable-convolution block (RuCB) in a recurrent manner without introducing extra parameters. With the help of the repetitive iteration in RuCB, the high-level information of small targets in the deep layers can be well maintained and further refined. Then, a dual interactive attention aggregation module (DIAAM) is proposed to promote the mutual enhancement and fusion of refined information. In this way, RRCA-Net can both achieve high-level feature refinement and enhance the correlation of contextual information between adjacent layers. Moreover, to achieve steady convergence, we design a target characteristic inspired loss function (DpT-k loss) by integrating physical and mathematical constraints. Experimental results on three benchmark datasets (e.g. NUAA-SIRST, IRSTD-1k, DenseSIRST) demonstrate that our RRCA-Net can achieve comparable performance to the state-of-the-art methods while maintaining a small number of parameters, and act as a plug and play module to introduce consistent performance improvement for several popular IRSTD methods.

红外检测小目标轻量化注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。