用深度学习自动识别8类腹部多参数MRI序列,提升放射科读片效率。
Classification of Multi-Parametric Body MRI Series Using Deep Learning
- 基于ResNet、EfficientNet、DenseNet训练分类模型,筛选最优架构
- DenseNet-121在内部数据上达0.966 F1-score和0.972准确率
- 模型在外部数据集上仍保持0.81~0.87准确率,适合跨机构部署
多参数磁共振成像(mpMRI)检查包含多种不同成像协议的序列,其DICOM头信息常因协议多样性及操作员失误而错误。为此,本文提出一种基于深度学习的分类模型,用于自动识别8类体部mpMRI序列,以提升放射科医生阅片效率。利用来自多个医疗机构的mpMRI数据,训练了ResNet、EfficientNet和DenseNet等多种深度学习分类器并进行对比。结果表明,DenseNet-121模型在所有模型中表现最佳,内部测试集F1-score为0.966,准确率为0.972(p<0.05)。当训练数据量超过729例时,模型准确率稳定高于0.95,且性能随数据量增加而提升。在外部数据集DLDS和CPTAC-UCEC上,模型分别取得0.872和0.810的准确率。结果表明,DenseNet-121在内部与外部数据上均能高效完成8类体部MRI序列分类任务。
原文摘要 · Abstract (English)
Multi-parametric magnetic resonance imaging (mpMRI) exams have various series types acquired with different imaging protocols. The DICOM headers of these series often have incorrect information due to the sheer diversity of protocols and occasional technologist errors. To address this, we present a deep learning-based classification model to classify 8 different body mpMRI series types so that radiologists read the exams efficiently. Using mpMRI data from various institutions, multiple deep learning-based classifiers of ResNet, EfficientNet, and DenseNet are trained to classify 8 different MRI series, and their performance is compared. Then, the best-performing classifier is identified, and its classification capability under the setting of different training data quantities is studied. Also, the model is evaluated on the out-of-training-distribution datasets. Moreover, the model is trained using mpMRI exams obtained from different scanners in two training strategies, and its performance is tested. Experimental results show that the DenseNet-121 model achieves the highest F1-score and accuracy of 0.966 and 0.972 over the other classification models with p-value$<$0.05. The model shows greater than 0.95 accuracy when trained with over 729 studies of the training data, whose performance improves as the training data quantities grew larger. On the external data with the DLDS and CPTAC-UCEC datasets, the model yields 0.872 and 0.810 accuracy for each. These results indicate that in both the internal and external datasets, the DenseNet-121 model attains high accuracy for the task of classifying 8 body MRI series types.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。