一个可同时分割、分型和分级胶质瘤的多模态影像基础模型。
Towards a Multimodal MRI-Based Foundation Model for Multi-Level Feature Exploration in Segmentation, Molecular Subtyping, and Grading of Glioma
- 基于预训练框架,融合肿瘤形态与分子特征进行多任务学习。
- 分割Dice达84%,IDH突变预测AUC超90%,1p/19q共缺失预测达69%。
- 适合医学影像研究者和临床辅助诊断系统开发者使用。
准确、非侵入性地刻画胶质瘤对临床管理至关重要。传统方法依赖侵入性组织采样,难以捕捉肿瘤空间异质性。尽管深度学习提升了分割与分子分析能力,但极少有方法能同时整合形态与分子特征。基础深度学习模型从大规模数据中学习通用表征,前景广阔但尚未在胶质瘤影像生物标志物中充分应用。本文提出多任务SWIN-UNET(MTS-UNET)模型,基于BrainSegFounder预训练框架,在7个公共数据集共2,249例患者上训练。该模型同时完成胶质瘤分割、组织学分级及分子亚型预测(IDH突变与1p/19q共缺失)。引入两个关键模块:肿瘤感知特征编码(TAFE)实现多尺度肿瘤聚焦特征提取,跨模态差异模块(CMD)突出与IDH突变相关的微弱T2-FLAIR不匹配信号。模型在测试中达到84%平均分割Dice分数,以及IDH突变预测AUC 90.58%、1p/19q共缺失预测AUC 69.22%、分级预测AUC 87.54%,显著优于基线模型(p≤0.05)。消融实验验证了各模块关键作用及框架鲁棒性。该基础模型有效整合分割与多层次分类,具有跨多中心数据集强泛化能力,有望推动非侵入性个性化胶质瘤管理。
原文摘要 · Abstract (English)
Accurate, noninvasive glioma characterization is crucial for effective clinical management. Traditional methods, dependent on invasive tissue sampling, often fail to capture the spatial heterogeneity of the tumor. While deep learning has improved segmentation and molecular profiling, few approaches simultaneously integrate tumor morphology and molecular features. Foundation deep learning models, which learn robust, task-agnostic representations from large-scale datasets, hold great promise but remain underutilized in glioma imaging biomarkers. We propose the Multi-Task SWIN-UNETR (MTS-UNET) model, a novel foundation-based framework built on the BrainSegFounder model, pretrained on large-scale neuroimaging data. MTS-UNET simultaneously performs glioma segmentation, histological grading, and molecular subtyping (IDH mutation and 1p/19q co-deletion). It incorporates two key modules: Tumor-Aware Feature Encoding (TAFE) for multi-scale, tumor-focused feature extraction and Cross-Modality Differential (CMD) for highlighting subtle T2-FLAIR mismatch signals associated with IDH mutation. The model was trained and validated on a diverse, multi-center cohort of 2,249 glioma patients from seven public datasets. MTS-UNET achieved a mean Dice score of 84% for segmentation, along with AUCs of 90.58% for IDH mutation, 69.22% for 1p/19q co-deletion prediction, and 87.54% for grading, significantly outperforming baseline models (p<=0.05). Ablation studies validated the essential contributions of the TAFE and CMD modules and demonstrated the robustness of the framework. The foundation-based MTS-UNET model effectively integrates tumor segmentation with multi-level classification, exhibiting strong generalizability across diverse MRI datasets. This framework shows significant potential for advancing noninvasive, personalized glioma management by improving predictive accuracy and interpretability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。