arXiv:2502.14928eess.IVcs.CV2025-02

用分布式U-Net提升肺癌肺部影像分割精度,助力早期诊断。

Distributed U-net model and Image Segmentation for Lung Cancer Detection

  • 采用分布式GPU与联邦学习优化U-Net模型训练效率。
  • 四块GPU并行时分割准确率达96.3%,显著优于单机配置。
  • 适合医疗AI研发、医院影像科及深度学习部署团队参考。

新冠疫情后,肺部疾病如肺癌和慢性阻塞性肺病(COPD)已成为全球重大健康问题。早期检测与精准诊断对治疗效果至关重要。本研究探索计算机辅助诊断(CAD)系统在肺部影像分析中的应用,重点评估U-Net模型在肺部CT图像分割任务中的表现。基于多所高校合作构建的肺部CT图像及对应分割掩码数据集,研究在单CPU、单GPU、分布式GPU和联邦学习等多种硬件配置下对U-Net进行严格测试。结果表明,使用四块GPU进行分布式训练时,模型性能最优,分割准确率达到96.3%,验证了基于U-Net的CAD系统在肺部疾病早期检测中的巨大潜力。

原文摘要 · Abstract (English)

Until now, in the wake of the COVID-19 pandemic in 2019, lung diseases, especially diseases such as lung cancer and chronic obstructive pulmonary disease (COPD), have become an urgent global health issue. In order to mitigate the goal problem, early detection and accurate diagnosis of these conditions are critical for effective treatment and improved patient outcomes. To further research and reduce the error rate of hospital diagnoses, this comprehensive study explored the potential of computer-aided design (CAD) systems, especially utilizing advanced deep learning models such as U-Net. And compared with the literature content of other authors, this study explores the capabilities of U-Net in detail, and enhances the ability to simulate CAD systems through the VGG16 algorithm. An extensive dataset consisting of lung CT images and corresponding segmentation masks, curated collaboratively by multiple academic institutions, serves as the basis for empirical validation. In this paper, the efficiency of U-Net model is evaluated rigorously and precisely under multiple hardware configurations, such as single CPU, single GPU, distributed GPU and federated learning, and the effectiveness and development of the method in the segmentation task of lung disease are demonstrated. Empirical results clearly affirm the robust performance of the U-Net model, most effectively utilizing four GPUs for distributed learning, and these results highlight the potential of U-Net-based CAD systems for accurate and timely lung disease detection and diagnosis huge potential.

医学影像U-Net分割分布式

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。