用患者相似图和图神经网络预测心衰,效果优于传统方法。
Graph Neural Networks for Heart Failure Prediction on an EHR-Based Patient Similarity Graph
- 构建患者相似图,用GNN捕捉医疗记录中的复杂关系
- Graph Transformer模型在心衰预测上F1达0.5361,表现最佳
- 通过注意力机制分析,可解释模型如何做出临床决策
目标:在现代医疗中,准确预测疾病至关重要。本研究提出一种新方法,利用图神经网络(GNN)和图Transformer(GT)在基于电子健康记录(EHR)的患者相似图上预测下一次住院时的心力衰竭(HF)发生。方法:使用MIMIC-III数据集的EHR,通过诊断、操作和药物嵌入计算患者间距离,并采用K近邻(KNN)算法构建患者相似图。比较了GraphSAGE、图注意力网络(GAT)和图Transformer(GT)三种模型的性能,评估指标包括F1分数、AUROC和AUPRC,同时进行可解释性分析。结果:GT模型表现最优(F1:0.5361,AUROC:0.7925,AUPRC:0.5168)。尽管随机森林(RF)基线在AUPRC上接近,但GT因利用图结构中的患者关系,具备更强可解释性。联合分析注意力权重、图连通性与临床特征,揭示了不同分类群体下的预测逻辑。结论:基于图的方法如GNN能有效建模复杂医疗数据,提升心衰预测准确率与临床可解释性。
原文摘要 · Abstract (English)
Objective: In modern healthcare, accurately predicting diseases is a crucial matter. This study introduces a novel approach using graph neural networks (GNNs) and a Graph Transformer (GT) to predict the incidence of heart failure (HF) on a patient similarity graph at the next hospital visit. Materials and Methods: We used electronic health records (EHR) from the MIMIC-III dataset and applied the K-Nearest Neighbors (KNN) algorithm to create a patient similarity graph using embeddings from diagnoses, procedures, and medications. Three models - GraphSAGE, Graph Attention Network (GAT), and Graph Transformer (GT) - were implemented to predict HF incidence. Model performance was evaluated using F1 score, AUROC, and AUPRC metrics, and results were compared against baseline algorithms. An interpretability analysis was performed to understand the model's decision-making process. Results: The GT model demonstrated the best performance (F1 score: 0.5361, AUROC: 0.7925, AUPRC: 0.5168). Although the Random Forest (RF) baseline achieved a similar AUPRC value, the GT model offered enhanced interpretability due to the use of patient relationships in the graph structure. A joint analysis of attention weights, graph connectivity, and clinical features provided insight into model predictions across different classification groups. Discussion and Conclusion: Graph-based approaches such as GNNs provide an effective framework for predicting HF. By leveraging a patient similarity graph, GNNs can capture complex relationships in EHR data, potentially improving prediction accuracy and clinical interpretability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。