Modelling the rapid detection of Carbapenemase-resistant Klebsiella pneumoniae based on machine learning and matrix-assisted laser desorption/ionization time-of-flight mass spectrometry.

ABSTRACT

In this study, 80 carbapenem-resistant Klebsiella pneumoniae (CR-KP) and 160 carbapenem-susceptible Klebsiella pneumoniae (CS-KP) strains detected in the clinic were selected and their matrix-assisted laser desorption/ionization time-of-flight mass spectrometry (MALDI-TOF MS) peaks were collected. K-means clustering was performed on the MS peak data to obtain the best "feature peaks", and four different machine learning models were built to compare the area under the ROC curve, specificity, sensitivity, test set score, and ten-fold cross-validation score of the models. By adjusting the model parameters, the test efficacy of the model is increased on the basis of reducing model overfitting. The area under the ROC curve of the Random Forest, Support Vector Machine, Logistic Regression, and Xgboost models used in this study are 0.99, 0.97, 0.96, and 0.97, respectively; the model scores on the test set are 0.94, 0.91, 0.90, and 0.93, respectively; and the results of the ten-fold cross-validation are 0.84, 0.81, 0.81, and 0.85, respectively. Based on the machine learning algorithms and MALDI-TOF MS assay data can realize rapid detection of CR-KP, shorten the in-laboratory reporting time, and provide fast and reliable identification results of CR-KP and CS-KP.