一种基于异构图神经网络的ICU心衰预测系统

摘要

本发明公开了一种基于异构图神经网络的ICU心衰预测系统，包括计算机存储器和计算机处理器，计算机存储器中存有心衰预测模型；心衰预测模型包括两层时间感知门控图注意力神经网络、一个环境Transformer层以及两层全连接层；计算机处理器执行计算机程序时实现以下步骤：提取心衰相关病人在住院第一次进入ICU后前48小时的实验检测、输液、处方药数据，构成EHR数据；通过提取得到的EHR数据构建时间感知EHR异构图，输入训练好的心衰预测模型得到预测的输出概率，该输出概率的结果用于医生评估当前的治疗方案。利用本发明，可以解决实际应用中通过心衰相关病人第一次进入ICU前48小时数据预测本次住院期间是否会死亡不够准确的问题，从而指导医生的后续治疗方案。