用动态稀疏压缩感知网络,精准分离红外图像中密集排列的小目标。
DSCSNet: A Dynamic Sparse Compression Sensing Network for Closely-Spaced Infrared Small Target Unmixing
- 将ADMM算法与可学习参数结合,嵌入严格ℓ₁范数约束以保留目标能量峰值
- 自注意力动态阈值机制根据迭代信息自适应调节稀疏化强度,提升精度
- 在复杂红外场景下显著优于现有方法,适合小目标检测与跟踪任务
由于光学镜头焦距和探测器分辨率的限制,远距离密集分布的红外小目标常表现为混合光斑。紧密小目标解混(CSOU)任务旨在从这些混合光斑中恢复出单个目标的数量、亚像素位置及辐射强度,属于高度病态的逆问题。现有方法难以兼顾模型驱动方法的严格稀疏性保证与数据驱动方法对动态场景的适应能力。为此,本文提出动态稀疏压缩感知网络(DSCSNet),一种深度展开网络,将交替方向乘子法(ADMM)与可学习参数相结合。具体地,在ADMM的辅助变量更新步骤中嵌入严格的ℓ₁范数稀疏约束,替代传统ℓ₂范数平滑项,有效保留小目标的离散能量峰。同时,在重构阶段引入基于自注意力的动态阈值机制,利用迭代过程中的稀疏增强信息自适应调整稀疏化强度。上述模块在ADMM的三个迭代步骤中端到端联合优化。保留压缩感知的物理逻辑,DSCSNet实现鲁棒的稀疏诱导与场景适应性,从而提升复杂红外场景下的解混精度与泛化能力。在合成红外数据集CSIST-100K上的大量实验表明,DSCSNet在关键指标如CSO-mAP和亚像素定位误差上均优于现有最先进方法。
原文摘要 · Abstract (English)
Due to the limitations of optical lens focal length and detector resolution, distant clustered infrared small targets often appear as mixed spots. The Close Small Object Unmixing (CSOU) task aims to recover the number, sub-pixel positions, and radiant intensities of individual targets from these spots, which is a highly ill-posed inverse problem. Existing methods struggle to balance the rigorous sparsity guarantees of model-driven approaches and the dynamic scene adaptability of data-driven methods. To address this dilemma, this paper proposes a Dynamic Sparse Compressed Sensing Network (DSCSNet), a deep-unfolded network that couples the Alternating Direction Method of Multipliers (ADMM) with learnable parameters. Specifically, we embed a strict $\ell_1$-norm sparsity constraint into the auxiliary variable update step of ADMM to replace the traditional $\ell_2$-norm smoothness-promoting terms, which effectively preserves the discrete energy peaks of small targets. We also integrate a self-attention-based dynamic thresholding mechanism into the reconstruction stage, which adaptively adjusts the sparsification intensity using the sparsity-enhanced information from the iterative process. These modules are jointly optimized end-to-end across the three iterative steps of ADMM. Retaining the physical logic of compressed sensing, DSCSNet achieves robust sparsity induction and scene adaptability, thus enhancing the unmixing accuracy and generalization in complex infrared scenarios. Extensive experiments on the synthetic infrared dataset CSIST-100K demonstrate that DSCSNet outperforms state-of-the-art methods in key metrics such as CSO-mAP and sub-pixel localization error.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。