LLM2D
自解释超图神经网络在诊断预测中的应用
Self-Explaining Hypergraph Neural Networks for Diagnosis Prediction
作者: Leisheng Yu, Yanxiao Cai, Minxing Zhang, Xia Hu
发布日期: 5/2/2025
arXiv ID: oai:arXiv.org:2502.10689v2

摘要

arXiv:2502.10689v2 通知类型: 替换-交叉 摘要:电子健康记录(EHRs)的急剧增长使深度学习模型在预测性医疗保健方面表现出色。然而,在诊断预测等高风险应用中,模型的可解释性仍然至关重要。现有的具有内在可解释性的深度学习诊断预测模型通常为每次过去的诊断或医院访问分配注意权重,提供的解释缺乏灵活性和简洁性。在本文中,我们介绍了SHy,一种自我解释的超图神经网络模型,旨在提供个性化、简洁且忠实的解释,这些解释允许临床专家进行干预。通过将每位患者建模为一个独特的超图,并采用消息传递机制,SHy捕捉了更高级别的疾病相互作用,并提取了作为个性化解释的独特时间表型。此外,通过考虑原始诊断记录中的重要遗漏,SHy解决了EHR数据的不完整性问题。在两个真实世界EHR数据集上的定性案例研究和广泛的定量评估表明,SHy的预测性能和解释性优于现有最先进的模型。