Abstract
In this article, we used the Fast Correlation-Based Feature Selection (FCBF) method to filter redundant and irrelevant characteristics in order to improve the quality of heart disease classification. Then, we proposed PA-KNN a classification based K-Nearest Neighbour model optimized by Particle Swarm Optimization (PSO) associated with Ant Colony Optimization (ACO). The proposed mixed approach is applied to the heart disease dataset. The results demonstrate the effectiveness and robustness of the proposed hybrid method in processing various types of data for the classification of heart disease. Therefore, this study examines the different automatic learning algorithms and compares the results using different performance measures, i.e. Accuracy, Precision, Recall, F1-Score, etc. The data set used in this study comes from the UCI’s automatic learning repository, entitled “Heart Disease” Data set. We can be concluded that PA-KNN has demonstrated efficiency and robustness compared to other classification methods.
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Kamkar, I., Akbarzadeh-T, M., Yaghoobi, M.: Intelligent water drops a new optimization algorithm for solving the vehicle routing problem. In: 2010 IEEE International Conference on Systems, Man and Cybernetics, pp. 4142–4146 (2010)
Gazzaz, N.M., Yusoff, M.K., Ramli, M.F., Juahir, H., Aris, Z.: Artificial neural network modeling of the water quality index using land use areas as predictors. Water Environ. Res. 87(2), 99–112 (2015)
Zhang, Y., Wang, S., Ji, G.: A comprehensive survey on particle swarm optimization algorithm and its applications. Math. Probl. Eng. 2015, 38 (2015)
Alshamlan, H.M., Badr, G.H., Alohali, Y.A.: Genetic Bee Colony (GBC) algorithm: a new gene selection method for microarray cancer classification. Comput. Biol. Chem. 56, 49–60 (2015)
Yu, L., Liu, H.: Feature selection for high-dimensional data: a fast correlation-based filter solution. In: Proceedings of 20th International Conference on Machine Learning, pp. 856–863 (2003)
Fatima, M., Pasha, M.: Survey of machine learning algorithms for disease diagnostic. J. Intell. Learn. Syst. Appl. 9(1), 1 (2017)
Tan, K.C., Teoh, E.J., Yu, Q., Goh, K.C.: A hybrid evolutionary algorithm for attribute selection in data mining. Expert Syst. Appl. 36(4), 8616–8630 (2009)
Otoom, A.F., Abdallah, E.E., Kilani, Y., Kefaye, A., Ashour, M.: Effective diagnosis and monitoring of heart disease. Int. J. Softw. Eng. Appl. 9(1), 143–156 (2015)
Vembandasamy, K., Sasipriya, R., Deepa, E.: Heart diseases detection using Naive Bayes algorithm. IJISET-Int. J. Innov. Sci. Eng. Technol. 2, 441–444 (2015)
Lichman, M.: UCI Machine Learning Repositry (2013). https://archive.ics.uci.edu/
Dataset, U.H.: UCI Machine Learning Repository. http://archive.ics.uci.edu/ml/machine-learning-databases/heartdisease/Hear
Van Cauwenberge, L.: Top 10 machine learning algorithms. Data Sci. Cent. 6 (2015)
Poli, R., Kennedy, J., Blackwell, T.: Particle swarm optimization. Swarm Intell. 1(1), 33–57 (2007)
Kennedy, J.: Particle swarm optimization. In: Sammut, C., Webb, G.I. (eds.) Encyclopedia of Machine Learning, pp. 760–766. Springer, Boston (2010)
Shi, Y.: Particle swarm optimization. IEEE Connect. 2(1), 8–13 (2004)
Dorigo, M., Birattari, M.: Ant colony optimization. In: Sammut, C., Webb, G.I. (eds.) Encyclopedia of Machine Learning, 36–39. Springer, Boston (2010)
Tabakhi, S., Moradi, P., Akhlaghian, F.: An unsupervised feature selection algorithm based on ant colony optimization. Eng. Appl. Artif. Intell. 32, 112–123 (2014)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Khourdifi, Y., Bahaj, M. (2019). K-Nearest Neighbour Model Optimized by Particle Swarm Optimization and Ant Colony Optimization for Heart Disease Classification. In: Farhaoui, Y., Moussaid, L. (eds) Big Data and Smart Digital Environment. ICBDSDE 2018. Studies in Big Data, vol 53. Springer, Cham. https://doi.org/10.1007/978-3-030-12048-1_23
Download citation
DOI: https://doi.org/10.1007/978-3-030-12048-1_23
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-12047-4
Online ISBN: 978-3-030-12048-1
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)