SpirDet: Toward Efficient, Accurate, and Lightweight Infrared Small-Target Detector

Qianchen Mao,Qiang Li,Bingshu Wang,Yongjun Zhang,Tao Dai,C. L. Philip Chen
DOI: https://doi.org/10.1109/tgrs.2024.3470514
IF: 8.2
2024-10-19
IEEE Transactions on Geoscience and Remote Sensing
Abstract:In recent years, the detection of infrared small targets using deep learning methods has garnered substantial attention due to notable advancements. To improve the detection capability of small targets, these methods commonly maintain a pathway that preserves high-resolution (HR) features of sparse and tiny targets. However, it can result in redundant and expensive computations. To tackle this challenge, we propose a sparse infrared-target detector (SpirDet) for the efficient detection of infrared small targets. Specifically, to cope with the computational redundancy issue, we employ a new dual-branch sparse decoder (DBSD) to restore the feature map. First, the fast branch directly predicts a sparse map indicating potential small-target locations. Second, the slow branch conducts fine-grained adjustments at the positions indicated by the sparse map. In addition, we design a lightweight DO-RepEncoder based on reparameterization with the downsampling orthogonality (DO), which can effectively reduce memory consumption and inference latency. Extensive experiments show that the proposed SpirDet significantly outperforms state-of-the-art (SOTA) models while achieving faster inference speed and fewer parameters. For example, on the NUDT-SIRST dataset, SpirDet improves mean intersection over union (MIoU) by 2.09 and has a frames/s acceleration compared to the previous SOTA model. The code is available at https://github.com/laCorse/SpirDet-Pytorch.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?