arXiv:2410.18698eess.IVcs.CV2024-10被引 4

用高质量MRI数据训练模型,提升非洲低质量影像的脑胶质瘤诊断准确率。

Transferring Knowledge from High-Quality to Low-Quality MRI for Adult Glioma Diagnosis

  • 从高质量数据集预训练,再在非洲本地数据微调,提升模型泛化能力。
  • 在验证集上达到0.926的全肿瘤Dice分数,优于纯本地训练方案。
  • 为资源有限地区提供可复现的AI辅助诊断范式,适合医疗资源匮乏地区使用。

脑胶质瘤是常见且致命的脑部肿瘤,早期诊断对改善预后至关重要。然而,撒哈拉以南非洲(SSA)地区的低质量磁共振成像(MRI)技术阻碍了准确诊断。本文参与了BraTS挑战赛中的SSA成人胶质瘤诊断任务。我们采用BraTS-GLI 2021冠军方案的模型,并尝试三种训练策略:(1)先在BraTS-GLI 2021数据集上训练,再在BraTS-Africa数据集上微调;(2)仅在BraTS-Africa数据集上训练;(3)仅在BraTS-Africa数据集上训练并结合2倍超分辨率增强。结果表明,先在高质量数据集上预训练再微调的策略表现最佳。该方法在验证阶段取得0.882、0.840和0.926的增强肿瘤、肿瘤核心和全肿瘤的Dice分数,以及15.324、37.518和13.971的Hausdorff距离(95%)得分。在竞赛最终阶段,该方法获得总体第二名,验证了模型与训练策略的有效性。本研究为改善非洲地区胶质瘤诊断提供了新思路,展示了深度学习在资源受限环境中的潜力,以及从高质量数据集迁移知识的重要性。

原文摘要 · Abstract (English)

Glioma, a common and deadly brain tumor, requires early diagnosis for improved prognosis. However, low-quality Magnetic Resonance Imaging (MRI) technology in Sub-Saharan Africa (SSA) hinders accurate diagnosis. This paper presents our work in the BraTS Challenge on SSA Adult Glioma. We adopt the model from the BraTS-GLI 2021 winning solution and utilize it with three training strategies: (1) initially training on the BraTS-GLI 2021 dataset with fine-tuning on the BraTS-Africa dataset, (2) training solely on the BraTS-Africa dataset, and (3) training solely on the BraTS-Africa dataset with 2x super-resolution enhancement. Results show that initial training on the BraTS-GLI 2021 dataset followed by fine-tuning on the BraTS-Africa dataset has yielded the best results. This suggests the importance of high-quality datasets in providing prior knowledge during training. Our top-performing model achieves Dice scores of 0.882, 0.840, and 0.926, and Hausdorff Distance (95%) scores of 15.324, 37.518, and 13.971 for enhancing tumor, tumor core, and whole tumor, respectively, in the validation phase. In the final phase of the competition, our approach successfully secured second place overall, reflecting the strength and effectiveness of our model and training strategies. Our approach provides insights into improving glioma diagnosis in SSA, showing the potential of deep learning in resource-limited settings and the importance of transfer learning from high-quality datasets.

脑胶质瘤医学影像迁移学习AI医疗

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