Development and validation of machine learning-driven prediction model for serious bacterial infection among febrile

Bongjin Lee1, Hyun Jung Chung2, Hyun Mi Kang3

  • 1Department of Pediatrics, Seoul National University Hospital, Seoul, Korea.

Plos One
|March 25, 2022
PubMed

Insights

A new machine learning model accurately predicts serious bacterial infections (SBI) in febrile children, even with missing data. This AI tool offers superior performance over traditional methods for diagnosing critical pediatric conditions.

Area of Science:

  • Pediatric Infectious Diseases
  • Machine Learning in Healthcare
  • Clinical Decision Support Systems

Background:

  • Serious bacterial infections (SBI) in children, including meningitis and sepsis, are life-threatening conditions requiring accurate and timely diagnosis.
  • Existing SBI prediction tools often fail when crucial data points are missing, limiting their clinical utility.
  • The need for robust diagnostic aids that can function effectively despite incomplete patient information is critical in emergency settings.

Purpose of the Study:

  • To develop and validate a machine learning (ML)-driven model for predicting SBI in febrile children.
  • To create a model capable of robust prediction even when patient data values are missing.
  • To compare the performance of the ML model against conventional methods in both internal and external validation cohorts.

Main Methods:

  • A multicenter retrospective observational study involving febrile children under 6 years old across three tertiary hospitals (2016-2018).
  • A derivation cohort (two hospitals) was used to train a Random Forest (RF) ML model, while a third hospital's data served as an external validation cohort.
  • Performance was evaluated using Area Under the Receiver Operating Characteristic Curve (AUROC) and Area Under the Precision-Recall Curve (AUPRC), comparing RF to a Conventional Logistic Regression (CLR) model.

Main Results:

  • In the derivation cohort, the RF model achieved an AUROC of 0.964 and AUPRC of 0.753, significantly outperforming the CLR model (AUROC 0.902, AUPRC 0.573).
  • External validation in the third hospital confirmed the RF model's superior performance with an AUROC of 0.950 and AUPRC of 0.605, compared to CLR (AUROC 0.815, AUPRC 0.586).
  • The ML-driven RF model demonstrated robust performance and effectiveness despite the presence of missing data points in patient records.

Conclusions:

  • A novel machine learning-driven prediction model for SBI in febrile children has been successfully developed and validated.
  • The developed model exhibits superior predictive performance compared to conventional methods, particularly in handling missing data.
  • This AI-powered tool offers a promising advancement for improving the accuracy and reliability of SBI diagnosis in pediatric emergency departments.