Disease classification via interpretable machine learning based on multi-center routine coagulation test

Read full text
Authors: Feng Dong, Yaqiong Zhang, W.H. Chen, Changmin Wang, Lei Zhang, Xiaoling Gao, Xiaoli Zhang, Minghua Jiang, Guobin Xu, Ruichuang Yang, Yutong Hou, Jiandang Ma
+2 more Chuanbao Li, Jun Wu
Cited by: 0
References: 18

Abstract

Background: This study aims to establish an interpretable disease classification model via machine learning and identify key features related to the disease to assist clinical disease diagnosis based on a multi-center CX9000 routine coagulation test. Methods: Data from 11 hospitals were collected. An unsupervised clustering model was used to extract classification patterns, and clinical experts assigned disease labels. Multiple machine learning models, including Random Forest, SVM, Decision Tree, Naive Bayes, MLP, XGBoost, and LightGBM, were trained. Ten-fold cross validation and external validation were performed. For external validation, models were trained with data from 8 hospitals (˜90%) and tested on the remaining 2 hospitals (˜10%). SHAP and Decision Tree analysis were used for interpretability. Results: Clear clustering patterns were observed for valvular heart disease (VHD) and pulmonary infection (PI). LightGBM achieved the best performance in both tasks. In cross validation, the mean F1-scores were 0.8890 and 0.7233, and the mean AUCs were 0.9500 and 0.8023. External validation showed strong generalization, with mean F1-scores of 0.9259 and 0.7464 and mean AUCs of 0.9493 and 0.8297. The sample visualization by t-SNE and the interpretable analysis by SHAP and Decision Trees identified some key classification features, i.e., international normalized ratio (INR) for VHD and age for PI. Conclusion: Machine learning models based on multi-center coagulation tests provide effective and interpretable disease classification, supporting clinical diagnostic automation.

Fields of Study & Subjects

Artificial Intelligence in Healthcare Machine Learning in Healthcare Clinical Laboratory Practices and Quality Control

Publication details

Journal / source
Frontiers in Molecular Biosciences
Volume
13
Pages
1788536–1788536
Publication date
4 March 2026
Language
EN