arXiv:2412.10826eess.IVcs.AI2024-12被引 9

用Pix2pix-GAN实现胸部X光肺部高效精准分割,助力早期疾病发现。

Generative AI: A Pix2pix-GAN-Based Machine Learning Approach for Robust and Efficient Lung Segmentation

  • 基于U-Net结构的Pix2pix-GAN框架,结合对抗损失与L1损失优化
  • 在Montgomery和Shenzhen数据集上均达到高精度,Dice系数表现优异
  • 适用于医学影像中的肺部异常分割,适合临床辅助诊断研究

胸片在识别肺部疾病中至关重要,但放射科医生工作负荷大、效率低可能导致误诊。自动、准确、高效的肺部分割对早期疾病检测极为关键。本研究提出一种基于Pix2pix生成对抗网络(GAN)的深度学习框架,用于从胸部X光片(CXR)中分割肺部异常。该框架整合了图像预处理与增强技术,并采用受U-Net启发的生成器-判别器架构。首先从Montgomery和Shenzhen数据集加载原始图像与人工标注掩码,进行预处理与尺寸调整;随后使用U-Net生成器输出分割掩码,判别器区分生成掩码与真实掩码。以Montgomery数据集训练模型,用Shenzhen数据集测试其鲁棒性,后者首次被用于此类研究。训练中采用对抗损失与L1距离联合优化。所有评估指标(精确率、召回率、F1分数、Dice系数)均证明该框架在肺部异常分割中的有效性,为后续利用多样化数据集验证其临床应用价值奠定了基础。

原文摘要 · Abstract (English)

Chest radiography is climacteric in identifying different pulmonary diseases, yet radiologist workload and inefficiency can lead to misdiagnoses. Automatic, accurate, and efficient segmentation of lung from X-ray images of chest is paramount for early disease detection. This study develops a deep learning framework using a Pix2pix Generative Adversarial Network (GAN) to segment pulmonary abnormalities from CXR images. This framework's image preprocessing and augmentation techniques were properly incorporated with a U-Net-inspired generator-discriminator architecture. Initially, it loaded the CXR images and manual masks from the Montgomery and Shenzhen datasets, after which preprocessing and resizing were performed. A U-Net generator is applied to the processed CXR images that yield segmented masks; then, a Discriminator Network differentiates between the generated and real masks. Montgomery dataset served as the model's training set in the study, and the Shenzhen dataset was used to test its robustness, which was used here for the first time. An adversarial loss and an L1 distance were used to optimize the model in training. All metrics, which assess precision, recall, F1 score, and Dice coefficient, prove the effectiveness of this framework in pulmonary abnormality segmentation. It, therefore, sets the basis for future studies to be performed shortly using diverse datasets that could further confirm its clinical applicability in medical imaging.

肺部分割Pix2pix-GAN医学影像深度学习

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