摘要 使用人工智能从 MRI 图像中检测和描绘脑肿瘤是医学 AI 面临的一项复杂挑战。最近的进展见证了各种技术被用于协助医疗专业人员完成这项任务。尽管机器学习算法在分割肿瘤方面很有效,但它们在决策过程中缺乏透明度,阻碍了信任和验证。在我们的项目中,我们构建了一个可解释的 U-Net 模型,专门用于脑肿瘤分割,利用梯度加权类激活映射 (Grad-CAM) 算法和 SHapley 加法解释 (SHAP) 库。我们依靠 BraTS2020 基准数据集进行训练和评估。我们采用的 U-Net 模型产生了有希望的结果。然后,我们利用 Grad-CAM 在图像中可视化模型关注的关键特征。此外,我们利用 SHAP 库来阐明用于预测患者生存天数的各种模型(包括随机森林、KNN、SVC 和 MLP)的预测,从而增强了可解释性。
主要关键词