1
|
Liu L, Li Y, Liu N, Luo J, Deng J, Peng W, Bai Y, Zhang G, Zhao G, Yang N, Li C, Long X. Establishment of machine learning-based tool for early detection of pulmonary embolism. COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE 2024; 244:107977. [PMID: 38113803 DOI: 10.1016/j.cmpb.2023.107977] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Subscribe] [Scholar Register] [Received: 04/07/2023] [Revised: 09/11/2023] [Accepted: 12/11/2023] [Indexed: 12/21/2023]
Abstract
BACKGROUND AND OBJECTIVES Pulmonary embolism (PE) is a complex disease with high mortality and morbidity rate, leading to increasing society burden. However, current diagnosis is solely based on symptoms and laboratory data despite its complex pathology, which easily leads to misdiagnosis and missed diagnosis by inexperienced doctors. Especially, CT pulmonary angiography, the gold standard method, is not widely available. In this study, we aim to establish a rapid and accurate screening model for pulmonary embolism using machine learning technology. Importantly, data required for disease prediction are easily accessed, including routine laboratory data and medical record information of patients. METHODS We extracted features from patients' routine laboratory results and medical records, including blood routine, biochemical group, blood coagulation routine and other test results, as well as symptoms and medical history information. Samples with a feature loss rate greater than 0.8 were deleted from the original database. Data from 4723 cases were retained, 231 of which were positive for pulmonary embolism. 50 features were retained through the positive and negative statistical hypothesis testing which was used to build the predictive model. In order to avoid identification as majority-class samples caused by the imbalance of sample proportion, we used the method of Synthetic Minority Oversampling Technique (SMOTE) to increase the amount of information on minority samples. Five typical machine learning algorithms were used to model the screening of pulmonary embolism, including Support Vector Machines, Logistic Regression, Random Forest, XGBoost, and Back Propagation Neural Networks. To evaluate model performance, sensitivity, specificity and AUC curve were analyzed as the main evaluation indicators. Furthermore, a baseline model was established using the characteristics of the pulmonary embolism guidelines as a comparison model. RESULTS We found that XGBoost showed better performance compared to other models, with the highest sensitivity and specificity (0.99 and 0.99, respectively). Moreover, it showed significant improvement in performance compared to the baseline model (sensitivity and specificity were 0.76 and 0.76 respectively). More important, our model showed low missed diagnosis rate (0.46) and high AUC value (0.992). Finally, the calculation time of our model is only about 0.05 s to obtain the possibility of pulmonary embolism. CONCLUSIONS In this study, five machine learning classification models were established to assess the likelihood of patients suffering from pulmonary embolism, and the XGBoost model most significantly improved the precision, sensitivity, and AUC for pulmonary embolism screening. Collectively, we have established an AI-based model to accurately predict pulmonary embolism at early stage.
Collapse
Affiliation(s)
- Lijue Liu
- School of Automation, Central South University, Changsha, Hunan 410083, China; Xiangjiang Laboratory, Changsha 410205, China; Hunan Zixing Intelligent Medical Technology Co., Ltd, Changsha, Hunan 410000, China
| | - Yaming Li
- School of Automation, Central South University, Changsha, Hunan 410083, China
| | - Na Liu
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Jingmin Luo
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Jinhai Deng
- Hunan Zixing Intelligent Medical Technology Co., Ltd, Changsha, Hunan 410000, China; Richard Dimbleby Laboratory of Cancer Research, School of Cancer & Pharmaceutical Sciences, King's College London, London SE1 1UL, UK
| | - Weixiong Peng
- Hunan Zixing Intelligent Medical Technology Co., Ltd, Changsha, Hunan 410000, China; Department of Electrical and Electronic Engineering, College of Engineering, Southern University of Science and Technology (SUSTech), Shenzhen, Guangdong 518055, China
| | - Yongping Bai
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Guogang Zhang
- Department of Cardiovascular Medicine, The Third Xiangya Hospital, Central South University, Tongzipo Road 138#, Changsha 410008,China.
| | - Guihu Zhao
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Ning Yang
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Chuanchang Li
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| | - Xueying Long
- Xiangya Hospital, Central South University, Xiangya Road 87#, Changsha 410008, China
| |
Collapse
|