Using Feature Selection from XGBoost to Predict MIC Values with Neural Networks
Cory Kromer-Edwards, Mariana Castanheira, Suely Oliveira · 2023
The minimum inhibitory concentration (MIC) is the lowest concentration of an antimicrobial agent (commonly known as an antibiotic) that inhibits bacterial growth. The interpretation of an MIC value for a bacterial organism, recovered from an infection guides the treatment of that infection by informing the physician if the treatment has a chance to be successful or not. Methods to generate the MIC values depend on bacterial growth and could take 24 hours to several days, delaying the implementation of appropriate therapy and increasing rates of morbidity and mortality. In this study, XGBoost and Neural Network (NN) models were used to predict the MIC values for 13 antimicrobial agents in parallel. The XGBoost model was then used for feature selection, and those features were used as input to the NN. The 29 selected features from the XGBoost model gave an F1 score of 0.808 when applied to a NN model. That NN model has a +-1 2-fold dilution accuracy of ׅ89 for all antibiotics. The results prove that the XGBoost model is capable of selecting important features from the dataset and those selected features can be used to generate a highly effective NN model that can predict MIC results.