Developing and validating clinical features-based machine learning algorithms to predict influenza infection in influenza-like illness patients

Background: Seasonal influenza poses a significant risk, and patients can benefit from early diagnosis and treatment. However, underdiagnosis and undertreatment remain wide-spread. We developed and compared clinical feature-based machine learning (ML) algorithms that can accurately predict influenza infection in emergency departments (EDs) among patients with influenza-like illness (ILI).Material and methods: We conducted a prospective cohort study in five EDs in the US and Taiwan from 2015 to 2020. Adult patients visiting the EDs with symptoms of ILI were recruited and tested by real-time RT-PCR for influenza. We evaluated seven ML algorithms and compared their results with previously developed clinical prediction models.Results: Out of the 2189 enrolled patients, 1104 tested positive for influenza. The eXtreme Gradient Boosting achieved superior performance with an area under the receiver operating characteristic curve of 0.82 (95% confidence interval [CI] = 0.79-0.85), with a sensitivity of 0.92 (95% CI = 0.88-0.95), specificity of 0.89 (95% CI = 0.86-0.92), and accuracy of 0.72 (95% CI = 0.69-0.76) in the testing set over cut-offs of 0.4, 0.6 and 0.5, respectively. These results were superior to those of previously proposed clinical prediction models. The model interpretation revealed that body temperature, cough, rhinorrhea, and exposure history were positively associated with and the days of illness and influenza vaccine were negatively associated with influenza infection. We also found the week of the influenza season, pulse rate, and oxygen saturation to be associated with influenza infection.Conclusions: The clinical feature-based ML model outperformed conventional models for predicting influenza infection.

Automated summary

The study developed and compared clinical feature-based machine learning algorithms for accurately predicting influenza infection in patients with influenza-like illness in emergency departments. The eXtreme Gradient Boosting model outperformed conventional models with superior performance, showing certain clinical features positively or negatively associated with influenza infection. This study highlights the potential of machine learning in improving early diagnosis and treatment of influenza.