arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.16860eess.IVcs.CV

医学图像生成与分析中基于 Diffusion 的方法

Diffusion-Based Approaches in Medical Image Generation and Analysis

Islamic University(伊斯兰大学);Texas Tech University(德克萨斯理工大学);University of Oklahoma(俄克拉荷马大学)

Abdullah al Nomaan Nafi, Md. Alamgir Hossain, Rakib Hossain Rifat, Md Mahabub Uz Zaman, Md Manjurul Ahsan, Shivakumar Raman

AI总结:

针对医学图像数据稀缺问题,本研究利用 Diffusion models 在脑肿瘤 MRI、ALL 和 SARS-CoV-2 CT 三个领域生成合成数据集训练 CNN,结果表明分类性能良好且 LIME 分析证实模型关注相关特征,有效减少对真实患者数据的依赖。

摘要:

由于隐私问题,医学成像中的数据稀缺性带来了重大挑战。Diffusion models(扩散模型)作为一种近期的生成式建模技术,通过生成合成且逼真的数据提供了一种潜在的解决方案。然而,关于卷积神经网络(CNN)模型在原始数据集和合成数据集上的性能问题仍然存在。如果 Diffusion 生成的样本能帮助 CNN 模型达到与在原始数据集上训练的模型相当的性能,那么对训练 CNN 时患者特定数据的依赖可能会减少。在本研究中,我们调查了 Diffusion models 在三个领域生成合成医学图像以训练 CNN 的有效性:Brain Tumor MRI、Acute Lymphoblastic Leukemia(ALL)和 SARS-CoV-2 CT 扫描。我们训练了一个 Diffusion model 来为每个领域生成合成数据集。然后,在这些合成数据集上训练预训练的 CNN 架构,并在未见过的真实数据上进行评估。使用在合成数据上训练的 CNN,所有三个数据集都取得了有前景的分类性能。Local Interpretable Model-Agnostic Explanations(LIME)分析显示,这些模型专注于用于分类的相关图像特征。本研究证明了 Diffusion models 生成合成医学图像以用于医学图像分析中训练 CNN 的潜力。

英文摘要:

Data scarcity in medical imaging poses significant challenges due to privacy concerns. Diffusion models, a recent generative modeling technique, offer a potential solution by generating synthetic and realistic data. However, questions remain about the performance of convolutional neural network (CNN) models on original and synthetic datasets. If diffusion-generated samples can help CNN models perform comparably to those trained on original datasets, reliance on patient-specific data for training CNNs might be reduced. In this study, we investigated the effectiveness of diffusion models for generating synthetic medical images to train CNNs in three domains: Brain Tumor MRI, Acute Lymphoblastic Leukemia (ALL), and SARS-CoV-2 CT scans. A diffusion model was trained to generate synthetic datasets for each domain. Pre-trained CNN architectures were then trained on these synthetic datasets and evaluated on unseen real data. All three datasets achieved promising classification performance using CNNs trained on synthetic data. Local Interpretable Model-Agnostic Explanations (LIME) analysis revealed that the models focused on relevant image features for classification. This study demonstrates the potential of diffusion models to generate synthetic medical images for training CNNs in medical image analysis.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