用分布式训练加速肺部X光去噪,兼顾隐私保护与效率。
Distributed Deep Learning for Medical Image Denoising with Data Obfuscation
- 采用分布式训练+高斯噪声遮蔽,提升医疗图像处理速度。
- 优化后训练时间减少60%以上,性能损失微小。
- 适合需要快速处理大规模医学影像的临床研究场景。
医学图像去噪对提升图像质量并降低敏感信息泄露风险至关重要,尤其在大规模临床数据集上。本研究针对NIH Chest X-ray14数据集中的胸部X光图像,探索分布式深度学习方法,采用加性高斯噪声作为轻量级数据遮蔽技术。在PyTorch中实现并评估U-Net和U-Net++模型,对比单GPU、标准多GPU(DataParallel)及优化的多GPU训练配置(使用DistributedDataParallel和Automatic Mixed Precision)。结果表明,U-Net++在所有噪声水平下均表现更优,达到有竞争力的峰值信噪比(PSNR)和结构相似性指数(SSIM),但在低中等噪声水平下感知相似度(LPIPS)高于U-Net,体现其更强的结构保真能力。同时,优化训练流程使两种模型的训练时间较单GPU减少60%以上,相较标准DataParallel提升超40%,仅带来轻微精度下降。这些发现证明了软件级优化在分布式学习中的有效性。本工作展示了结合架构设计、轻量遮蔽与先进分布式训练策略,在真实临床与科研环境中加速并增强医学图像处理的可行性。完整代码已公开:https://github.com/Suadey/medical-image-denoising-ddp。
原文摘要 · Abstract (English)
Medical image denoising is essential for improving image quality while minimizing the exposure of sensitive information, particularly when working with large-scale clinical datasets. This study explores distributed deep learning for denoising chest X-ray images from the NIH Chest X-ray14 dataset, using additive Gaussian noise as a lightweight obfuscation technique. We implement and evaluate U-Net and U-Net++ architectures under single-GPU, standard multi-GPU (DataParallel), and optimized multi-GPU training configurations using PyTorch's DistributedDataParallel (DDP) and Automatic Mixed Precision (AMP). Our results show that U-Net++ consistently delivers superior denoising performance, achieving competitive Peak Signal to Noise Ratio (PSNR) and Structured Similarity Index Method (SSIM) scores, though with less performance in Learned Perceptual Image Patch Similarity (LPIPS) compared to U-Net under low and moderate noise levels. This indicates U-Net++'s enhanced structural fidelity and low perceptual similarity. Meanwhile, our optimized training pipeline reduces training time by over 60% for both models compared to single-GPU training, and outperforms standard DataParallel by over 40%, with only a minor accuracy drop for both models (trading some accuracy for speed). These findings highlight the effectiveness of software-level optimization in distributed learning for medical imaging. This work demonstrates the practical viability of combining architectural design, lightweight obfuscation, and advanced distributed training strategies to accelerate and enhance medical image processing pipelines in real-world clinical and research environments. The full implementation is publicly available at: https://github.com/Suadey/medical-image-denoising-ddp.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。