Open AccessFeature optimization using Backward Elimination and Support Vector Machines (SVM) algorithm for diabetes classification
Author(s) -
Faisa Maulidina,
Zuherman Rustam,
Sri Hartini,
Velery Virgina Putri Wibowo,
Ilsya Wirasati,
Wismaji Sadewo
Publication year - 2021
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1821/1/012006
Subject(s) - support vector machine , feature selection , artificial intelligence , diabetes mellitus , feature (linguistics) , computer science , machine learning , algorithm , pattern recognition (psychology) , medicine , linguistics , philosophy , endocrinology
Diabetes is a disease that occurs when the blood glucose level is higher than normal and also leads to health problems. Early and accurate diagnosis needs to be carried out on individuals affected by this disease. Furthermore, excellent treatment needs to be provided to prevent worse situations. Some studies have used several machine learning methods to diagnose diabetes. Furthermore, in this study, the Backward Elimination and Support Vector Machine (SVM) algorithm was used to classify the PIMA Indians diabetes dataset. It consisted of 268 diabetic and 500 non-diabetic patients with eight attributes. Backward Elimination is a feature selection method used to remove irrelevant features based on the linear regression model. Using this method, the right features for the model was expected. This method has some advantages which include increasing training time, decreasing complexity and improving performance and accuracy. Therefore, the performance of SVM improved. Based on the experiments, it was discovered that by combining feature selection algorithm (backward elimination) and SVM, the highest accuracy obtained was 85.71% using 90% data training. Therefore, it was concluded that Backward Elimination combined with SVM algorithm is an excellent method to classify diabetes by using the PIMA Indians diabetes dataset.