基于多尺度双分支Transformer的图像去噪方法

Image denoising method based on multi-scale dual-branch Transformer

  • 摘要: 针对现有图像去噪算法在特征处理阶段忽视对局部特征的充分利用、无法有效恢复边缘细节、甚至导致失真的问题,提出了一种基于多尺度双分支Transformer的图像去噪方法. 首先,该方法设计了一种双分支的Transformer结构,有效融合浅层和深层特征的多尺度信息,在恢复图像细节的同时保留整体的特征;其次,加入残差块组以缓解因网络加深可能导致梯度消失的问题;最后,应用极化自注意力机制,提高模型对多尺度特征的感知能力,并在参数量受控的情况下减少下采样过程中特征信息的丢失. 实验结果表明,相较于现有主流图像去噪方法,新方法不仅能有效去除图像噪声,还能够恢复出更精细的纹理效果,在定性和定量分析中均表现出优异的去噪性能.

     

    Abstract: A novel image denoising method based on a multi-scale dual-branch Transformer is proposed to address the limitations of existing algorithms, such as insufficient utilization of local features and the inability to effectively restore edge details, leading to distortions. Firstly, this method designs a dual-branch Transformer structure that effectively integrates multi-scale information from both shallow and deep features, allowing the preservation of global characteristics while restoring image details. Secondly, residual blocks are introduced to mitigate the potential gradient vanishing problem caused by network depth. Finally, a polarized self-attention mechanism is applied to enhance the model’s perception of multi-scale features, reducing the loss of feature information during down-sampling, while controlling the number of parameters. Experimental results demonstrate that the proposed method not only effectively removes noise but also restores finer texture details, outperforming existing mainstream denoising methods in both qualitative and quantitative evaluations.

     

/

返回文章
返回