Skip to main content
Top

2021 | OriginalPaper | Chapter

A Model for Heart Disease Prediction Using Feature Selection with Deep Learning

Authors : Vaishali Baviskar, Madhushi Verma, Pradeep Chatterjee

Published in: Advanced Computing

Publisher: Springer Singapore

Activate our intelligent search to find suitable subject content or patents.

search-config
loading …

Abstract

The heart disease is considered as the most widespread disease. It is challenging for most of the physicians to diagnose at an early stage to avoid the risk of death rate. The main objective of this study involves the prediction of heart disease by using efficient techniques based on feature selection and classification. For feature selection, the enhanced genetic algorithm (GA) and particle swarm optimization (PSO) have been implemented. For classification, the recurrent neural network (RNN) and long short term memory (LSTM) has been implemented in this study. The data set used is the Cleveland heart disease data set available on UCI machine learning repository, and the performance of the proposed techniques has been evaluated by using various metrics like accuracy, precision, recall and f-measure. Finally, the results thus obtained have been compared with the existing models in terms of accuracy. It has been observed that LSTM when combined with PSO showed an accuracy of 93.5% whereas the best known existing model had an accuracy of 93.33%. Therefore, the proposed approach can be applied in the medical field for accurate heart disease prediction.

Dont have a licence yet? Then find out more about our products and how to get one now:

Springer Professional "Wirtschaft+Technik"

Online-Abonnement

Mit Springer Professional "Wirtschaft+Technik" erhalten Sie Zugriff auf:

  • über 102.000 Bücher
  • über 537 Zeitschriften

aus folgenden Fachgebieten:

  • Automobil + Motoren
  • Bauwesen + Immobilien
  • Business IT + Informatik
  • Elektrotechnik + Elektronik
  • Energie + Nachhaltigkeit
  • Finance + Banking
  • Management + Führung
  • Marketing + Vertrieb
  • Maschinenbau + Werkstoffe
  • Versicherung + Risiko

Jetzt Wissensvorsprung sichern!

Springer Professional "Technik"

Online-Abonnement

Mit Springer Professional "Technik" erhalten Sie Zugriff auf:

  • über 67.000 Bücher
  • über 390 Zeitschriften

aus folgenden Fachgebieten:

  • Automobil + Motoren
  • Bauwesen + Immobilien
  • Business IT + Informatik
  • Elektrotechnik + Elektronik
  • Energie + Nachhaltigkeit
  • Maschinenbau + Werkstoffe




 

Jetzt Wissensvorsprung sichern!

Springer Professional "Wirtschaft"

Online-Abonnement

Mit Springer Professional "Wirtschaft" erhalten Sie Zugriff auf:

  • über 67.000 Bücher
  • über 340 Zeitschriften

aus folgenden Fachgebieten:

  • Bauwesen + Immobilien
  • Business IT + Informatik
  • Finance + Banking
  • Management + Führung
  • Marketing + Vertrieb
  • Versicherung + Risiko




Jetzt Wissensvorsprung sichern!

