학술논문

A diabetes prediction model based on Boruta feature selection and ensemble learning
Document Type
article
Source
BMC Bioinformatics, Vol 24, Iss 1, Pp 1-34 (2023)
Subject
Diabetes detection
Machine learning
Boruta feature selection
K-Means++
Ensemble learning
Computer applications to medicine. Medical informatics
R858-859.7
Biology (General)
QH301-705.5
Language
English
ISSN
1471-2105
Abstract
Abstract Background and objective As a common chronic disease, diabetes is called the “second killer” among modern diseases. Currently, there is no medical cure for diabetes. We can only rely on medication for auxiliary treatment. However, many diabetic patients still die each year. In addition, a considerable number of people do not pay attention to their physical health or opt out of treatment due to lack of money, which eventually leads to various complications. Therefore, diagnosing diabetes at an early stage and intervening early is necessary; thus, developing an early detection method for diabetes is essential. Methods In this study, a diabetes prediction model based on Boruta feature selection and ensemble learning is proposed. The model contains the use of Boruta feature selection, the extraction of salient features from datasets, the use of the K-Means++ algorithm for unsupervised clustering of data and stacking of an ensemble learning method for classification. It has been validated on a diabetes dataset. Results The experiments were performed on the PIMA Indian diabetes dataset. The model was evaluated by accuracy, precision and F1 index. The obtained results show that the accuracy rate of the model reaches 98% and achieves good results. Conclusion Compared with other diabetes prediction models, this model achieved better results, and the obtained results indicate that this model is superior to other models in diabetes prediction and has better performance.