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.

An evolutionary machine learning for pulmonary hypertension animal model from arterial blood gas analysis

Identification of Pulmonary Hypertension Animal Models Using a New Evolutionary Machine Learning Framework Based on Blood Routine Indicators

Early Detection of Coronary Microvascular Dysfunction Using Machine Learning Algorithm Based on Vectorcardiography and Cardiodynamicsgram Features

Lasso Algorithm and Support Vector Machine Strategy to Screen Pulmonary Arterial Hypertension Gene Diagnostic Markers

An optimized machine learning method for predicting wogonin therapy for the treatment of pulmonary hypertension

Machine Learning for Diagnosis of Pulmonary Hypertension by Echocardiography

Exploring IRGs as a Biomarker of Pulmonary Hypertension Using Multiple Machine Learning Algorithms

Pulmonary Hypertension Detection Non-Invasively at Point-of-Care Using a Machine-Learned Algorithm

Predicting Coronary Atherosclerotic Heart Disease: An Extreme Learning Machine with Improved Salp Swarm Algorithm

Detection of pulmonary hypertension with six training strategies based on deep learning technology

Classification of pulmonary arterial pressure using photoplethysmography and bi-directional LSTM

Echocardiographic artificial intelligence for pulmonary hypertension classification

A comprehensive study on machine learning models combining with oversampling for bronchopulmonary dysplasia-associated pulmonary hypertension in very preterm infants

Establishment of Machine Learning-Based Tool for Early Detection of Pulmonary Embolism

Detection of pulmonary embolism severity using clinical characteristics, hematological indices, and machine learning techniques

Noninvasive Estimation of Mean Pulmonary Artery Pressure Using MRI, Computer Models, and Machine Learning

Support vector machine deep mining of electronic medical records to predict the prognosis of severe acute myocardial infarction

Identification of Potential Biomarkers for Group I Pulmonary Hypertension Based on Machine Learning and Bioinformatics Analysis

Predicting Pulmonary Hypertension by Electrocardiograms Using Machine Learning

Multi-Task Learning for Pulmonary Arterial Hypertension Prognosis Prediction Via Memory Drift and Prior Prompt Learning on 3D Chest CT

Identification of MACC1 as a potential biomarker for pulmonary arterial hypertension based on bioinformatics and machine learning