Abstract:Background and objective Tuberculosis (Tuberculosis, TB) is a public health problem in China, which not only endangers the population’s health but also affects economic and social development. It requires an accurate prediction analysis to help to make policymakers with early warning and provide effective precautionary measures. In this study, ARIMA, GM(1,1), and LSTM models were constructed and compared, respectively. The results showed that the LSTM was the optimal model, which can be achieved satisfactory performance for TB cases predictions in mainland China. Methods The data of tuberculosis cases in mainland China were extracted from the National Health Commission of the People’s Republic of China website. According to the TB data characteristics and the sample requirements, we created the ARIMA, GM(1,1), and LSTM models, which can make predictions for the prevalence trend of TB. The mean absolute error (MAE), root mean square error (RMSE), and mean absolute percentage error (MAPE) were applied to evaluate the effects of model fitting predicting accuracy. Results There were 3,021,995 tuberculosis cases in mainland China from January 2018 to December 2020. And the overall TB cases in mainland China take on a downtrend trend. We established ARIMA, GM(1,1), and LSTM models, respectively. The optimal ARIMA model is the ARIMA (0,1,0) × (0,1,0)12. The equation for GM(1,1) model was X(k+1) = -10057053.55e (-0.01k) + 10153178.55 the Mean square deviation ratio C value was 0.49, and the Small probability of error P was 0.94. LSTM model consists of an input layer, a hidden layer and an output layer, the parameters of epochs, learning rating are 60, 0.01, respectively. The MAE, RMSE, and MAPE values of LSTM model were smaller than that of GM(1,1) and ARIMA models. Conclusions Our findings showed that the LSTM model was the optimal model, which has a higher accuracy performance than that of ARIMA and GM (1,1) models. Its prediction results can act as a predictive tool for TB prevention measures in mainland China.

Prediction of outpatients with conjunctivitis in Xinjiang based on LSTM and GRU models

Research on Prediction of Daily Admissions of Respiratory Diseases with Comorbid Diabetes in Beijing Based on Long Short-Term Memory Recurrent Neural Network

A Prediction Model Based on Machine Learning for Diagnosing the Early COVID-19 Patients

Short-term prediction for dynamic blood glucose trends based on ARIMA-LSSVM-GRU model

The Prediction of Influenza-like Illness and Respiratory Disease Using LSTM and ARIMA

The research of ARIMA, GM(1,1), and LSTM models for prediction of TB cases in China

Using Gated Recurrent Units Models for Early Prediction of Sepsis in The Intensive Care Unit

Establishment and Application of the Number of Outpatient and Emergence Visits and Discharged Patients Prediction Linear Regression Model

Hospital daily outpatient visits forecasting using a combinatorial model based on ARIMA and SES models

Forecasting rheumatoid arthritis patient arrivals by including meteorological factors and air pollutants

Application of a long short-term memory neural network: a burgeoning method of deep learning in forecasting HIV incidence in Guangxi, China

The comparative analysis of SARIMA, Facebook Prophet, and LSTM for road traffic injury prediction in Northeast China

[LSTM-XGBoost Based RR Intervals Time Series Prediction Method in Hypertensive Patients]

Machine learning approaches for practical predicting outpatient near-future AECOPD based on nationwide electronic medical records

A Power Load Prediction by LSTM Model Based on the Double Attention Mechanism for Hospital Building

Time series models in prediction of severe fever with thrombocytopenia syndrome cases in Shandong province, China

[Visual field prediction based on temporal-spatial feature learning]

A novel multivariable time series prediction model for acute kidney injury in general hospitalization

Comparison of autoregressive integrated moving average model and generalised regression neural network model for prediction of haemorrhagic fever with renal syndrome in China: a time-series study

Predictive study of tuberculosis incidence by time series method and Elman neural network in Kashgar, China