Journal of Machine and Computing


SIRD-ABiGRU-AE: A Modified Compartmental Model with Attention-Driven BiGRU Autoencoder for COVID-19 Outbreak Prediction and Hospitalizations



Journal of Machine and Computing

Received On : 25 April 2024

Revised On : 18 September 2024

Accepted On : 20 January 2025

Published On : 05 April 2025

Volume 05, Issue 02

Pages : 694-708


Abstract


Several epidemiological studies have been undertaken using a compartmental model to predict disease spread effectively. However, knowledge about the epidemiological cycle lacks existing techniques and fails to promote the vaccines and medications that the government issues to overcome the pandemic disease. Many researchers implemented a Susceptible-Infected-Recovered-Deceased (SIRD) based compartmental approach to determine the methods emphasized by the government to eradicate the spread of COVID-19. The traditional SIRD-based compartmental model produces high prediction errors and is time-consuming. Hence, this article presents a novel Deep Learning (DL) based Attention-driven bi-directional gated recurrent unit Autoencoder (A-Bi-GRU-AE) model, which is hybridized with the SIRD model to enhance the system performance. The proposed approach is implemented in the PYTHON platform, and the publicly available covid19Italy dataset is utilized for the experimental process. The proposed method obtains the overall predicted R2 of 0.97 and time complexity of 2634.01ms.


Keywords


COVID-19, Italy, Attention Mechanism, Bi-Directional Gated Recurrent Unit, Autoencoder, Hospitalizations, Compartmental Models.


  1. S. Abrams et al., “Modelling the early phase of the Belgian COVID-19 epidemic using a stochastic compartmental model and studying its implied future trajectories,” Epidemics, vol. 35, p. 100449, Jun. 2021, doi: 10.1016/j.epidem.2021.100449.
  2. L. Yang et al., “COVID-19: immunopathogenesis and Immunotherapeutics,” Signal Transduction and Targeted Therapy, vol. 5, no. 1, Jul. 2020, doi: 10.1038/s41392-020-00243-2.
  3. N. Kandel, S. Chungong, A. Omaar, and J. Xing, “Health security capacities in the context of COVID-19 outbreak: an analysis of International Health Regulations annual report data from 182 countries,” The Lancet, vol. 395, no. 10229, pp. 1047–1053, Mar. 2020, doi: 10.1016/s0140-6736(20)30553-5.
  4. L. Di Domenico, G. Pullano, C. E. Sabbatini, P.-Y. Boëlle, and V. Colizza, “Impact of lockdown on COVID-19 epidemic in Île-de-France and possible exit strategies,” BMC Medicine, vol. 18, no. 1, Jul. 2020, doi: 10.1186/s12916-020-01698-4.
  5. G. H.-H. Jen, S.-Y. Chen, W.-J. Chang, C.-N. Chen, A. M.-F. Yen, and R.-E. Chang, “Evaluating medical capacity for hospitalization and intensive care unit of COVID-19: A queue model approach,” Journal of the Formosan Medical Association, vol. 120, pp. S86–S94, Jun. 2021, doi: 10.1016/j.jfma.2021.05.002.
  6. A. Bousquet, W. H. Conrad, S. O. Sadat, N. Vardanyan, and Y. Hong, “Deep learning forecasting using time-varying parameters of the SIRD model for Covid-19,” Scientific Reports, vol. 12, no. 1, Feb. 2022, doi: 10.1038/s41598-022-06992-0.
  7. J. Bollon et al., “Predicted Effects of Stopping COVID-19 Lockdown on Italian Hospital Demand,” Disaster Medicine and Public Health Preparedness, vol. 14, no. 5, pp. 638–642, May 2020, doi: 10.1017/dmp.2020.157.
  8. R. N. Thompson, “Epidemiological models are important tools for guiding COVID-19 interventions,” BMC Medicine, vol. 18, no. 1, May 2020, doi: 10.1186/s12916-020-01628-4.
  9. X. Liu, X. Zheng, and B. Balachandran, “COVID-19: data-driven dynamics, statistical and distributed delay models, and observations,” Nonlinear Dynamics, vol. 101, no. 3, pp. 1527–1543, Aug. 2020, doi: 10.1007/s11071-020-05863-5.
  10. S. S. Singh and D. K. Mohapatra, “Predictive Analysis for COVID-19 Spread in India by Adaptive Compartmental Model,” Jul. 2020, doi: 10.1101/2020.07.08.20148619.
  11. H. Sjödin et al., “COVID-19 healthcare demand and mortality in Sweden in response to non-pharmaceutical mitigation and suppression scenarios,” International Journal of Epidemiology, vol. 49, no. 5, pp. 1443–1453, Sep. 2020, doi: 10.1093/ije/dyaa121.
  12. Gatto, Andrea, Gabriele Accarino, Valeria Aloisi, Francesco Immorlano, Francesco Donato, and Giovanni Aloisio. "Limits of Compartmental Models and New Opportunities for Machine Learning: A Case Study to Forecast the Second Wave of COVID-19 Hospitalizations in Lombardy, Italy." In Informatics, vol. 8, no. 3, p. 57. MDPI, 2021.
  13. J. Tolles and T. Luong, “Modeling Epidemics with Compartmental Models,” JAMA, vol. 323, no. 24, p. 2515, Jun. 2020, doi: 10.1001/jama.2020.8420.
  14. G. E. Weissman et al., “Locally Informed Simulation to Predict Hospital Capacity Needs During the COVID-19 Pandemic,” Annals of Internal Medicine, vol. 173, no. 1, pp. 21–28, Jul. 2020, doi: 10.7326/m20-1260.
  15. D. Rafiq, A. Batool, and M. A. Bazaz, “Three months of COVID‐19: A systematic review and meta‐analysis,” Reviews in Medical Virology, vol. 30, no. 4, May 2020, doi: 10.1002/rmv.2113.
  16. A. Teslya, T. M. Pham, N. G. Godijk, M. E. Kretzschmar, M. C. J. Bootsma, and G. Rozhnova, “Impact of self-imposed prevention measures and short-term government-imposed social distancing on mitigating and delaying a COVID-19 epidemic: A modelling study,” PLOS Medicine, vol. 17, no. 7, p. e1003166, Jul. 2020, doi: 10.1371/journal.pmed.1003166.
  17. A. Leontitsis, A. Senok, A. Alsheikh-Ali, Y. Al Nasser, T. Loney, and A. Alshamsi, “SEAHIR: A Specialized Compartmental Model for COVID-19,” International Journal of Environmental Research and Public Health, vol. 18, no. 5, p. 2667, Mar. 2021, doi: 10.3390/ijerph18052667.
  18. M. J. Keeling et al., “Predictions of COVID-19 dynamics in the UK: Short-term forecasting and analysis of potential exit strategies,” PLOS Computational Biology, vol. 17, no. 1, p. e1008619, Jan. 2021, doi: 10.1371/journal.pcbi.1008619.
  19. S. B. Ramezani, A. Amirlatifi, and S. Rahimi, “A novel compartmental model to capture the nonlinear trend of COVID-19,” Computers in Biology and Medicine, vol. 134, p. 104421, Jul. 2021, doi: 10.1016/j.compbiomed.2021.104421.
  20. M. Pájaro, N. M. Fajar, A. A. Alonso, and I. Otero-Muras, “Stochastic SIR model predicts the evolution of COVID-19 epidemics from public health and wastewater data in small and medium-sized municipalities: A one year study,” Chaos, Solitons & Fractals, vol. 164, p. 112671, Nov. 2022, doi: 10.1016/j.chaos.2022.112671.
  21. K. S. Sharov, “Creating and applying SIR modified compartmental model for calculation of COVID-19 lockdown efficiency,” Chaos, Solitons & Fractals, vol. 141, p. 110295, Dec. 2020, doi: 10.1016/j.chaos.2020.110295.
  22. G. Rainisch, E. A. Undurraga, and G. Chowell, “A dynamic modeling tool for estimating healthcare demand from the COVID19 epidemic and evaluating population-wide interventions,” International Journal of Infectious Diseases, vol. 96, pp. 376–383, Jul. 2020, doi: 10.1016/j.ijid.2020.05.043.
  23. L. Wang et al., “SEIARN: Intelligent Early Warning Model of Epidemic Spread Based on LSTM Trajectory Prediction,” Mathematics, vol. 10, no. 17, p. 3046, Aug. 2022, doi: 10.3390/math10173046.
  24. R. Delli Compagni, Z. Cheng, S. Russo, and T. P. Van Boeckel, “A hybrid Neural Network-SEIR model for forecasting intensive care occupancy in Switzerland during COVID-19 epidemics,” PLOS ONE, vol. 17, no. 3, p. e0263789, Mar. 2022, doi: 10.1371/journal.pone.0263789.
  25. https://github.com/RamiKrispin/covid19Italy.

