提出SDS-Net模型,提升红外小目标检测精度与速度
SDS-Net: Shallow-Deep Synergism-detection Network for infrared small target detection
- 双分支架构分别处理浅层结构与深层语义特征
- 在三个数据集上优于现有方法,推理速度显著提升
- 适合需要高效高精度红外检测的应用场景
现有基于CNN的红外小目标检测方法普遍忽略浅层与深层特征间的异质性,导致细粒度结构信息与高层语义表征协作效率低下。不同层级特征间的依赖关系与融合机制缺乏系统建模,难以充分挖掘多层级特征的互补性,限制了检测性能并带来高昂计算成本。为此,本文提出浅-深协同检测网络(SDS-Net),通过双分支架构分别建模特征的结构特性与语义属性,有效保留浅层空间细节,同时捕获深层语义信息,实现高精度检测与显著提升的推理速度。此外,引入自适应特征融合模块,动态建模跨层特征相关性,增强整体特征协同与表达能力。在NUAA-SIRST、NUDT-SIRST和IRSTD-1K三个公开数据集上的实验表明,SDS-Net优于当前最优方法,兼具低计算复杂度与高推理效率,展现出优异的检测性能与广泛应用前景。
原文摘要 · Abstract (English)
Current CNN-based infrared small target detection(IRSTD) methods generally overlook the heterogeneity between shallow and deep features, leading to inefficient collaboration between shallow fine grained structural information and deep high-level semantic representations. Additionally, the dependency relationships and fusion mechanisms across different feature hierarchies lack systematic modeling, which fails to fully exploit the complementarity of multilevel features. These limitations hinder IRSTD performance while incurring substantial computational costs. To address these challenges, this paper proposes a shallow-deep synergistic detection network (SDS-Net) that efficiently models multilevel feature representations to increase both the detection accuracy and computational efficiency in IRSTD tasks. SDS-Net introduces a dual-branch architecture that separately models the structural characteristics and semantic properties of features, effectively preserving shallow spatial details while capturing deep semantic representations, thereby achieving high-precision detection with significantly improved inference speed. Furthermore, the network incorporates an adaptive feature fusion module to dynamically model cross-layer feature correlations, enhancing overall feature collaboration and representation capability. Comprehensive experiments on three public datasets (NUAA-SIRST, NUDT-SIRST, and IRSTD-1K) demonstrate that SDS-Net outperforms state-of-the-art IRSTD methods while maintaining low computational complexity and high inference efficiency, showing superior detection performance and broad application prospects. Our code will be made public at https://github.com/PhysiLearn/SDS-Net.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。