Abstract
Cardiovascular diseases kill approximately millions of people globally every year. Heart failure (HF) occurs when the heart cannot pump enough blood to meet the needs of the body. This means that the heart is overworked and unable to respond to the speed and demands of other activities. This will lead to fatigue and shortness of breath while performing daily activities. In this research, the authors apply a machine learning model to predict heart failure patients’ survival based on the original set of available medical features. Therefore, a novel wrapper-based feature selection utilizing an Adaptive Particle Swarm Grey Wolf Optimization (APSGWO) is proposed to enhance the architecture of Multilayer Perceptron (MLP) and reduce the number of required input attributes. Moreover, we also compared the results of our proposed method and several conventional machine learning models such as Support Vector Machine (SVM), Decision Tree (DT), K–Nearest Neighbor, Naïve Bayesian Classifier (NBC), Random Forest (RF), and Logistic Regression (LR). The results of our method show not only that much fewer features are needed, but also higher accuracy can be accomplished, 81% for Adaptive Particle Swarm Grey Wolf Optimization—Multilayer Perceptron (APSGWO—MLP). This work can be applied in practice to become an effective tool to support the diagnosis for doctors.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
L. Yue, D. Tian, W. Chen, X. Han, M. Yin, Deep learning for heterogeneous medical data analysis. World Wide Web 23(5), 2715–2737 (2020). https://doi.org/10.1007/s11280-019-00764-z
Mining telemonitored physiological data and patient-reported outcomes of congestive heart failure patients. https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0190323. Accessed 13 Nov 2020
T. Ee, P. Tg, K. Gs, N. Kk, F. Di, Heart failure: diagnosis, severity estimation and prediction of adverse events through machine learning techniques. Comput. Struct. Biotechnol. J. 15, 26–47 (2016). https://doi.org/10.1016/j.csbj.2016.11.001
L. Hussain, I.A. Awan, W. Aziz, S. Saeed, A. Ali, Detecting congestive heart failure by extracting multimodal features and employing machine learning techniques. Biomed. Res. Int. 2020, 1–19 (2020). https://doi.org/10.1155/2020/4281243
C.S. Dangare, S.S. Apte, A data mining approach for prediction of heart disease using neural networks. Res. Gate 3(3), 30–40 (2012)
S. Smiley, Diagnostic for heart disease with machine learning. Medium (2020). https://towardsdatascience.com/diagnostic-for-heart-disease-with-machine-learning-81b064a3c1dd. Accessed 19 Sep 2020
G. Marques, D. Agarwal, I. de la Torre Díez, Automated medical diagnosis of COVID-19 through EfficientNet convolutional neural network. Appl. Soft Comput. 96, 106691 (2020). https://doi.org/10.1016/j.asoc.2020.106691
R.E. Wright, “Logistic regression”, in Reading and understanding multivariate statistics (Am. Psychol. Assoc., Washington, DC, US, 1995), pp. 217–244
An Introduction to Kernel and Nearest-Neighbor Nonparametric Regression: The American Statistician: Vol. 46, No 3. https://www.tandfonline.com/doi/abs/10.1080/00031305.1992.10475879. Accessed 06 Sep 2020
K.M. Ting, Z. Zheng, Improving the performance of boosting for Naive Bayesian classification, in Methodologies for Knowledge Discovery and Data Mining (Berlin, Heidelberg, 1999), pp. 296–305. https://doi.org/10.1007/3-540-48912-6_41
Statistical Learning Theory, Wiley. https://www.wiley.com/en-us/Statistical+Learning+Theory-p-9780471030034. Accessed 06 Sep 2020
J.R. Quinlan, Induction of decision trees. Mach. Learn. 1(1), 81–106 (1986). https://doi.org/10.1007/BF00116251
D. Chicco, G. Jurman, Machine learning can predict survival of patients with heart failure from serum creatinine and ejection fraction alone. BMC Med. Inform. Decis. Mak. 20(1), 16 (2020). https://doi.org/10.1186/s12911-020-1023-5
G. Guidi, M.C. Pettenati, P. Melillo, E. Iadanza, A machine learning system to improve heart failure patient assistance. IEEE J. Biomed. Health Inform. 18(6), 1750–1756 (2014). https://doi.org/10.1109/JBHI.2014.2337752
L. Yu, H. Liu, Efficient feature selection via analysis of relevance and redundancy. J. Mach. Learn. Res. 5, 1205–1224 (2004)
Y. Yang, J. Pedersen, A Comparative Study on Feature Selection in Text Categorization (1997)
K. Yan, D. Zhang, Feature selection and analysis on correlated gas sensor data with recursive feature elimination. Sens. Actuators B Chem. 212, 353–363 (2015). https://doi.org/10.1016/j.snb.2015.02.025
A. Jain, D. Zongker, Feature selection: evaluation, application, and small sample performance. IEEE Trans. Pattern Anal. Mach. Intell. 19(2), 153–158 (1997). https://doi.org/10.1109/34.574797
C. De Stefano, F. Fontanella, C. Marrocco, A. Scotto di Freca, A GA-based feature selection approach with an application to handwritten character recognition. Pattern Recogn. Lett. 35, 130–141 (2014). https://doi.org/10.1016/j.patrec.2013.01.026
E. Zorarpacı, S.A. Özel, A hybrid approach of differential evolution and artificial bee colony for feature selection. Expert Syst. Appl. 62, 91–103 (2016). https://doi.org/10.1016/j.eswa.2016.06.004
J. Too, A.R. Abdullah, N. Mohd Saad, N. Mohd Ali, W. Tee, A new competitive binary grey wolf optimizer to solve the feature selection problem in EMG signals classification. Computers 7(4), Art. no. 4 (2018). https://doi.org/10.3390/computers7040058
E. Emary, H.M. Zawbaa, A.E. Hassanien, Binary grey wolf optimization approaches for feature selection. Neurocomputing 172, 371–381 (2016). https://doi.org/10.1016/j.neucom.2015.06.083
S. Mirjalili, S.M. Mirjalili, A. Lewis, Grey Wolf optimizer. Adv. Eng. Softw. 69, 46–61 (2014). https://doi.org/10.1016/j.advengsoft.2013.12.007
L.-Y. Chuang, H.-W. Chang, C.-J. Tu, C.-H. Yang, Improved binary PSO for feature selection using gene expression data. Comput. Biol. Chem. 32(1), 29–38 (2008). https://doi.org/10.1016/j.compbiolchem.2007.09.005
M.H. Aghdam, N. Ghasem-Aghaee, M.E. Basiri, Text feature selection using ant colony optimization. Expert Syst. Appl. 36(3), Part 2, pp. 6843–6853 (2009). https://doi.org/10.1016/j.eswa.2008.08.022
Feature Selection with Discrete Binary Differential Evolution—IEEE Conference Publication. https://ieeexplore.ieee.org/document/5376334. Accessed 06 Sep 2020
Q. Li et al., An enhanced grey wolf optimization based feature selection wrapped Kernel extreme learning machine for medical diagnosis. Comput. Math. Methods Med. (2017). https://www.hindawi.com/journals/cmmm/2017/9512741/. Accessed 8 Nov 2020
H. Faris, I. Aljarah, M.A. Al-Betar, S. Mirjalili, Grey wolf optimizer: a review of recent variants and applications. Neural Comput. Appl. 30(2), 413–435 (2018). https://doi.org/10.1007/s00521-017-3272-5
M.D. Wilson, Support vector machines, in Encyclopedia of Ecology, ed. by S.E. Jørgensen, B.D. Fath (Academic Press, Oxford, 2008), pp. 3431–3437
J. Kennedy, R. Eberhart, Particle swarm optimization, in Proceedings of ICNN’95—International Conference on Neural Networks, vol. 4 (1995), pp. 1942–1948. https://doi.org/10.1109/icnn.1995.488968
M. Jahangir, H. Afzal, M. Ahmed, K. Khurshid, R. Nawaz, An expert system for diabetes prediction using auto tuned multi-layer perceptron, in IEEE Intelligent Systems (2018), pp. 722–728
Auto-MeDiSine: an auto-tunable medical decision support engine using an automated class outlier detection method and AutoMLP, springerprofessional.de. https://www.springerprofessional.de/en/auto-medisine-an-auto-tunable-medical-decision-support-engine-us/16630282. Accessed 19 Sep 2020
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this chapter
Cite this chapter
Le, T.M., Pham, T.N., Dao, S.T. (2021). A Novel Wrapper-Based Feature Selection for Heart Failure Prediction Using an Adaptive Particle Swarm Grey Wolf Optimization. In: Marques, G., Kumar Bhoi, A., de la Torre Díez, I., Garcia-Zapirain, B. (eds) Enhanced Telemedicine and e-Health. Studies in Fuzziness and Soft Computing, vol 410. Springer, Cham. https://doi.org/10.1007/978-3-030-70111-6_15
Download citation
DOI: https://doi.org/10.1007/978-3-030-70111-6_15
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-70110-9
Online ISBN: 978-3-030-70111-6
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)