arXiv:2502.15204eess.IVcs.CV2025-02中稿 · IEEE Transactions …被引 11

用语义布局引导生成肺部CT图像,解决医疗数据稀缺问题。

Lung-DDPM: Semantic Layout-guided Diffusion Models for Thoracic CT Image Synthesis

  • 基于语义布局的扩散模型,可从不完整布局生成解剖合理图像
  • 生成图像在质量评估中优于现有方法7.4至29.5倍
  • 合成数据提升肺结节分割性能,提升18.6%敏感度

随着人工智能快速发展,辅助医学影像分析在肺癌早期筛查中表现突出。然而,高昂的标注成本和隐私问题限制了大规模医疗数据集的构建,制约了AI在医疗中的进一步应用。为缓解肺癌筛查中的数据稀缺问题,我们提出Lung-DDPM,一种基于语义布局引导的去噪扩散概率模型(DDPM),用于生成高质量3D合成胸部CT图像,显著提升下游肺结节分割任务表现。该方法即使在不完整语义布局条件下,也能生成解剖合理、无缝且一致的样本。实验结果表明,Lung-DDPM在图像质量评估和下游任务中均优于其他先进生成模型。在大型验证队列上,其弗雷切特起始距离(FID)为0.0047,最大均值差异(MMD)为0.0070,均方误差(MSE)为0.0024,分别比第二佳方案提升7.4×、3.1×和29.5×。使用真实与合成样本联合训练的肺结节分割模型,获得Dice系数0.3914和灵敏度0.4393,较仅用真实数据训练的模型分别提升8.8%和18.6%。实验表明Lung-DDPM在肿瘤分割、生存预测等医疗影像任务中具有广泛应用潜力。代码与预训练模型已开源。

原文摘要 · Abstract (English)

With the rapid development of artificial intelligence (AI), AI-assisted medical imaging analysis demonstrates remarkable performance in early lung cancer screening. However, the costly annotation process and privacy concerns limit the construction of large-scale medical datasets, hampering the further application of AI in healthcare. To address the data scarcity in lung cancer screening, we propose Lung-DDPM, a thoracic CT image synthesis approach that effectively generates high-fidelity 3D synthetic CT images, which prove helpful in downstream lung nodule segmentation tasks. Our method is based on semantic layout-guided denoising diffusion probabilistic models (DDPM), enabling anatomically reasonable, seamless, and consistent sample generation even from incomplete semantic layouts. Our results suggest that the proposed method outperforms other state-of-the-art (SOTA) generative models in image quality evaluation and downstream lung nodule segmentation tasks. Specifically, Lung-DDPM achieved superior performance on our large validation cohort, with a Fréchet inception distance (FID) of 0.0047, maximum mean discrepancy (MMD) of 0.0070, and mean squared error (MSE) of 0.0024. These results were 7.4$\times$, 3.1$\times$, and 29.5$\times$ better than the second-best competitors, respectively. Furthermore, the lung nodule segmentation model, trained on a dataset combining real and Lung-DDPM-generated synthetic samples, attained a Dice Coefficient (Dice) of 0.3914 and sensitivity of 0.4393. This represents 8.8% and 18.6% improvements in Dice and sensitivity compared to the model trained solely on real samples. The experimental results highlight Lung-DDPM's potential for a broader range of medical imaging applications, such as general tumor segmentation, cancer survival estimation, and risk prediction. The code and pretrained models are available at https://github.com/Manem-Lab/Lung-DDPM/.

医学影像扩散模型数据生成

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