Evaluating the effectiveness of self-attention mechanism in tuberculosis time series forecasting

被引:0
作者
Lv, Zhihong [1 ]
Sun, Rui [1 ]
Liu, Xin [1 ]
Wang, Shuo [2 ]
Guo, Xiaowei [1 ]
Lv, Yuan [1 ]
Yao, Min [3 ]
Zhou, Junhua [1 ]
机构
[1] Hunan Normal Univ, Sch Med, Key Lab Mol Epidemiol Hunan Prov, Changsha 410013, Hunan, Peoples R China
[2] Changsha Univ Sci & Technol, Changsha 410114, Hunan, Peoples R China
[3] Hunan Prov Ctr Dis Control & Prevent, Changsha 410005, Hunan, Peoples R China
基金
中国国家自然科学基金;
关键词
Tuberculosis; Time series forecasting; Self-attention mechanism; ARIMA model; LSTM model;
D O I
10.1186/s12879-024-10183-9
中图分类号
R51 [传染病];
学科分类号
100401 ;
摘要
BackgroundWith the increasing impact of tuberculosis on public health, accurately predicting future tuberculosis cases is crucial for optimizing of health resources and medical service allocation. This study applies a self-attention mechanism to predict the number of tuberculosis cases, aiming to evaluate its effectiveness in forecasting.MethodsMonthly tuberculosis case data from Changde City between 2010 and 2021 were used to construct a self-attention model, a long short-term memory (LSTM) model, and an autoregressive integrated moving average (ARIMA) model. The performance of these models was evaluated using three metrics: root mean square error (RMSE), mean absolute error (MAE), and mean absolute percentage error (MAPE).ResultsThe self-attention model outperformed the other models in terms of prediction accuracy. On the test set, the RMSE of the self-attention model was approximately 7.41% lower than that of the LSTM model, MAE was reduced by about 10.99%, and MAPE was reduced by approximately 9.87%. Compared to the ARIMA model, RMSE was reduced by about 28.86%, MAE by about 32.22%, and MAPE by approximately 29.89%.ConclusionThe self-attention model can effectively improve the prediction accuracy of tuberculosis cases, providing guidance for health departments optimizing of health resources and medical service allocation.
引用
收藏
页数:13
相关论文
共 41 条
[1]   On the Implementation of the Artificial Neural Network Approach for Forecasting Different Healthcare Events [J].
Alshanbari, Huda M. M. ;
Iftikhar, Hasnain ;
Khan, Faridoon ;
Rind, Moeeba ;
Ahmad, Zubair ;
El-Bagoury, Abd Al-Aziz Hosni .
DIAGNOSTICS, 2023, 13 (07)
[2]  
Bai W., 2024, Global, regional and national trends in tuberculosis incidence and main risk factors: a study using data from 2000 to 2021, DOI [10.1186/s12889-023-17495-6, DOI 10.1186/S12889-023-17495-6]
[3]  
Bergstra J, 2012, J MACH LEARN RES, V13, P281
[4]  
Bhanja Samit, 2018, arXiv, DOI 10.48550/arXiv.1812.05519
[5]  
Box G E., 2015, Time series analysis: forecasting and control, V5
[6]  
Cleveland RB, 1990, J Official Stat, V6
[7]   Modeling and Analysis of Monkeypox Outbreak Using a New Time Series Ensemble Technique [J].
Cuba, Wilfredo Meza ;
Alfaro, Juan Carlos Huaman ;
Iftikhar, Hasnain ;
Lopez-Gonzales, Javier Linkolk .
AXIOMS, 2024, 13 (08)
[8]  
Das S., 2023, Neuromethods, P117, DOI [DOI 10.1007/978-1-0716-3195-94, 10.1007/978-1-0716-3195-9_4, DOI 10.1007/978-1-0716-3195-9_4]
[9]   Tuberculosis [J].
Furin, Jennifer ;
Cox, Helen ;
Pai, Madhukar .
LANCET, 2019, 393 (10181) :1642-1656
[10]   Epidemiological characteristics of pulmonary tuberculosis in mainland China from 2004 to 2015: a model-based analysis [J].
Guo, Zuiyuan ;
Xiao, Dan ;
Wang, Xiuhong ;
Wang, Yayu ;
Yan, Tiecheng .
BMC PUBLIC HEALTH, 2019, 19 (1)