DOI number:

10.1007/s00521-024-10166-6

Journal:

Neural Computing and Applications

Abstract:

Left atrial thrombus (LAT) impacts humans greatly and can result in ischemia and necrosis in severe cases. Therefore, health workers appeal to the social community to emphasize the importance of preventive treatment for LAT. This paper proposes an ensemble framework for risk prediction of LAT based on undersampling with replacement (EFRP-UR), addressing the problem of data imbalance. Firstly, in the feature selection process, we adopt the method of separately counting the essential features of data subsets. In view of the characteristics of class imbalance in medical data, we apply our improved undersamling method, “undersampling with replacement", to obtain a number of training subsets, train multiple base-classifiers, and use an iterative method to select the classifiers with better performance for subsequent integration, improving the prediction accuracy of the proposed EFRP-UR. To aim for disease risk prediction, we synthesize the results of different ensemble algorithms in the end to increase the recall rate. Applied to the LAT dataset obtained from the Regional Medical Center, our experimental results prove that the proposed EFRP-UR has improved in accuracy, recall rate and F1 value, compared with any single base-classifier. In addition, if comprehensive data on other diseases exist, EFRP-UR can also be transferred to predict other diseases.

Indexed by:

Journal paper

Document Type:

J

Volume:

36

Translation or Not:

no

Date of Publication:

2024-07-30

Included Journals:

SCI

Links to published journals:

https://doi.org/10.1007/s00521-024-10166-6