An ensemble framework for risk prediction of left atrial thrombus based on undersampling with replacement
发布时间:2024-08-09
点击次数:
- DOI码:
- 10.1007/s00521-024-10166-6
- 发表刊物:
- Neural Computing and Applications
- 摘要:
- 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.
- 第一作者:
- Li Li
- 论文类型:
- 期刊论文
- 文献类型:
- J
- 卷号:
- 36
- 是否译文:
- 否
- 发表时间:
- 2024-07-30
- 收录刊物:
- SCI