Pulmonary embolism (PE) is a cardiopulmonary condition that can be fatal. PE can lead to sudden cardiovascular collapse and is potentially life-threatening, necessitating risk classification to modify therapy following the diagnosis of PE. We collected clinical characteristics, routine blood data, and arterial blood gas analysis data from all 139 patients.
Combining these data, this paper proposes a PE risk stratified prediction framework based on machine learning technology. An improved algorithm is proposed by adding sobol sequence and black hole mechanism to the cuckoo search algorithm (CS), called SBCS. Based on the coupling of the enhanced algorithm and the kernel extreme learning machine (KELM), a prediction framework is also proposed.
To confirm the overall performance of SBCS, we run benchmark function experiments in this work. The results demonstrate that SBCS has great convergence accuracy and speed. Then, tests based on seven open data sets are carried out in this study to verify the performance of SBCS on the feature selection problem. To further demonstrate the usefulness and applicability of the SBCS-KELM framework, this paper conducts aided diagnosis experiments on PE data collected from the hospital.
The experiment findings show that the indicators chosen, such as syncope, systolic blood pressure (SBP), oxygen saturation (SaO2%), white blood cell (WBC), neutrophil percentage (NEUT%), and others, are crucial for the feature selection approach presented in this study to assess the severity of PE. The classification results reveal that the prediction model’s accuracy is 99.26% and its sensitivity is 98.57%. It is expected to become a new and accurate method to distinguish the severity of PE.