MCDNet: An Infrared Small Target Detection Network Using Multi-Criteria Decision and Adaptive Labeling Strategy
Tianlei Ma,Qi Ma,Zhen Yang,Jing Liang,Jun Fu,Yu Dou,Yanan Ku,Usman Ahmad,Liangqiong Qu
DOI: https://doi.org/10.1109/tgrs.2024.3368059
IF: 8.2
2024-03-15
IEEE Transactions on Geoscience and Remote Sensing
Abstract:The success of deep learning methods heavily relies on the availability of adequate samples. However, in the task of infrared small target detection (ISTD), the lack of high-quality training samples is a challenging problem due to the confidentiality of the application field and the difficulty of labeling. This limitation often leads to suboptimal detection performance of convolutional neural networks (CNNs). To address this challenge, we propose an adaptive labeling strategy and an ISTD network called the multi-criteria decision network (MCDNet) to achieve higher-quality sample labeling and more accurate detection results. In the adaptive labeling strategy, we propose a second-order differential autocorrelation method to determine the size of fuzzy edge targets accurately. In addition, we introduce local backgrounds to enhance the saliency information in the labels and improve the richness and contrast of training label content. To obtain accurate and robust detection results with limited target feature information, we design MCDNet. In particular, we propose a multi-criteria decision (MCD) method that can combine the CNN decisions and the infrared small target prior saliency decisions through weighted fusion, and set decision weights based on the importance of different decision criteria in the decision-making process. This method can integrate the advantages of both the CNN decisions and the prior saliency decisions, avoid the one-sidedness of a single criterion, and improve the reliability and stability of the decision-making. The experimental results indicate that our method has a higher accuracy compared to other contrastive methods.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics