DWTFreqNet: Infrared Small Target Detection via Wavelet-Driven Frequency Matching and Saliency-Difference Optimization
Qianwen Ma, Shangwei Deng, Bincheng Li, Zhen Zhu, Ziying Song, Xiaobo Li, Haofeng Hu
Abstract
In the field of infrared small target detection, targets generally exhibit dim characteristics, and difficult to distinguish from background clutter. Learning-based methods enhance feature representation through layer-by-layer propagation, but the sparse target information often diminishes. To address this, we propose DWTFreqNet, a network that splits input data to enhance both local saliency and global contextual differences. It incorporates complementary feature extraction modules designed to match the data distribution characteristics. Specifically, it first utilizes the discrete wavelet transform (DWT) to decompose the input data into low- and high-frequency components. For the low-frequency part, which carries key target information, we apply component-differential dense connections and DWT-based downsampling to maintain feature integrity. For the high-frequency part, rich in target-background contrast, an Adaptive Wavelet Guidance Mechanism optimizes multi-component fusion via adaptive weighting, while a Layer-wide Discrepancy Relationship Capture Module enhances target discrimination by linking multi-scale feature maps. Comparative experiments on public datasets demonstrate its superiority over state-of-the-art methods. The code will be available at https://github.com/Kingwin97/DWTFreqNet.