MDAFNet: Multiscale Differential Edge and Adaptive Frequency Guided Network for Infrared Small Target Detection
Shuying Li, Qiang Ma, San Zhang, Wuwei Wang, Chuang Yang
TL;DR
Problem: infrared small target detection (IRSTD) suffers from cumulative edge degradation with network depth and insufficient frequency processing to separate high-frequency targets from high-frequency noise. Approach: MDAFNet integrates a Multi-Scale Differential Edge (MSDE) module and a Dual-Domain Adaptive Feature Enhancement (DAFE) module within a U-shaped encoder–decoder, leveraging Haar wavelet decomposition and adaptive strip pooling for frequency-aware enhancement. Contributions: MSDE preserves edge geometry across scales via differential edge operations and attention refinements, while DAFE performs adaptive frequency modulation to boost target signals and suppress noise; the method yields state-of-the-art performance on multiple IRSTD benchmarks, with improvements in $IoU$, $P_d$, and reductions in $F_a$. Significance: the framework improves detection reliability in cluttered infrared scenes and motivates future lightweight, real-time variants.
Abstract
Infrared small target detection (IRSTD) plays a crucial role in numerous military and civilian applications. However, existing methods often face the gradual degradation of target edge pixels as the number of network layers increases, and traditional convolution struggles to differentiate between frequency components during feature extraction, leading to low-frequency backgrounds interfering with high-frequency targets and high-frequency noise triggering false detections. To address these limitations, we propose MDAFNet (Multi-scale Differential Edge and Adaptive Frequency Guided Network for Infrared Small Target Detection), which integrates the Multi-Scale Differential Edge (MSDE) module and Dual-Domain Adaptive Feature Enhancement (DAFE) module. The MSDE module, through a multi-scale edge extraction and enhancement mechanism, effectively compensates for the cumulative loss of target edge information during downsampling. The DAFE module combines frequency domain processing mechanisms with simulated frequency decomposition and fusion mechanisms in the spatial domain to effectively improve the network's capability to adaptively enhance high-frequency targets and selectively suppress high-frequency noise. Experimental results on multiple datasets demonstrate the superior detection performance of MDAFNet.
