MLAR-UNet: LDCT image denoising based on U-Net with multiple lightweight attention-based modules and residual reinforcement.
Journal:
Physics in medicine and biology
PMID:
39899989
Abstract
Computed tomography (CT) is a crucial medical imaging technique which uses x-ray radiation to identify cancer tissues. Since radiation poses a significant health risk, low dose acquisition procedures need to be adopted. However, low-dose CT (LDCT) can cause higher noise and artifacts which massively degrade the diagnosis.To denoise LDCT images more effectively, this paper proposes a deep learning method based on U-Net with multiple lightweight attention-based modules and residual reinforcement (MLAR-UNet). We integrate a U-Net architecture with several advanced modules, including Convolutional Block Attention Module (CBAM), Cross Residual Module (CR), Attention Cross Reinforcement Module (ACRM), and Convolution and Transformer Cross Attention Module (CTCAM). Among these modules, CBAM applies channel and spatial attention mechanisms to enhance local feature representation. However, serious detail loss caused by incorrect embedding of CBAM for LDCT denoising is verified in this study. To relieve this, we introduce CR to reduce information loss in deeper layers, preserving features more effectively. To address the excessive local attention of CBAM, we design ACRM, which incorporates Transformer to adjust the attention weights. Furthermore, we design CTCAM, which leverages a complex combination of Transformer and convolution to capture multi-scale information and compute more accurate attention weights.Experiments verify the embedding rationality and validity of each module and show that the proposed MLAR-UNet denoises LDCT images more effectively and preserves more details than many state-of-the-art methods on clinical chest and abdominal CT datasets.The proposed MLAR-UNet not only demonstrates superior LDCT image denoising capability but also highlights the strong detail comprehension and negligible overheads of our designed ACRM and CTCAM. These findings provide a novel approach to integrating Transformer more efficiently in image processing.