用MedNeXt模型提升脑肿瘤分割精度,适配非洲与儿童群体。
Optimizing Brain Tumor Segmentation with MedNeXt: BraTS 2024 SSA and Pediatrics
- 基于MedNeXt架构,结合集成学习与后处理优化分割效果。
- 在非洲和儿科数据集上分别取得0.896和0.830的Dice系数。
- 方法对不同人群和影像质量差异有良好鲁棒性,适合临床部署。
识别脑部MRI中的关键病理特征对胶质瘤患者长期生存至关重要。然而,人工分割耗时且易受人为误差影响。为此,研究者致力于开发能准确分割3D多模态脑部MRI中肿瘤的机器学习方法。尽管已有显著进展,当前先进模型常受限于训练数据,当应用于不同人群时可能因分布偏移导致可靠性下降,例如来自撒哈拉以南非洲地区低质量MRI设备或儿童患者群体的差异。2024年BraTS挑战赛为此提供评估平台。本研究采用MedNeXt模型、全面的模型集成及细致的后处理策略,完成BraTS-2024 SSA与儿科肿瘤任务的分割。在未见验证集上表现优异:BraTS-2024 SSA数据集平均Dice相似性系数(DSC)达0.896,平均95%豪斯多夫距离(HD95)为14.682;儿科数据集平均DSC为0.830,平均HD95为37.508。项目代码已开源。
原文摘要 · Abstract (English)
Identifying key pathological features in brain MRIs is crucial for the long-term survival of glioma patients. However, manual segmentation is time-consuming, requiring expert intervention and is susceptible to human error. Therefore, significant research has been devoted to developing machine learning methods that can accurately segment tumors in 3D multimodal brain MRI scans. Despite their progress, state-of-the-art models are often limited by the data they are trained on, raising concerns about their reliability when applied to diverse populations that may introduce distribution shifts. Such shifts can stem from lower quality MRI technology (e.g., in sub-Saharan Africa) or variations in patient demographics (e.g., children). The BraTS-2024 challenge provides a platform to address these issues. This study presents our methodology for segmenting tumors in the BraTS-2024 SSA and Pediatric Tumors tasks using MedNeXt, comprehensive model ensembling, and thorough postprocessing. Our approach demonstrated strong performance on the unseen validation set, achieving an average Dice Similarity Coefficient (DSC) of 0.896 on the BraTS-2024 SSA dataset and an average DSC of 0.830 on the BraTS Pediatric Tumor dataset. Additionally, our method achieved an average Hausdorff Distance (HD95) of 14.682 on the BraTS-2024 SSA dataset and an average HD95 of 37.508 on the BraTS Pediatric dataset. Our GitHub repository can be accessed here: Project Repository : https://github.com/python-arch/BioMbz-Optimizing-Brain-Tumor-Segmentation-with-MedNeXt-BraTS-2024-SSA-and-Pediatrics
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。