A new hybrid model for enhancing low-dose CT images using EfficientNetV2 and WGAN-GP: a multi-loss approach

一种利用 EfficientNetV2 和 WGAN-GP 增强低剂量 CT 图像的新型混合模型:一种多损失方法

阅读:1

Abstract

BACKGROUND: Low-dose computed tomography (LDCT) is widely used for medical imaging due to its reduced radiation exposure. However, LDCT images often suffer from significant noise, which can compromise diagnostic accuracy. This study aims to develop an effective denoising method that preserves critical anatomical structures while reducing noise, using a deep learning approach. METHODS: We propose a novel LDCT image denoising method that integrates EfficientNetV2-M as a multi-scale feature extractor with a Wasserstein generative adversarial network with gradient penalty (WGAN-GP). The EfficientNetV2-M backbone (54.1 M parameters, depth scaling 1.2) employs seven stages of MBConv blocks with expansion ratios from 1 to 6, extracting hierarchical features at stages 3, 5, and 7. The model is optimized using three weighted loss functions: adversarial loss (Wasserstein distance), pixel-wise L1 loss (λ₂ = 1.0), and perceptual loss (λ₃ = 0.1). The discriminator employs gradient penalty with coefficient λ = 10 for training stability. Training employed 64 × 64 patches with batch size 128, Adam optimizer (learning rate: 1e-5) on the AAPM-Mayo Dataset. Image quality was assessed using peak signal-to-noise ratio (PSNR) and structural similarity index measure (SSIM). RESULTS: The proposed method achieved a PSNR of 33.24 ± 0.15 dB and an SSIM of 0.92 ± 0.005 on the AAPM-Mayo Dataset across 10 independent runs, representing improvements of 4.0 dB and 0.04 over baseline LDCT images. Inference speed reached 12.5 FPS (0.08 s per 512 × 512 image) on NVIDIA Tesla T4 GPU, meeting real-time clinical requirements. CONCLUSIONS: Our EfficientNetV2-WGAN-GP-based method provides a robust solution for LDCT image denoising, significantly improving image clarity while maintaining diagnostic structures. This approach holds potential for enhancing diagnostic accuracy and improving patient safety in clinical practice.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。