基于2015-2019年江苏省死亡数据双向长短期记忆网络–时空图卷积网络的日死亡数预测模型的建立与评估

Development and evaluation of a daily mortality prediction model based on bi-directional long short-term memory and spatiotemporal graph convolutional network using Jiangsu mortality data from 2015 to 2019

  • 摘要:
    目的 基于死因监测数据开展人群死亡监测预警具有重要的公共卫生意义。 本研究旨在基于死因监测数据探索适用于预测区(县)级日死亡数的时空建模方法,以期为建立人群死亡时空预警模型提供方法参考,为死因监测系统实现主动预警功能提供新的思路。
    方法 采用全国死因监测系统报告的2015-2019年江苏省各区(县)每日全因死亡人数数据,在时空图卷积网络(STGCN)的基础上融合双向长短期记忆网络(BiLSTM),构建BiLSTM-STGCN模型用于预测各区(县)每日死亡数,模型共纳入了92个区(县)的每日全因死亡数据,通过计算均方根误差(RMSE)、平均绝对误差(MAE)以及对称平均绝对百分比误差(SMAPE)评价模型预测性能,同时以贝叶斯分层时空模型(BHSTM)和STGCN作为基准模型与该模型进行比较,评价模型在死因监测数据上的适用性。
    结果 BiLSTM-STGCN模型预测2019年7月1日至2019年12月31日江苏省各区(县)每日全因死亡数的RMSE值为4.81,MAE值为3.64,SMAPE值为27.42%,预测性能良好且优于BHSTM和STGCN。
    结论 BiLSTM-STGCN模型具有良好的预测性能,相较于经典的BHSTM和单纯的STGCN模型,更适配基于死因监测数据的区(县)级日死亡数的预测任务。

     

    Abstract:
    Objective Mortality surveillance and early warning in population based on death cause surveillance data has great public health significance. This study explores spatiotemporal modeling for predicting daily all-cause deaths at the county level using death cause surveillance data, aiming to provide methodological support for early warning models and new insights for active surveillance systems.
    Methods This study used daily, county-level and all-cause deaths data in Jiangsu province during 2015-2019, which were obtained from the National Mortality Surveillance System. Based on this dataset, a hybrid deep learning model that integrates a Bidirectional Long Short-Term Memory network (BiLSTM) with a Spatio-Temporal Graph Convolutional Network (STGCN) was developed to predict daily all-cause deaths. Data from 92 counties were incorporated into the model. Meanwhile, this model was compared with a Bayesian hierarchical spatiotemporal model and a STGCN model, with performance assessed via root mean squared error (RMSE), mean absolute error (MAE), and symmetric mean absolute percentage error (SMAPE). The models were developed by using R 4.5.1 and Python 3.10.
    Results The BiLSTM-STGCN model demonstrated a RMSE value of 4.81, an MAE value of 3.64, and an SMAPE value of 27.42% in prediction of daily all-cause deaths across all counties in Jiangsu from July 1, 2019, to December 31, 2019. The model exhibited good prediction performance and outperformed the Bayesian hierarchical spatiotemporal model and STGCN model.
    Conclusion The BiLSTM-STGCN model exhibits strong prediction performance and, compared with the classical Bayesian hierarchical spatiotemporal model and single STGCN model, is more suitable for the prediction daily deaths at the county level based on death cause surveillance data.

     

/

返回文章
返回