Automatic Recognition of High-Density Epileptic EEG Using Support Vector Machine and Gradient-Boosting Decision Tree

被引:4
作者
He, Jiaxiu [1 ]
Yang, Li [1 ]
Liu, Ding [1 ]
Song, Zhi [1 ]
机构
[1] CSU, Dept Neurol, Xiangya Hosp 3, Tongzipo St, Changsha 410013, Peoples R China
基金
中国国家自然科学基金;
关键词
EEG; epilepsy; machine learning; SVM; GBDT; SEIZURE DETECTION; DECOMPOSITION;
D O I
10.3390/brainsci12091197
中图分类号
Q189 [神经科学];
学科分类号
071006 ;
摘要
Background: Epilepsy (Ep) is a chronic neural disease. The diagnosis of epilepsy depends on detailed seizure history and scalp electroencephalogram (EEG) examinations. The automatic recognition of epileptic EEG is an artificial intelligence application developed from machine learning (ML). Purpose: This study compares the classification effects of two kinds of classifiers by controlling the EEG data source and characteristic values. Method: All EEG data were collected by GSN HydroCel 256 leads and high-density EEG from Xiangya Third Hospital. This study used time-domain features (mean, kurtosis and skewness processed by empirical mode decomposition (EMD) and three IMFs), a frequency-domain feature (power spectrum density, PSD) and a non-linear feature (Shannon entropy). Support vector machine (SVM) and gradient-boosting decision tree (GBDT) classifiers were used to recognize epileptic EEG. Result: The result of the SVM classifier showed an accuracy of 72.00%, precision of 73.98%, and an F1_score of 82.28%. Meanwhile, the result of the GBDT classifier showed a sensitivity of 98.57%, precision of 89.13%, F1_score of 93.40%, and an AUC of 0.9119. Conclusion: The comparison of GBDT and SVM by controlling the variables of the feature values and parameters of a classifier is presented. GBDT obtained the better classification accuracy (90.00%) and F1_score (93.40%).
引用
收藏
页数:10
相关论文
共 48 条
  • [1] EEG Microstate Features as an Automatic Recognition Model of High-Density Epileptic EEG Using Support Vector Machine
    Yang, Li
    He, Jiaxiu
    Liu, Ding
    Zheng, Wen
    Song, Zhi
    BRAIN SCIENCES, 2022, 12 (12)
  • [2] Automatic detection of abnormal EEG signals using wavelet feature extraction and gradient boosting decision tree
    Albaqami, Hezam
    Hassan, Ghulam Mubashar
    Subasi, Abdulhamit
    Datta, Amitava
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 70 (70)
  • [3] Human Activity Recognition Using Portable EEG Sensor and Support Vector Machine
    Mieee, Shafaq Zia
    Mieee, Ali Nawaz Khan
    Mukhtar, Mayyda
    Ali, Shan E.
    2021 7TH INTERNATIONAL CONFERENCE ON ENGINEERING AND EMERGING TECHNOLOGIES (ICEET 2021), 2021, : 241 - 246
  • [4] Automated Recognition of Epileptic EEG States Using a Combination of Symlet Wavelet Processing, Gradient Boosting Machine, and Grid Search Optimizer
    Wang, Xiashuang
    Gong, Guanghong
    Li, Ni
    SENSORS, 2019, 19 (02):
  • [5] A Review of Machine Learning Techniques using Decision Tree and Support Vector Machine
    Somvanshi, Madan
    Tambade, Shital
    Chavan, Pranjali
    Shinde, S. V.
    2016 INTERNATIONAL CONFERENCE ON COMPUTING COMMUNICATION CONTROL AND AUTOMATION (ICCUBEA), 2016,
  • [6] Seizure Detection in Epileptic EEG Using Short-Time Fourier Transform and Support Vector Machine
    Rizal, Achmad
    Priharti, Wahmisari
    Hadiyoso, Sugondo
    INTERNATIONAL JOURNAL OF ONLINE AND BIOMEDICAL ENGINEERING, 2021, 17 (14) : 65 - 78
  • [7] Speech Emotion Recognition in Machine Learning to Improve Accuracy using Novel Support Vector Machine and Compared with Decision Tree Algorithm
    Amartya, J. Guru Monish
    Kumar, S. Magesh
    JOURNAL OF PHARMACEUTICAL NEGATIVE RESULTS, 2022, 13 : 185 - 192
  • [8] Speech Emotion Recognition in Machine Learning to Improve Accuracy using Novel Support Vector Machine and Compared with Decision Tree Algorithm
    Amartya, J. Guru Monish
    Kumar, S. Magesh
    JOURNAL OF PHARMACEUTICAL NEGATIVE RESULTS, 2022, 13 : 185 - 192
  • [9] Predicting metabolic syndrome using decision tree and support vector machine methods
    Karimi-Alavijeh, Farzaneh
    Jalili, Saeed
    Sadeghi, Masoumeh
    ARYA ATHEROSCLEROSIS, 2016, 12 (03) : 146 - 152
  • [10] A Hybrid Machine Learning Framework of Gradient Boosting Decision Tree and Sequence Model for Predicting Escalation in Customer Support
    Gupta, Shubham
    2020 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2020,