Generative Adversarial Differential Analysis for Infrared Small Target Detection
Zongfang Ma, Shuo Pang, Fan Hao
Abstract
Infrared small target detection refers to the extraction of small targets in a complex, low signal-to-noise ratio background. Depthwise convolution makes it difficult to comprehensively characterize small infrared targets and ignores the importance of image background for the detection task. In this paper, we propose the generative adversarial differential analysis (GADA) model for infrared small target detection, the core of which aims to weaken the reliance on target features and enhance the use of background information. Specifically, we first construct pseudo background labels by the fast marching method. Then, the background-guided generative adversarial network is used to learn the background data distribution. On this basis, the differential image containing interest regions of small targets is obtained by differential analysis. Finally, the detection results are obtained by performing an elaborate characterization of the interest regions. The effectiveness of GADA is verified with three public datasets. Compared to several state-of-the-art methods, GADA achieves better performance in terms of <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$F1$</tex-math></inline-formula> , <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$IoU$</tex-math></inline-formula> , and <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$AUC$</tex-math></inline-formula> .