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 |
||
|
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 |