arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2411.00609eess.IVcs.CVcs.LG

肿瘤位置加权的MRI-报告对比学习:一种提升儿童脑肿瘤诊断可解释性的框架

Tumor Location-weighted MRI-Report Contrastive Learning: A Framework for Improving the Explainability of Pediatric Brain Tumor Diagnosis

The Hospital for Sick Children(病童医院);University of Toronto(多伦多大学);Vector Institute for Artificial Intelligence(矢量人工智能研究所);University Hospital Augsburg(奥格斯堡大学医院)

Sara Ketabi, Matthias W. Wagner, Cynthia Hawkins, Uri Tabori, Birgit Betina Ertl-Wagner, Farzad Khalvati

AI总结:

针对卷积神经网络在脑肿瘤MRI诊断中可解释性不足的问题,提出整合肿瘤位置的MRI-放射报告对比学习框架,提升了儿童低级别胶质瘤基因标记分类的性能与可解释性。

摘要:

尽管卷积神经网络(CNNs)在基于磁共振成像(MRI)的脑肿瘤诊断中表现出可观性能,但它们在临床工作流中的整合仍受到限制。这主要是因为放射科医生不清楚模型预测所依据的特征,因此这些特征不具备临床相关性,即缺乏可解释性。作为放射科医生知识与专业技能的宝贵来源,放射报告可与MRI整合到对比学习(CL)框架中,通过学习图像-报告关联来提升CNN的可解释性。在本研究中,我们在3D脑部MRI扫描和放射报告上训练多模态对比学习架构,以学习具有信息量的MRI表征。此外,我们将对多项脑肿瘤分析任务至关重要的肿瘤位置整合到该框架中,以提升其泛化能力。随后,我们将学习到的图像表征用于改善下游任务——儿童最常见脑肿瘤儿童低级别胶质瘤的基因标记分类的可解释性与性能。我们的结果显示,模型注意力图与手动肿瘤分割(作为可解释性度量)之间的Dice系数为31.1%,测试集分类性能达87.7%,显著优于基线模型。这些提升可建立放射科医生对我们模型的信任,推动其整合到临床实践中以实现更高效的肿瘤诊断。

英文摘要:

Despite the promising performance of convolutional neural networks (CNNs) in brain tumor diagnosis from magnetic resonance imaging (MRI), their integration into the clinical workflow has been limited. That is mainly due to the fact that the features contributing to a model's prediction are unclear to radiologists and hence, clinically irrelevant, i.e., lack of explainability. As the invaluable sources of radiologists' knowledge and expertise, radiology reports can be integrated with MRI in a contrastive learning (CL) framework, enabling learning from image-report associations, to improve CNN explainability. In this work, we train a multimodal CL architecture on 3D brain MRI scans and radiology reports to learn informative MRI representations. Furthermore, we integrate tumor location, salient to several brain tumor analysis tasks, into this framework to improve its generalizability. We then apply the learnt image representations to improve explainability and performance of genetic marker classification of pediatric Low-grade Glioma, the most prevalent brain tumor in children, as a downstream task. Our results indicate a Dice score of 31.1% between the model's attention maps and manual tumor segmentation (as an explainability measure) with test classification performance of 87.7%, significantly outperforming the baselines. These enhancements can build trust in our model among radiologists, facilitating its integration into clinical practices for more efficient tumor diagnosis.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