A Prediction Technique for Heart Disease Based on Long Short Term Memory Recurrent Neural Network
Manohar Manur, Alok Kumar Pani, Pankaj Kumar
Abstract
In recent years, heart disease is one of the leading cause of death for both women and men. So, heart disease prediction is considered as a significant part in the clinical data analysis. Standard data mining techniques like Support Vector Machine (SVM), Nave Bayes and other machine learning techniques used in the earlier research for heart disease prediction. These methods are not sufficient for effective heart disease prediction due to insufficient test data. In this research, Bi-directional Long Short Term Memory with Conditional Random Field (BiLSTM-CRF) has been proposed to increase the efficiency of heart disease prediction. The input medical data were analyzed in a bidirectional manner for effective analysis, and CRF provided the linear relationship between the features. The BiLSTM-CRF method has been tested on the Cleveland dataset to analyze the performance and compared with existing methods. The results showed that the proposed BiLSTM-CRF outperformed the existing methods in heart disease prediction. The average accuracy of the proposed BiLSTM-CRF is 90.04%, which is higher than the existing methods.