An Infrared Image Denoising Method Based on Wavelet Dual-Branch Diffusion Model

Removing noise from infrared images is of great significance in various fields. The existing infrared image denoising methods have the problem of imbalance between noise suppression and detail preservation. In this paper, we propose a wavelet dual-branch diffusion infrared image denoising method, namely WD-Diff. Firstly, a diffusion process with noisy images as the endpoint is constructed. In the reverse recovery process, we propose a new image restoration network. Among them, the low frequency restoration network (LFRNet) and the high frequency restoration network (HFRNet) are designed to restore wavelet low-frequency and high-frequency degradation subgraphs, respectively. The two networks are separately trained under two-step training strategy and single step training strategy. Secondly, the self-attention residual module (SARM) and the equidistant convolution residual module (ECRM) are specially designed and applied to LFRNet and HFRNet, respectively. Through self-attention mechanism and residual learning, SARM can effectively extract low frequency feature information. By constructing equidistant convolution large kernel, ECRM can accurately remove high frequency noise. Thirdly, we design a high-low frequency fusion Transformer network (HLFTNet), which achieves high-low frequency feature fusion through spatio-temporal cross fusion attention mechanism. Finally, extensive comparative and ablation experiments have demonstrated the superiority and effectiveness of the proposed method.