A Machine Learning Method for Differentiation Crohn's Disease and Intestinal Tuberculosis

Yufeng Shu,Zhe Chen,Jingshu Chi,Sha Cheng,Huan Li,Peng Liu,Ju Luo

DOI: https://doi.org/10.2147/JMDH.S470429

2024-08-08

Abstract:Background: Whether machine learning (ML) can assist in the diagnosis of Crohn's disease (CD) and intestinal tuberculosis (ITB) remains to be explored. Methods: We collected clinical data from 241 patients, and 51 parameters were included. Six ML methods were tested, including logistic regression, decision tree, k-nearest neighbor, multinomial NB, multilayer perceptron, and XGBoost. SHAP and LIME were subsequently introduced as interpretability methods. The ML model was tested in a real-world clinical practice and compared with a multidisciplinary team (MDT) meeting. Results: XGBoost displays the best performance among the six ML models. The diagnostic AUROC and the accuracy of XGBoost were 0.946 and 0.884, respectively. The top three clinical features affecting our ML model's result prediction were T-spot, pulmonary tuberculosis, and onset age. The ML model's accuracy, sensitivity, and specificity in clinical practice were 0.860, 0.833, and 0.871, respectively. The agreement rate and kappa coefficient of the ML and MDT methods were 90.7% and 0.780, respectively (P<0.001). Conclusion: We developed an ML model based on XGBoost. The ML model could provide effective and efficient differential diagnoses of ITB and CD with diagnostic bases. The ML model performs well in real-world clinical practice, and the agreement between the ML model and MDT is strong.