用迁移学习提升非洲脑瘤影像分割,小数据下效果领先
Adult Glioma Segmentation in Sub-Saharan Africa using Transfer Learning on Stratified Finetuning Data
- 基于分层微调策略,融合预训练模型与本地数据
- 在非洲数据集上实现0.926的全肿瘤分割Dice分数
- 适合医疗资源有限地区开发可落地的AI辅助诊断
胶质瘤是致死率高的脑肿瘤,在低收入和中等收入国家,特别是撒哈拉以南非洲地区,诊断面临巨大挑战。本文提出一种新型胶质瘤分割方法,利用迁移学习应对资源匮乏地区有限且低质量的MRI数据问题。我们采用预训练深度学习模型nnU-Net和MedNeXt,结合BraTS2023-Adult-Glioma与BraTS-Africa数据集,通过放射组学分析构建分层训练折,先在大型脑肿瘤数据集上训练,再迁移到非洲场景。使用加权模型集成与自适应后处理提升分割精度。在BraTS-Africa 2024挑战赛未见验证集上,病灶级平均Dice分数分别为:增强肿瘤0.870、肿瘤核心0.865、全肿瘤0.926,排名首位。该方法展示了集成机器学习技术缩小资源匮乏地区与发达地区医学影像能力差距的潜力。通过针对目标人群需求与限制定制方法,旨在提升孤立环境中的诊断能力。研究强调了本地数据整合与分层优化对缓解医疗不平等、确保实际应用性与提升影响力的重要性。BraTS-Africa 2024优胜算法的Docker版本已发布于https://hub.docker.com/r/aparida12/brats-ssa-2024。
原文摘要 · Abstract (English)
Gliomas, a kind of brain tumor characterized by high mortality, present substantial diagnostic challenges in low- and middle-income countries, particularly in Sub-Saharan Africa. This paper introduces a novel approach to glioma segmentation using transfer learning to address challenges in resource-limited regions with minimal and low-quality MRI data. We leverage pre-trained deep learning models, nnU-Net and MedNeXt, and apply a stratified fine-tuning strategy using the BraTS2023-Adult-Glioma and BraTS-Africa datasets. Our method exploits radiomic analysis to create stratified training folds, model training on a large brain tumor dataset, and transfer learning to the Sub-Saharan context. A weighted model ensembling strategy and adaptive post-processing are employed to enhance segmentation accuracy. The evaluation of our proposed method on unseen validation cases on the BraTS-Africa 2024 task resulted in lesion-wise mean Dice scores of 0.870, 0.865, and 0.926, for enhancing tumor, tumor core, and whole tumor regions and was ranked first for the challenge. Our approach highlights the ability of integrated machine-learning techniques to bridge the gap between the medical imaging capabilities of resource-limited countries and established developed regions. By tailoring our methods to a target population's specific needs and constraints, we aim to enhance diagnostic capabilities in isolated environments. Our findings underscore the importance of approaches like local data integration and stratification refinement to address healthcare disparities, ensure practical applicability, and enhance impact. A dockerized version of the BraTS-Africa 2024 winning algorithm is available at https://hub.docker.com/r/aparida12/brats-ssa-2024 .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。