# Revisiting the Scale Loss Function and Gaussian-Shape Convolution for Infrared Small Target Detection

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

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 12 of its 31 references that are in the map.
- First posted on arXiv 11 Apr 2026.

## 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)
- [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)
- [Attentional Local Contrast Networks for Infrared Small Target Detection](https://mlsanity.com/paper/2012.08573/) (2020)
- [Infrared Small Target Detection with Scale and Location Sensitivity](https://mlsanity.com/paper/2403.19366/) (2024)
- [SCTransNet: Spatial-channel Cross Transformer Network for Infrared Small Target Detection](https://mlsanity.com/paper/2401.15583/) (2024)
- [MTU-Net: Multi-level TransUNet for Space-based Infrared Tiny Ship Detection](https://mlsanity.com/paper/2209.13756/) (2022)
- [Pinwheel-shaped Convolution and Scale-based Dynamic Loss for Infrared Small Target Detection](https://mlsanity.com/paper/2412.16986/) (2024)
- [Alpha-IoU: A Family of Power Intersection over Union Losses for Bounding Box Regression](https://mlsanity.com/paper/2110.13675/) (2021)
- [TBC-Net: A real-time detector for infrared small target detection using semantic constraint](https://mlsanity.com/paper/2001.05852/) (2019)

## Built on by (most cited first, 2 of 2)

- [Multi-Adapter PPO: A Cross-Attention Enhanced Wavelength Selection Framework for LIBS Quantitative Analysis](https://mlsanity.com/paper/2606.17476/) (2026)
- [A Standard Processing Pipeline for High-accuracy Measurement of Few-shot Regression on Laser Induced Breakdown Spectroscopy](https://mlsanity.com/paper/2606.21960/) (2026)

## Abstract (arXiv)

Infrared small target detection still faces two persistent challenges: training instability from non-monotonic scale loss functions, and inadequate spatial attention due to generic convolution kernels that ignore the physical imaging characteristics of small targets. In this paper, we revisit both aspects. For the loss side, we propose a diff-based scale loss that weights predictions according to the signed area difference between the predicted mask and the ground truth, yielding strictly monotonic gradients and stable convergence. We further analyze a family of four scale loss variants to understand how their geometric properties affect detection behavior. For the spatial side, we introduce Gaussian-shaped convolution with a learnable scale parameter to match the center-concentrated intensity profile of infrared small targets, and augment it with a rotated pinwheel mask that adaptively aligns the kernel with target orientation via a straight-through estimator. Extensive experiments on IRSTD-1k, NUDT-SIRST, and SIRST-UAVB demonstrate consistent improvements in mIoU, Pd, and Fa over state-of-the-art methods. We release our anonymous code and pretrained models.