多任务学习实现胶质瘤多模态MRI的联合诊断与可视化分析
Glioma Multimodal MRI Analysis System for Tumor Layered Diagnosis via Multi-task Semi-supervised Learning
- 基于不确定性多任务学习,同步完成肿瘤分割、分型、基因突变等诊断任务
- 在未标注数据上采用两阶段半监督学习,提升模型性能并减少对标注依赖
- 支持模态缺失场景,可生成个性化预后建议,适合临床医生和患者使用
胶质瘤是中枢神经系统最常见的原发性肿瘤。多模态MRI广泛用于胶质瘤的初步筛查,在辅助诊断、疗效评估和预后判断中起关键作用。现有研究多独立处理肿瘤分割、分级和放射基因组分类等任务,忽视了各任务间的关联性。本文提出胶质瘤多模态MRI分析系统(GMMAS),通过深度学习网络同时处理多个诊断任务,利用基于不确定性的多任务学习架构,同步输出肿瘤区域分割、组织学亚型、IDH突变基因型及1p/19q染色体缺失状态。相比单任务模型,GMMAS在各项诊断任务中均提升了精度。此外,采用两阶段半监督学习方法,充分利用有标签与无标签MRI样本,增强模型表现。通过基于知识自蒸馏与对比学习的跨模态特征提取适配模块,GMMAS在模态缺失情况下仍具鲁棒性,并揭示了不同MRI模态的重要性差异。基于GMMAS分析结果,我们构建了面向医患的可视化平台,引入GMMAS-GPT生成个性化预后评估与建议。
原文摘要 · Abstract (English)
Gliomas are the most common primary tumors of the central nervous system. Multimodal MRI is widely used for the preliminary screening of gliomas and plays a crucial role in auxiliary diagnosis, therapeutic efficacy, and prognostic evaluation. Currently, the computer-aided diagnostic studies of gliomas using MRI have focused on independent analysis events such as tumor segmentation, grading, and radiogenomic classification, without studying inter-dependencies among these events. In this study, we propose a Glioma Multimodal MRI Analysis System (GMMAS) that utilizes a deep learning network for processing multiple events simultaneously, leveraging their inter-dependencies through an uncertainty-based multi-task learning architecture and synchronously outputting tumor region segmentation, glioma histological subtype, IDH mutation genotype, and 1p/19q chromosome disorder status. Compared with the reported single-task analysis models, GMMAS improves the precision across tumor layered diagnostic tasks. Additionally, we have employed a two-stage semi-supervised learning method, enhancing model performance by fully exploiting both labeled and unlabeled MRI samples. Further, by utilizing an adaptation module based on knowledge self-distillation and contrastive learning for cross-modal feature extraction, GMMAS exhibited robustness in situations of modality absence and revealed the differing significance of each MRI modal. Finally, based on the analysis outputs of the GMMAS, we created a visual and user-friendly platform for doctors and patients, introducing GMMAS-GPT to generate personalized prognosis evaluations and suggestions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。