A Unified Hierarchical XGBoost model for classifying priorities for COVID-19 vaccination campaign

PATTERN RECOGNITION

卷:121

Article

Romeo, Luca^1,2  Frontoni, Emanuele¹ 

[1] Univ Politecn Marche, Dept Informat Engn DII, Ancona, Italy

[2] Ist Italiano Tecnol, Computat Stat & Machine Learning, Genoa, Italy

关键词: COVID-19;    Vaccination;    Machine learning;    XGBoost;    Clinical decision support system;    Model interpretability;   

DOI  :  10.1016/j.patcog.2021.108197

来源: Elsevier

【 摘 要 】

The current ML approaches do not fully focus to answer a still unresolved and topical challenge, namely the prediction of priorities of COVID-19 vaccine administration. Thus, our task includes some additional methodological challenges mainly related to avoiding unwanted bias while handling categorical and ordinal data with a highly imbalanced nature. Hence, the main contribution of this study is to propose a machine learning algorithm, namely Hierarchical Priority Classification eXtreme Gradient Boosting for priority classification for COVID-19 vaccine administration using the Italian Federation of General Practitioners dataset that contains Electronic Health Record data of 17k patients. We measured the effectiveness of the proposed methodology for classifying all the priority classes while demonstrating a significant improvement with respect to the state of the art. The proposed ML approach, which is integrated into a clinical decision support system, is currently supporting General Pracitioners in assigning COVID-19 vaccine administration priorities to their assistants. (c) 2021 Elsevier Ltd. All rights reserved.

【 授权许可】

Free