Bulletin of Electrical Engineering and Informatics
Vol 11, No 4: August 2022

Electrocardiogram feature selection and performance improvement of sleep stages classification using grid search

Lyra Vega Ugi (Telkom University)
Fiky Yosef Suratman (Telkom University)
Unang Sunarya (Telkom University)



Article Info

Publish Date
01 Aug 2022

Abstract

Sleep analysis is often used to identify sleep-related human health. In many cases, sleep disorders could cause a particular disease. One of the approaches to detect sleep disorders is by investigating human sleep stages. However, the selection of the proper electrocardiogram (ECG) features is still considered challenging and becomes an issue to achieve the performance of the algorithm used. Therefore, it is necessary to investigate which ECG features are very significant to the performance of the algorithm. In this study, the support vector machine (SVM) method has been utilized to classify sleep stages into two classes namely awake and sleep. In order to improve the classification performances, an optimization method of grid search was used to find the best parameters of the SVM. Feature selection of information gain was then used to find the most significant ECG features. To validate the performance results, one leave-subject out cross-validation has been conducted during the implementation. There were ten subjects involved in this implementation. The ECG signals from those ten subjects were used to differentiate awake from sleep state. Based on the results, our method obtained an average accuracy of 85.46% a precision of 84.05% and a recall of 85.44% respectively.

Copyrights © 2022






Journal Info

Abbrev

EEI

Publisher

Subject

Electrical & Electronics Engineering

Description

Bulletin of Electrical Engineering and Informatics (Buletin Teknik Elektro dan Informatika) ISSN: 2089-3191, e-ISSN: 2302-9285 is open to submission from scholars and experts in the wide areas of electrical, electronics, instrumentation, control, telecommunication and computer engineering from the ...