Boosting Infrared Small Target Detection via Logit-Domain Contrast and Adaptive Shape Refinement

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

A new method, Adaptive-Contrastive SLSIoU (AC-SLSIoU), is proposed to enhance Infrared Small Target Detection (IRSTD), addressing challenges like tiny target size, low signal-to-noise ratio, and blurred boundaries. Existing IRSTD approaches often struggle with weak-target discrimination due to saturated probabilities in post-activation supervision and produce halo-like predictions from thermal diffusion. AC-SLSIoU integrates three key components: a Logit-Domain Margin Constraint (LDMC) to increase the response gap between targets and hard negatives in the logit space, Adaptive Boundary Suppression (ABS) for scale-aware annular penalties to refine target contours, and False-Alarm Focal Loss to penalize high-confidence false alarms. This plug-and-play loss function requires no extra inference overhead, seamlessly integrates into current detectors, and consistently improves both detection accuracy and shape quality, as demonstrated by extensive experiments and cross-backbone evaluations.

Key takeaway

For Computer Vision Engineers developing Infrared Small Target Detection (IRSTD) systems, consider integrating Adaptive-Contrastive SLSIoU (AC-SLSIoU) into your existing detectors. This plug-and-play loss function, which includes logit-domain contrast and adaptive boundary suppression, directly addresses weak-target discrimination and blurred boundary issues. Implementing AC-SLSIoU can significantly improve both detection accuracy and the quality of target shapes without adding inference overhead, streamlining your model's performance in complex scenes.

Key insights

Enhancing IRSTD requires logit-domain contrast, adaptive boundary refinement, and false-alarm focused loss to overcome weak-target and blurred boundary issues.

Principles

Method

AC-SLSIoU combines Logit-Domain Margin Constraint, Adaptive Boundary Suppression, and False-Alarm Focal Loss to create a plug-and-play, discriminative, and shape-aware loss function for IRSTD.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.