Literature
1.
go back to reference Kirubha, V., Priya, S.M.: Survey on data mining algorithms in disease prediction. Int. J. Comput. Trends Tech. 38, 124–128 (2016)CrossRef Kirubha, V., Priya, S.M.: Survey on data mining algorithms in disease prediction. Int. J. Comput. Trends Tech. 38, 124–128 (2016)CrossRef
2.
go back to reference Sharma, H., Rizvi, M.: Prediction of heart disease using machine learning algorithms: a survey. Int. J. Recent Innov. Trends Comput. Commun. 5, 99–104 (2017) Sharma, H., Rizvi, M.: Prediction of heart disease using machine learning algorithms: a survey. Int. J. Recent Innov. Trends Comput. Commun. 5, 99–104 (2017)
3.
go back to reference Choi, E., Schuetz, A., Stewart, W.F., Sun, J.: Using recurrent neural network models for early detection of heart failure onset. J. Am. Med. Inform. Assoc. 24, 361–370 (2017)CrossRef Choi, E., Schuetz, A., Stewart, W.F., Sun, J.: Using recurrent neural network models for early detection of heart failure onset. J. Am. Med. Inform. Assoc. 24, 361–370 (2017)CrossRef
4.
go back to reference Jin, B., Che, C., Liu, Z., Zhang, S., Yin, X., Wei, X.: Predicting the risk of heart failure with EHR sequential data modelling. IEEE Access 6, 9256–9261 (2018)CrossRef Jin, B., Che, C., Liu, Z., Zhang, S., Yin, X., Wei, X.: Predicting the risk of heart failure with EHR sequential data modelling. IEEE Access 6, 9256–9261 (2018)CrossRef
5.
go back to reference Salem, T.: Study and analysis of prediction model for heart disease: an optimization approach using genetic algorithm. Int. J. Pure Appl. Math. 119, 5323–5336 (2018) Salem, T.: Study and analysis of prediction model for heart disease: an optimization approach using genetic algorithm. Int. J. Pure Appl. Math. 119, 5323–5336 (2018)
6.
go back to reference Bobaoglu, I., Findik, O., Ulker, E.: A comparison of feature selection models utilizing binary particle swarm optimization and genetic algorithm in determining coronary artery disease using support vector. Expert Syst. Appl. 37, 3177–3183 (2010)CrossRef Bobaoglu, I., Findik, O., Ulker, E.: A comparison of feature selection models utilizing binary particle swarm optimization and genetic algorithm in determining coronary artery disease using support vector. Expert Syst. Appl. 37, 3177–3183 (2010)CrossRef
9.
go back to reference Arabasadi, Z., Alizadehsani, R., Roshanzamir, M., Moosaei, H., Yarifard, A.A.: Computer aided decision making for heart disease detection using hybrid neural network-Genetic algorithm. Comput. Methods Programs Biomed. 141, 19–26 (2017)CrossRef Arabasadi, Z., Alizadehsani, R., Roshanzamir, M., Moosaei, H., Yarifard, A.A.: Computer aided decision making for heart disease detection using hybrid neural network-Genetic algorithm. Comput. Methods Programs Biomed. 141, 19–26 (2017)CrossRef
10.
go back to reference Uyar, K., İlhan, A.: Diagnosis of heart disease using genetic algorithm based trained recurrent fuzzy neural networks. Proc. Comput. Sci. 120, 588–593 (2017)CrossRef Uyar, K., İlhan, A.: Diagnosis of heart disease using genetic algorithm based trained recurrent fuzzy neural networks. Proc. Comput. Sci. 120, 588–593 (2017)CrossRef
12.
go back to reference Khourdifi, Y., Bahaj, M.: Heart disease prediction and classification using machine learning algorithms optimized by particle swarm optimization and ant colony optimization. Int. J. Intell. Eng. Syst. 12, 242–252 (2019) Khourdifi, Y., Bahaj, M.: Heart disease prediction and classification using machine learning algorithms optimized by particle swarm optimization and ant colony optimization. Int. J. Intell. Eng. Syst. 12, 242–252 (2019)
13.
go back to reference Feshki, M.G., Shijani, O.S.: Improving the heart disease diagnosis by evolutionary algorithm of PSO and Feed Forward Neural Network. In: Artificial Intelligence and Robotics (IRANOPEN) 2016, pp. 48–53 (2016) Feshki, M.G., Shijani, O.S.: Improving the heart disease diagnosis by evolutionary algorithm of PSO and Feed Forward Neural Network. In: Artificial Intelligence and Robotics (IRANOPEN) 2016, pp. 48–53 (2016)
14.
go back to reference Narasimhan, B., Malathi, A.: Altered particle swarm optimization based attribute selection strategy with improved fuzzy Artificial Neural Network classifier for coronary artery heart disease risk prediction. Int. J. Adv. Res. Ideas Innov. Technol. 5, 1196–1203 (2019) Narasimhan, B., Malathi, A.: Altered particle swarm optimization based attribute selection strategy with improved fuzzy Artificial Neural Network classifier for coronary artery heart disease risk prediction. Int. J. Adv. Res. Ideas Innov. Technol. 5, 1196–1203 (2019)
15.
go back to reference Pillai, N.S.R., Bee, K.K., Kiruthika, J.: Prediction of heart disease using RNN algorithm (2019) Pillai, N.S.R., Bee, K.K., Kiruthika, J.: Prediction of heart disease using RNN algorithm (2019)
16.
go back to reference Babu, S.B., Suneetha, A., Babu, G.C., Kumar, Y.J.N., Karuna, G.: Medical disease prediction using grey wolf optimization and auto encoder based recurrent neural network. Period. Eng. Nat. Sci. 6, 229–240 (2018) Babu, S.B., Suneetha, A., Babu, G.C., Kumar, Y.J.N., Karuna, G.: Medical disease prediction using grey wolf optimization and auto encoder based recurrent neural network. Period. Eng. Nat. Sci. 6, 229–240 (2018)
17.
go back to reference Islam, M.S., Umran, H.M., Umran, S.M., Karim, M.: Intelligent healthcare platform: cardiovascular disease risk factors prediction using attention module based LSTM. In: 2019 2nd International Conference on Artificial Intelligence and Big Data (ICAIBD), pp. 167–175 (2019) Islam, M.S., Umran, H.M., Umran, S.M., Karim, M.: Intelligent healthcare platform: cardiovascular disease risk factors prediction using attention module based LSTM. In: 2019 2nd International Conference on Artificial Intelligence and Big Data (ICAIBD), pp. 167–175 (2019)
19.
go back to reference Bandyopadhyay, S.K., Dutta, S.: Stacked bi-directional LSTM layer based model for prediction of possible heart disease during lockdown period of COVID-19: bidirectional LSTM. J. Adv. Res. Med. Sci. Technol. 7, 10–14 (2020). ISSN 2394-6539CrossRef Bandyopadhyay, S.K., Dutta, S.: Stacked bi-directional LSTM layer based model for prediction of possible heart disease during lockdown period of COVID-19: bidirectional LSTM. J. Adv. Res. Med. Sci. Technol. 7, 10–14 (2020). ISSN 2394-6539CrossRef
20.
go back to reference Sherstinsky, A.: Fundamentals of recurrent neural network (RNN) and long short-term memory (LSTM) network. Phys. D: Nonlinear Phenom. 404, 132306 (2020)MathSciNetCrossRef Sherstinsky, A.: Fundamentals of recurrent neural network (RNN) and long short-term memory (LSTM) network. Phys. D: Nonlinear Phenom. 404, 132306 (2020)MathSciNetCrossRef
22.
go back to reference Ali, L., Rahman, A., Khan, A., Zhou, M., Javeed, A., Khan, J.A.: An automated diagnostic system for heart disease prediction based on \({\chi ^{2}} \) statistical model and optimally configured deep neural network. IEEE Access 7, 34938–34945 (2019)CrossRef Ali, L., Rahman, A., Khan, A., Zhou, M., Javeed, A., Khan, J.A.: An automated diagnostic system for heart disease prediction based on \({\chi ^{2}} \) statistical model and optimally configured deep neural network. IEEE Access 7, 34938–34945 (2019)CrossRef
Metadata
Title
A Model for Heart Disease Prediction Using Feature Selection with Deep Learning
Authors
Vaishali Baviskar
Madhushi Verma
Pradeep Chatterjee
Copyright Year
2021
Publisher
Springer Singapore
DOI
https://doi.org/10.1007/978-981-16-0401-0_12

Premium Partner