Forest-EMCBE: an evolutionary ensemble learning algorithm for multiclass diagnosis of bacterial pneumonia using the...

Rapid diagnosis of bacterial pneumonia is crucial for clinical diagnosis and treatment, but traditional methods are time-consuming. The wide application of machine learning techniques in medical diagnosis provides an effective way to solve this problem. However, the complexity of medical datasets and the problem of class imbalance poses serious challenges to classical machine learning algorithms.

Aiming at the multiclass imbalanced problem in complete blood count (CBC) datasets, this study proposes a novel ensemble learning algorithm, Forest of Evolutionary Multi-Classifiers Based on Bagging with Error-Correcting Output Coding (Forest-EMCBE). The algorithm integrates Multi-Objective Genetic Algorithm, Error-Correcting Output Codes (ECOC), and balanced sampling strategy, which enhances the generalization ability of the classifiers through a three-layer integrated structure.

To validate the effectiveness of the proposed method, we trained the diagnostic model on a CBC dataset, which contains 1,457 samples and 4 different classes of bacterial pneumonia results, and compared it with 11 state-of-the-art algorithms. The experimental results demonstrate the superior performance of the Forest-EMCBE algorithm on the CBC dataset, outperforming all other compared algorithms.

Based on the Shapley value-based feature importance analysis method, this study dissects the contributions of key features to the prediction outcomes and further elucidates the differential impacts of features such as age, gender, and neutrophil percentage on predicting infections by different bacterial species.