Infrared Technology, Volume. 47, Issue 6, 757(2025)

Diffusion Model for Infrared Small Target Detection

Chenhao TU1, Wenya YE1、*, Nini DU2, Binhao ZHENG1, and Sheng XU2
Author Affiliations
  • 1School of Architecture and Transportation Engineering, Ningbo Institute of Technology, Ningbo 315211, China
  • 2School of Architecture and Art, Zhejiang Business Technology Institute, Ningbo 315100, China
  • show less

    Infrared small-target detection, a complex and critical task in computer vision, faces numerous challenges—including tiny target sizes, low contrast, severe background noise, and limited data availability. These factors significantly impair detection accuracy and real-time performance. Existing deep learning–based algorithms, which predominantly adopt segmentation paradigms via deep encoder–decoder architectures for generating segmentation masks, often exhibit limited precision in complex scenarios due to inadequate feature representation and learning capabilities. Inspired by the notable success of diffusion models in artificial intelligence, this paper introduces a novel approach by reframing infrared small-target detection as a generative task and proposes a conditional denoising network, termed diff-ISTD. By leveraging the strengths of progressive denoising and image reconstruction, diff-ISTD captures the deep statistical properties of infrared images, enabling more precise identification of weak and ambiguous small-target features. The proposed network consists of conditional branching modules for extracting prior knowledge from infrared inputs and denoising branches for refining noisy segmentation masks. In addition, a parallel dual-dimensional self-attention (PDSA) block is introduced to integrate spatial and channel information, significantly enhancing the model's sensitivity to global structures and local details. This design effectively addresses the challenges of target blurring caused by resolution limitations and environmental variability. Comprehensive experiments demonstrate that, under rigorous detection conditions, diff-ISTD outperforms current state-of-the-art segmentation methods in terms of performance and detection efficiency, offering a promising direction for advancing infrared small-target detection technologies.

    Tools

    Get Citation

    Copy Citation Text

    TU Chenhao, YE Wenya, DU Nini, ZHENG Binhao, XU Sheng. Diffusion Model for Infrared Small Target Detection[J]. Infrared Technology, 2025, 47(6): 757

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: May. 24, 2024

    Accepted: Jul. 3, 2025

    Published Online: Jul. 3, 2025

    The Author Email: YE Wenya (763425011@qq.com)

    DOI:

    Topics