A Hybrid ANN-GWO Algorithm for Prediction of Heart Disease

Abstract

The paper investigates the powerful of hybridizing two computational intelligence methods viz., Gray Wolf Optimization (GWO) and Artificial Neural Networks (ANN) for prediction of heart disease. Gray wolf optimization is a global search method while gradient-based back propagation method is a local search one. The proposed algorithm implies the ability of ANN to find a relationship between the input and the output variables while the stochastic search ability of GWO is used for finding the initial optimal weights and biases of the ANN to reduce the probability of ANN getting stuck at local minima and slowly converging to global optimum. For evaluation purpose, the performance of hybrid model (ANN-GWO) was compared with standard back-propagation neural network (BPNN) using Root Mean Square Error (RMSE). The results demonstrate that the proposed model increases the convergence speed and the accuracy of prediction.