用梯度基分解提升红外小目标检测精度,兼顾效果与速度
STGBD-Net: Spatio-temporal Gradient Basis Decomposition Network for Infrared Small Target Detection
- 将多方向梯度特征视为基向量,动态重构新特征以增强细节
- 在多个数据集上达到顶尖检测性能,参数量少、推理快
- 适合需要高精度实时检测的红外成像场景,如军事侦察
红外小目标检测(IRSTD)的核心挑战在于微弱目标信号易被强背景干扰掩盖,导致漏检。传统基于梯度的方法虽能捕捉细节,但因多方向梯度特征静态融合,鲁棒性不足。本文从基分解理论出发,提出新型显式自适应分解-重构范式。设计基分解模块(BDM)及其专用变体梯度分解模块(GDM),将归一化梯度特征作为基向量,重建新特征以保持结构细节并突出红外小目标。将GDM嵌入轻量级三阶段U-Net,构建两种统一架构:单帧检测的时空梯度基分解网络(Spatial GDM Net)和多帧场景的时序梯度基分解网络(STGBD-Net)。大量实验表明,该方法在多个基准测试中达到当前最优(SOTA)性能,同时在检测精度与计算效率间实现更优平衡。代码将公开于:https://github.com/greekinRoma/IRSTD_HC_Platform。
原文摘要 · Abstract (English)
A key challenge in infrared small target detection (IRSTD) is that weak target signal responses are easily obscured by strong background clutter, frequently resulting in missed detections. While traditional gradient-based methods attempt to capture fine details, their robustness is limited by the static fusion of multi-directional gradient features. In this paper, we rethink feature fusion from the perspective of Basis Decomposition Theory and propose a novel framework that reformulates the process into an explicit and adaptive decomposition-and-reconstruction paradigm. Specifically, we introduce the Basis Decomposition Module (BDM) and its specialized variant, the Gradient Decomposition Module (GDM) for IRSTD. GDMs treat the normalized gradient features as basis vectors to reconstruct a new feature, thereby maintaining detailed structures and highlighting infrared small targets. By integrating GDMs into a lightweight three-stage U-Net, we develop two unified architectures: the Spatial Gradient Basis Decomposition Network for single-frame detection and the Spatio-temporal Gradient Basis Decomposition Network for multi-frame scenarios. Extensive experiments demonstrate that our networks achieve state-of-the-art (SOTA) performance across multiple benchmarks, offering a superior balance between detection accuracy and computational efficiency. Our codes will be made public at: https://github.com/greekinRoma/IRSTD_HC_Platform.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。