Explainable Machine Learning Models for Pneumonia Mortality Risk Prediction Using MIMIC-III Data
James Sanii, Wai Yip Chan · 2022
To gain trust, machine learning (ML) models used in high stake applications such as clinical decision support need to provide explainable behaviours and outputs. To assess whether interpretable explanations can be obtained without sacrificing prediction performance, we compare using “black box” versus “glass box” models for predicting the mortality risk of patients diagnosed with pneumonia, using data in the MIMIC-III dataset. We examine five types of black box models: random forest (RF), support vector machine (SVM), gradient boosting classifier (GBC), AdaBoost (ADA), and multilayer perceptron (MLP), and three types of glassbox models: K-nearest neighbor (KNN), explainable boosting machine (EBM), and generalized additive models (GAM). When trained using 417 features, a black box RF model performs best with AUC of 0.896. With the feature set size reduced to 19, an EBM model performs the best with AUC 0.872. Both models exceed the AUC of 0.661, the best previously reported for the task. Our results suggest that ML models with inbuilt explainability may provide prediction power as attractive as black box models.