1. Employing artificial bee colony algorithm to optimize the artificial neural network in heart disease prediction.
- Author
-
Asaad, Manal Mohammed Othman Farea, Wahid, Juliana, and Rahmat, Abdul Razak
- Subjects
- *
BEES algorithm , *HEART diseases , *HONEYBEES , *ARTIFICIAL neural networks , *MACHINE learning , *MACHINE tools , *HONEY - Abstract
Heart disease forecasting is a key issue in the clinical data analysis field. Artificial Neural Network (ANN) is a machine learning tool that can help doctors diagnose heart diseases more accurately and quickly. However, the design of ANN is complicated because it needs to identify optimal weight values and suitable network structures. This paper aims to optimize the weights of ANN for forecasting the existence of heart disease among humans by using Artificial Bee Colony (ABC) algorithm to train ANN and select optimal network weights. The experimentations are carried out on the UC Irvine Machine Learning Repository (UCI) heart disease datasets and tested by the MATLAB software machine learning tools. The accuracy, specificity, sensitivity (recall), precision, and F-score of the ANN trained by the ABC model (ABC-ANN) and ANN trained by ABC with backpropagation (ABC-BpNN) model are investigated. Between ABC-ANN and ABC-BpNN, it was shown that the ABC-BpNN model has a better predicting ability and could reach considerably higher accuracy with 90% and 90.9%, 88.9%, 90.8% and 90.9% for precision, specificity, recall, and F-score, respectively. The findings showed significant enhancements compared with previous studies that have utilized the same dataset. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF