Point-to-Mask: From Arbitrary Point Annotations to Mask-Level Infrared Small Target Detection
Organizations: Key Laboratory of Electronics and Information Technology for Space Systems, National Space Science Center, Chinese Academy of Sciences, Beijing 100190, China. · School of Computer Science and Technology, University of Chinese Academy of Sciences, Beijing 100049, China.
Abstract
Infrared small target detection (IRSTD) methods predominantly formulate the task as pixel-level segmentation, which requires costly dense annotations and is not well suited to tiny targets with weak texture and ambiguous boundaries. To address this issue, we propose Point-to-Mask, a framework that bridges low-cost point supervision and mask-level detection through two components: a Physics-driven Adaptive Mask Generation (PAMG) module that converts point annotations into compact target masks and geometric cues, and a lightweight Radius-aware Point Regression Network (RPR-Net) that reformulates IRSTD as target center localization and effective radius regression using spatiotemporal motion cues. The two modules form a closed loop: PAMG generates pseudo masks and geometric supervision during training, while the geometric predictions of RPR-Net are fed back to PAMG for pixel-level mask recovery during inference. To facilitate systematic evaluation, we further construct SIRSTD-Pixel, a sequential dataset with refined pixel-level annotations. Experiments show that the proposed framework achieves strong pseudo-label quality, high detection accuracy, and efficient inference, approaching full-supervision performance under point-supervised settings with substantially lower annotation cost. Code and datasets will be available at: https://github.com/GaoScience/point-to-mask.
Figures & tables
| Method | Setting | SIRSTD | NUDT-SIRST | IRSTD-1K | Mean |
|---|---|---|---|---|---|
| MCLC [ 16 ] | Prior | 63.08 | 69.44 | 71.33 | 67.95 |
| Blind | 54.42 | 62.69 | 66.97 | 61.36 | |
| -8.66 | -6.75 | -4.36 | -6.59 | ||
| MCGC [ 15 ] | Prior | 63.58 | 73.04 | 63.62 | 66.75 |
| Blind | 61.01 | 70.93 | 63.26 | 65.07 | |
| -2.57 | -2.11 | -0.36 | -1.68 |
| Type | Method | Sup. | Out | SIRSTD-Pixel | IRSTD-1K / NUDT-SIRST | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| mIoU | AUC | mIoU | AUC | ||||||||
| Single-frame | ACM [ 3 ] | Pixel | Mask | 43.13 | 69.04 | 1.80 | 0.8452 | 58.34 / 65.65 | 94.59 / 97.56 | 5.23 / 3.05 | 0.9730 / 0.9878 |
| AGPCNet [ 40 ] | Pixel | Mask | 54.55 | 71.42 | 0.33 | 0.8571 | 64.66 / 79.78 | 96.28 / 99.15 | 3.22 / 2.01 | 0.9814 / 0.9958 | |
| ALCNet [ 4 ] | Pixel | Mask | 45.16 | 75.50 | 3.11 | 0.8775 | 59.61 / 52.85 | 96.62 / 98.20 | 5.12 / 8.57 | 0.9831 / 0.9910 | |
| DNANet [ 17 ] | Pixel | Mask | 55.06 | 67.81 | 0.26 | 0.8390 | 68.92 / 84.50 | 94.26 / 98.73 | 2.58 / 0.82 | 0.9713 / 0.9936 | |
| ISNet [ 39 ] | Pixel | Mask | 47.14 | 74.19 | 3.89 | 0.8710 | 64.01 / 61.57 | 95.27 / 98.52 | 3.74 / 20.5 | 0.9763 / 0.9926 | |
| Seed Location | mIoU(%) | Area Ratio | Centroid Error | Radius Error |
|---|---|---|---|---|
| Center | 87.19 | 1.1793 | 12.1526 | 0.2635 |
| Random Interior | 87.62 | 1.2432 | 14.6276 | 0.2673 |
| Boundary | 87.81 | 1.2530 | 15.0028 | 0.2654 |
| (a) Impact of Fixed Support Radius (Blind Mode) | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Radius | 2 | 4 | 8 | 12 | 16 | 20 | 24 | 28 | 32 | 40 |
| mIoU (%) | 8.75 | 34.43 | 72.69 | 85.09 | 88.37 | 88.34 | 87.73 | 87.03 | 86.25 | 84.99 |
| (b) Impact of Adaptive Scale Factor (Guided Mode, ) | ||||||||||
| Scale Factor | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
| mIoU (%) | 8.63 | 52.53 | 82.98 | 88.21 | 89.12 | 89.25 | 88.97 | 88.74 | 88.34 | 87.96 |
| Variant | mIoU (%) | Area Ratio |
|---|---|---|
| Full PAMG | 88.34 | 1.22 |
| w/o Size Prior | 75.68 | 0.86 |
| w/o Saliency | 39.63 | 3.29 |
| w/o Homogeneity | 49.26 | 0.88 |
| w/o Geometric Prior | 75.02 | 13.40 |
| Model Variant | Key Change | Backbone Config | Detection Metrics | Shape | |||
|---|---|---|---|---|---|---|---|
| Width | Max Stride | (%) | ( ) | AUC | mIoU (%) | ||
| Ablation on Backbone Capacity & Resolution | |||||||
| Variant-Width-S | Small Width | 0.375 | 16 | 89.69 | 6.44 | 0.9485 | 34.63 |
| Variant-Width-L | Large Width | 0.75 | 16 | 90.49 | 12.00 | 0.9524 | 29.90 |
| Variant-Stride | Standard Downsample | 0.5 | 32 | 88.69 | 2.60 | 0.9435 | 39.27 |
| Ablation on Temporal Modeling | |||||||
| Type | Method | Params | FLOPs | Latency | AUC |
| Single-frame | ACM [ 3 ] | 0.40 | 2.01 | 2.95 | 0.8452 |
| AGPCNet [ 40 ] | 12.36 | 215.91 | 37.75 | 0.8571 | |
| ALCNet [ 4 ] | 0.52 | 1.89 | 2.81 | 0.8775 | |
| DNANet [ 17 ] | 4.70 | 71.31 | 79.49 | 0.8390 | |
| ISNet [ 39 ] | 1.09 | 153.09 | 25.60 | 0.8710 | |
| ISTDU-Net [ 10 ] | 2.76 | 39.72 | 13.70 | 0.8577 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Abbreviation | Full Expression |
|---|---|
| IRSTD | Infrared Small Target Detection |
| PAMG | Physics-driven Adaptive Mask Generation |
| RPR-Net | Radius-aware Point Regression Network |
| PSF | Point Spread Function |
| MAP | Maximum A Posteriori |
| Hw-SNR | Homogeneity-Weighted Signal-to-Noise Ratio |
Infrared Small Target Detection
Infrared small target detection finds faint, often sub-pixel objects against cluttered thermal backgrounds while limiting false alarms.
See all 43 papers