arrow
Return

LSTM algorithm optimization for COVID-19 prediction model

delete2024-02-01
delete5
delete
OA
AI
I
Irwan Sembiring
S
Sri Wahyuni *
E
Eko Sediyono
DOI:10.1016/j.heliyon.2024.e26158delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
The development of predictive models for infectious diseases, specifically COVID-19, is an important step in early control efforts to reduce the mortality rate. However, traditional time series prediction models used to analyze the disease spread trends often encounter challenges related to accuracy, necessitating the need to develop prediction models with enhanced accuracy. Therefore, this research aimed to develop a prediction model based on the Long Short-Term Memory (LSTM) networks to better predict the number of confirmed COVID-19 cases. The proposed optimized LSTM (popLSTM) model was compared with Basic LSTM and improved MinMaxScaler developed earlier using COVID-19 dataset taken from previous research. The dataset was collected from four countries with a high daily increase in confirmed cases, including Hong Kong, South Korea, Italy, and Indonesia. The results showed significantly improved accuracy in the optimized model compared to the previous research methods. The contributions of popLSTM included 1) Incorporating the output results on the output gate to effectively filter more detailed information compared to the previous model, and 2) Reducing the error value by considering the hidden state on the output gate to improve accuracy. popLSTM in this experiment exhibited a significant 4% increase in accuracy.
Keywords:
COVID-19
Time series prediction
LSTM model
Optimization
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Heliyon cover
Heliyon
IF:
3.6
Papers:
3.8W
Citations:
10.5W

Organization

U
universitas kristen satya wacana
Scholars:
152
Papers: 112
Citations: 0