Optimization of feature selection in Support Vector Machines (SVM) using recursive feature elimination (RFE) and particle swarm optimization (PSO) for heart disease detection

Bayuaji, Luhur and Kusnadi, Kusnadi and Amzah, Mochamad Yamin and Pebrianti, Dwi (2024) Optimization of feature selection in Support Vector Machines (SVM) using recursive feature elimination (RFE) and particle swarm optimization (PSO) for heart disease detection. In: 9th International Conference on Mechatronics Engineering (ICOM 2024), 13th - 14th August 2024, Kuala Lumpur, Malaysia.

	PDF (Full Paper) - Published Version Restricted to Repository staff only Download (539kB) | Request a copy
Preview	PDF (Scopus) - Supplemental Material Download (160kB) | Preview

Abstract

Heart disease is one of the main health problems throughout the world. Early and accurate detection of heart disease is very important to reduce the death rate caused by this condition. One effective approach to detect heart disease is to use Support Vector Machine (SVM) as a machine learning algorithm. However, when using SVM, selecting the right features is very important to improve prediction performance. Inappropriate feature selection can lead to excess dimensions, high computing time, as well as the possibility of adding irrelevant information. Apart from that, the choice of C and gamma parameters can affect SVM performance. The RFE method is used to select the most informative and relevant subset of features from a given feature set. RFE combines a dimensionality reduction process with a machine learning process where the least important features are iteratively removed until the best feature subset is obtained, while the PSO method is used to optimize the C and gamma parameters of SVM. The results of the SVM-RFE PSO model trial showed increased accuracy in classifying heart disease. The accuracy value increased from 86.41% to 89.13%, indicating that this approach compares favorably with conventional SVM models or even SVM with RFE alone. These results illustrate that using the combination of RFE and PSO together is effective in improving the ability of SVM in classifying heart disease.

Item Type:	Proceeding Paper (Slide Presentation)
Additional Information:	10732/115360
Uncontrolled Keywords:	Detection, SVM, RFE, PSO, Heart Disease, Machine Learning
Subjects:	T Technology > T Technology (General)
Kulliyyahs/Centres/Divisions/Institutes (Can select more than one option. Press CONTROL button):	Kulliyyah of Engineering > Department of Mechanical Engineering Kulliyyah of Engineering
Depositing User:	Dr Dwi Pebrianti
Date Deposited:	29 Oct 2024 15:45
Last Modified:	29 Oct 2024 15:55
URI:	http://irep.iium.edu.my/id/eprint/115360

Actions (login required)

View Item