Object detection in infrared images poses a considerable challenge due to its small-scale targets, low contrast and poor signal-to-clutter ratio, often resulting in a high false alarm rate. To improve the detection accuracy on infrared small targets, we introduce Light-SGMTLM, a lightweight and saliency-guided multi-task learning model. This model integrates saliency detection into the YOLOv5x framework through a parallel multi-task learning structure and employs a joint loss function during training. Such integration significantly alleviates the impact of complex backgrounds and improves the precision of small target localization. Moreover, we have developed a streamlined module, termed SIWD, to create a more agile backbone, which establishes an optimal balance between precision and efficiency, making the model more suitable for situations with limited computational resources. Comprehensive comparative experiments were conducted on six infrared small target datasets, namely, Small-ExtIRShip, Small-SSDD, IHAST, NUAA-SIRST, IRSTD-1k, and IRDST, and we assessed the model’s performance against ten leading target detection models, such as YOLOv7, YOLOv8, DINO, and Relation-DETR. The findings reveal that our method’s unique joint learning architecture, combining saliency and object detection tasks, significantly improves accuracy for infrared small target detection. Notably, it achieved impressive mean average precision (mAP) values of 92.60% and 75.71% on the NUAA-SIRST and IRSTD-1k datasets, respectively.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Enhancing Infrared Small Target Detection: A Saliency-Guided Multi-Task Learning Approach


    Contributors:


    Publication date :

    2025-03-01


    Size :

    4816157 byte




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English



    Infrared small target detection based on visual saliency

    Zhang Hui / Liu Yan / Zhou Bin et al. | IEEE | 2016


    Robust visual saliency detection method for infrared small target

    Wang, Gang / Chen, Yongguang / Yang, Suochang et al. | British Library Online Contents | 2015


    Investigation of infrared dim and small target detection algorithm based on the visual saliency feature

    Li, Shaoyi / Wang, Xiaotian / Yang, Xi et al. | SAGE Publications | 2021


    Probabilistic Multi-Task Learning for Visual Saliency Estimation in Video

    Li, J. / Tian, Y. / Huang, T. et al. | British Library Online Contents | 2010