CRediT Author Statement


The authors confirm contribution to the paper as follows:

Conceptualization: Parthiban M, Anna Alphy and Sreedevi B; Methodology: Parthiban M; Software: Parthiban M, Anna Alphy and Sreedevi B; Data Curation: Anna Alphy and Sreedevi B; Writing- Original Draft Preparation: Parthiban M, Anna Alphy and Sreedevi B; Visualization: Parthiban M, Anna Alphy and Sreedevi B; Investigation: Parthiban M; Supervision: Anna Alphy and Sreedevi B; Validation: Anna Alphy and Sreedevi B; Writing- Reviewing and Editing: Parthiban M, Anna Alphy and Sreedevi B; All authors reviewed the results and approved the final version of the manuscript.


Acknowledgements


Authors thank Reviewers for taking the time and effort necessary to review the manuscript.


Funding


No funding was received to assist with the preparation of this manuscript.


Ethics declarations


Conflict of interest

The authors have no conflicts of interest to declare that are relevant to the content of this article.


Availability of data and materials


Data sharing is not applicable to this article as no new data were created or analysed in this study.


Author information


Contributions

All authors have equal contribution in the paper and all authors have read and agreed to the published version of the manuscript.


Corresponding author


Rights and permissions


Open Access This article is licensed under a Creative Commons Attribution NoDerivs is a more restrictive license. It allows you to redistribute the material commercially or non-commercially but the user cannot make any changes whatsoever to the original, i.e. no derivatives of the original work. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-nd/4.0/


Cite this article


Parthiban M, Anna Alphy and Sreedevi B, “SIRD-ABiGRU-AE: A Modified Compartmental Model with Attention-Driven BiGRU Autoencoder for COVID-19 Outbreak Prediction and Hospitalizations”, Journal of Machine and Computing, pp. 694-708, April 2025, doi: 10.53759/7669/jmc202505055.


Copyright


© 2025 Parthiban M, Anna Alphy and Sreedevi B. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.