Difference Decomposition Networks for Infrared Small Target Detection

Code

• github.com/greekinroma/irstd_hc_platform
  OfficialIn paper★ 0

Abstract

Infrared small target detection (ISTD) faces two major challenges: a lack of discernible target texture and severe background clutter, which results in the background obscuring the target. To enhance targets and suppress backgrounds, we propose the Basis Decomposition Module (BDM) as an extensible and lightweight module based on basis decomposition, which decomposes a complex feature into several basis features and enhances certain information while eliminating redundancy. Extending BDM leads to a series of modules, including the Spatial Difference Decomposition Module (SD^2M), Spatial Difference Decomposition Downsampling Module (SD^3M), and Temporal Difference Decomposition Module (TD^2M). Based on these modules, we develop the Spatial Difference Decomposition Network (SD^2Net) for single-frame ISTD (SISTD) and the Spatiotemporal Difference Decomposition Network (STD^2Net) for multi-frame ISTD (MISTD). SD^2Net integrates SD^2M and SD^3M within an adapted U-shaped architecture. We employ TD^2M to introduce motion information, which transforms SD^2Net into STD^2Net. Extensive experiments on SISTD and MISTD datasets demonstrate state-of-the-art (SOTA) performance. On the SISTD task, SD^2Net performs well compared to most established networks. On the MISTD datasets, STD^2Net achieves a mIoU of 87.68\%, outperforming SD^2Net, which achieves a mIoU of 64.97\%. Our codes are available: https://github.com/greekinRoma/IRSTD_HC_Platform.

Reproductions