# Infrared Small Target Detection with Scale and Location Sensitivity

arXiv: https://arxiv.org/abs/2403.19366 · HTML: https://mlsanity.com/paper/2403.19366/

## Key facts

- Cited by 73 papers in the map (data through 14 Oct 2026).
- Builds on 12 of its 32 references that are in the map.
- First posted on arXiv 28 Mar 2024.

## Builds on (most cited first, 12 of 12)

- [Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regression](https://mlsanity.com/paper/1902.09630/) (2019)
- [Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentations](https://mlsanity.com/paper/1707.03237/) (2017)
- [Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression](https://mlsanity.com/paper/1911.08287/) (2019)
- [Asymmetric Contextual Modulation for Infrared Small Target Detection](https://mlsanity.com/paper/2009.14530/) (2020)
- [Dense Nested Attention Network for Infrared Small Target Detection](https://mlsanity.com/paper/2106.00487/) (2021)
- [UIU-Net: U-Net in U-Net for Infrared Small Object Detection](https://mlsanity.com/paper/2212.00968/) (2022)
- [Reweighted Infrared Patch-Tensor Model With Both Non-Local and Local Priors for Single-Frame Small Target Detection](https://mlsanity.com/paper/1703.09157/) (2017)
- [Mapping Degeneration Meets Label Evolution: Learning Infrared Small Target Detection with Single Point Supervision](https://mlsanity.com/paper/2304.01484/) (2023)
- [Instance Segmentation in the Dark](https://mlsanity.com/paper/2304.14298/) (2023)
- [TFPnP: Tuning-free Plug-and-Play Proximal Algorithm with Applications to Inverse Imaging Problems](https://mlsanity.com/paper/2012.05703/) (2020)
- [RIFormer: Keep Your Vision Backbone Effective While Removing Token Mixer](https://mlsanity.com/paper/2304.05659/) (2023)
- [Image Restoration from Patch-based Compressed Sensing Measurement](https://mlsanity.com/paper/1706.00597/) (2017)

## Built on by (most cited first, 30 of 73)

- [Pinwheel-shaped Convolution and Scale-based Dynamic Loss for Infrared Small Target Detection](https://mlsanity.com/paper/2412.16986/) (2024)
- [Triple-domain Feature Learning with Frequency-aware Memory Enhancement for Moving Infrared Small Target Detection](https://mlsanity.com/paper/2406.06949/) (2024)
- [Multi-Scale Direction-Aware Network for Infrared Small Target Detection](https://mlsanity.com/paper/2406.02037/) (2024)
- [RPCANet++: Deep Interpretable Robust PCA for Sparse Object Segmentation](https://mlsanity.com/paper/2508.04190/) (2025)
- [Probing Deep into Temporal Profile Makes the Infrared Small Target Detector Much Better](https://mlsanity.com/paper/2506.12766/) (2025)
- [DRPCA-Net: Make Robust PCA Great Again for Infrared Small Target Detection](https://mlsanity.com/paper/2507.09541/) (2025)
- [Seeing Through the Noise: Improving Infrared Small Target Detection and Segmentation from Noise Suppression Perspective](https://mlsanity.com/paper/2508.06878/) (2025)
- [Text-IRSTD: Leveraging Semantic Text to Promote Infrared Small Target Detection in Complex Scenes](https://mlsanity.com/paper/2503.07249/) (2025)
- [Detection-Friendly Nonuniformity Correction: A Union Framework for Infrared UAVTarget Detection](https://mlsanity.com/paper/2504.04012/) (2025)
- [Rethinking Generalizable Infrared Small Target Detection: A Real-scene Benchmark and Cross-view Representation Learning](https://mlsanity.com/paper/2504.16487/) (2025)
- [SDS-Net: Shallow-Deep Synergism-detection Network for infrared small target detection](https://mlsanity.com/paper/2506.06042/) (2025)
- [Lightweight Deep Unfolding Networks with Enhanced Robustness for Infrared Small Target Detection](https://mlsanity.com/paper/2509.08205/) (2025)
- [Rethinking Evaluation of Infrared Small Target Detection](https://mlsanity.com/paper/2509.16888/) (2025)
- [Dynamic High-frequency Convolution for Infrared Small Target Detection](https://mlsanity.com/paper/2602.02969/) (2026)
- [Lost in UNet: Improving Infrared Small Target Detection by Underappreciated Local Features](https://mlsanity.com/paper/2406.13445/) (2024)
- [Hybrid Mask Generation for Infrared Small Target Detection with Single-Point Supervision](https://mlsanity.com/paper/2409.04011/) (2024)
- [SeqCSIST: Sequential Closely-Spaced Infrared Small Target Unmixing](https://mlsanity.com/paper/2507.09556/) (2025)
- [SSTD: Stripe-Like Space Target Detection Using Single-Point Weak Supervision](https://mlsanity.com/paper/2407.18097/) (2024)
- [Paying more attention to local contrast: improving infrared small target detection performance via prior knowledge](https://mlsanity.com/paper/2411.13260/) (2024)
- [RRCANet: Recurrent Reusable-Convolution Attention Network for Infrared Small Target Detection](https://mlsanity.com/paper/2506.02393/) (2025)
- [It's Not the Target, It's the Background: Rethinking Infrared Small Target Detection via Deep Patch-Free Low-Rank Representations](https://mlsanity.com/paper/2506.10425/) (2025)
- [TY-RIST: Tactical YOLO Tricks for Real-time Infrared Small Target Detection](https://mlsanity.com/paper/2509.22909/) (2025)
- [Revisiting the Scale Loss Function and Gaussian-Shape Convolution for Infrared Small Target Detection](https://mlsanity.com/paper/2604.09991/) (2026)
- [10K is Enough: An Ultra-Lightweight Binarized Network for Infrared Small-Target Detection](https://mlsanity.com/paper/2503.02662/) (2025)
- [Target Driven Adaptive Loss For Infrared Small Target Detection](https://mlsanity.com/paper/2506.01349/) (2025)
- [Weakly-supervised Contrastive Learning with Quantity Prompts for Moving Infrared Small Target Detection](https://mlsanity.com/paper/2507.02454/) (2025)
- [Exploiting Gaussian Agnostic Representation Learning with Diffusion Priors for Enhanced Infrared Small Target Detection](https://mlsanity.com/paper/2507.18260/) (2025)
- [Beyond Motion Cues and Structural Sparsity: Revisiting Small Moving Target Detection](https://mlsanity.com/paper/2509.07654/) (2025)
- [Ivan-ISTD: Rethinking Cross-domain Heteroscedastic Noise Perturbations in Infrared Small Target Detection](https://mlsanity.com/paper/2510.12241/) (2025)
- [Point-to-Mask: From Arbitrary Point Annotations to Mask-Level Infrared Small Target Detection](https://mlsanity.com/paper/2603.16257/) (2026)

## Abstract (arXiv)

Recently, infrared small target detection (IRSTD) has been dominated by deep-learning-based methods. However, these methods mainly focus on the design of complex model structures to extract discriminative features, leaving the loss functions for IRSTD under-explored. For example, the widely used Intersection over Union (IoU) and Dice losses lack sensitivity to the scales and locations of targets, limiting the detection performance of detectors. In this paper, we focus on boosting detection performance with a more effective loss but a simpler model structure. Specifically, we first propose a novel Scale and Location Sensitive (SLS) loss to handle the limitations of existing losses: 1) for scale sensitivity, we compute a weight for the IoU loss based on target scales to help the detector distinguish targets with different scales: 2) for location sensitivity, we introduce a penalty term based on the center points of targets to help the detector localize targets more precisely. Then, we design a simple Multi-Scale Head to the plain U-Net (MSHNet). By applying SLS loss to each scale of the predictions, our MSHNet outperforms existing state-of-the-art methods by a large margin. In addition, the detection performance of existing detectors can be further improved when trained with our SLS loss, demonstrating the effectiveness and generalization of our SLS loss. The code is available at https://github.com/ying-fu/MSHNet.
