Research Article
Automated Pipeline for Continual Data Gathering and Retraining of the Machine Learning-Based COVID-19 Spread Models
@ARTICLE{10.4108/eai.4-5-2021.169582, author={S. Baressi Šegota and I. Lorencin and N. Anđelić and D. Štifanić and J. Musulin and S. Vlahinić and T. Šušteršič and A. Blagojević and Z. Car}, title={Automated Pipeline for Continual Data Gathering and Retraining of the Machine Learning-Based COVID-19 Spread Models}, journal={EAI Endorsed Transactions on Bioengineering and Bioinformatics}, volume={1}, number={3}, publisher={EAI}, journal_a={BEBI}, year={2021}, month={5}, keywords={Artificial Intelligence, Bio-engineering, Bio-inspired systems, Bio-inspired models, COVID-19, Epidemiology Curves, Machine Learning, Multilayer Perceptron}, doi={10.4108/eai.4-5-2021.169582} }
- S. Baressi Šegota
I. Lorencin
N. Anđelić
D. Štifanić
J. Musulin
S. Vlahinić
T. Šušteršič
A. Blagojević
Z. Car
Year: 2021
Automated Pipeline for Continual Data Gathering and Retraining of the Machine Learning-Based COVID-19 Spread Models
BEBI
EAI
DOI: 10.4108/eai.4-5-2021.169582
Abstract
INTRODUCTION: The development of epidemiological curve models is one of the key factors in the combat of epidemiological diseases such as COVID-19.
OBJECTIVES: The goal of this paper is to develop a system for automatic training and testing of AI-based regressive models of epidemiological curves using public data, which involves automating the data acquisition and speeding up the training of the models.
METHODS: The research applies Multilayer Perceptron (MLP) for the creation of models, implemented within a system for automatic data fetching and training, and e valuated using the coefficient of determination (R2). Training time is lowered through the application of data filtering and simplifying the model selection.
RESULTS: The developed system can train high precision models rapidly, allowing for quick model delivery All trained models achieve scores which are higher than 0.95.
CONCLUSION: The results show that the development of a quick COVID-19 spread modeling system is possible.
Copyright © 2021 S. Baressi Šegota et al., licensed to EAI. This is an open access article distributed under the terms of the Creative Commons Attribution license, which permits unlimited use, distribution and reproduction in any medium so long as the original work is properly cited