Application of routine laboratory data in combination with machine learning in the differential diagnosis of lung tuberculosis / 中华检验医学杂志

ABSTRACT

Objective:To investigate the application value of establishing the differential diagnosis model of pulmonary tuberculosis using routine laboratory data.Methods:The retrospective study was conducted. The routine laboratory data of newly diagnosed patients with pulmonary tuberculosis and other pulmonary diseases in Beijng Jishuitan Hospital and Beijing Hepingli Hospital from May 2015 to November 2021were collected. According to the random numbers showed in the computer, all the 11516 patients were divided into training dataset and test dataset with a ratio of 9∶1. Four machine learning algorithms, Support Vector Machine, Random Forest, K-Nearest Neighbor and Logistic Regression, were used to build models and select features. The diagnostic accuracy of each model was verified by using the 10-fold cross-validation method and the performance of each model was evaluated by using the receptor operator of characteristic (ROC) curve.Results:Random Forest was selected as the optimal machine learning algorithm to build the best feature model in the study. According to importance scale of factors, the differential diagnosis model of pulmonary tuberculosis consisting of 37 non-specific test indexes. In the validation set and test set the accuracy and area under curve (AUC) of the models were 0.747 and 0.736, the sensitivity, specificity and accuracy were 68.03% and 68.75%, 70.91% and 67.90%, 70.30% and 68.12%, respectively.Conclusion:A key tool in the differential diagnosis model of pulmonary tuberculosis was established by routine laboratory data in combination with machine learning. The results of this study need to be further verified by more data from medical institutions.