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 条
  • [21] Performance evaluation for epileptic electroencephalogram (EEG) detection by using Neyman-Pearson criteria and a support vector machine
    Wang Chun-mei
    Zhang Chong-ming
    Zou Jun-zhong
    Zhang Jian
    PHYSICA A-STATISTICAL MECHANICS AND ITS APPLICATIONS, 2012, 391 (04) : 1602 - 1609
  • [22] Automatic depression diagnosis through hybrid EEG and near-infrared spectroscopy features using support vector machine
    Yi, Li
    Xie, Guojun
    Li, Zhihao
    Li, Xiaoling
    Zhang, Yizheng
    Wu, Kai
    Shao, Guangjian
    Lv, Biliang
    Jing, Huan
    Zhang, Chunguo
    Liang, Wenting
    Sun, Jinyan
    Hao, Zhifeng
    Liang, Jiaquan
    FRONTIERS IN NEUROSCIENCE, 2023, 17
  • [23] Support vector machine classification for large datasets using decision tree and Fisher linear discriminant
    Lopez Chau, Asdrubal
    Li, Xiaoou
    Yu, Wen
    FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2014, 36 : 57 - 65
  • [24] An EEG-Based Attentiveness Recognition System Using Hilbert–Huang Transform and Support Vector Machine
    Chia-Ju Peng
    Yi-Chun Chen
    Chun-Chuan Chen
    Shih-Jui Chen
    Barthélemy Cagneau
    Luc Chassagne
    Journal of Medical and Biological Engineering, 2020, 40 : 230 - 238
  • [25] Automatic recognition of sleep spindles in EEG via radial basis support vector machine based on a modified feature selection algorithm
    Nurettin Acır
    Cüneyt Güzeliş
    Neural Computing & Applications, 2005, 14 : 56 - 65
  • [26] Automatic recognition of sleep spindles in EEG via radial basis support vector machine based on a modified feature selection algorithm
    Acir, N
    Güzelis, C
    NEURAL COMPUTING & APPLICATIONS, 2005, 14 (01) : 56 - 65
  • [27] Prediction of electrochemical impedance spectroscopy of high-entropy alloys corrosion by using gradient boosting decision tree
    Wei, Boxin
    Xu, Jin
    Pang, Jingyu
    Huang, Zheng
    Wu, Jiajun
    Cai, Zheng
    Yan, Mengdi
    Sun, Cheng
    MATERIALS TODAY COMMUNICATIONS, 2022, 32
  • [28] Artificial Intelligence-based classification of schizophrenia: A high density EEG-support vector machine study
    Tikka, Sai Krishna
    Singh, Bikesh Kumar
    Nizamie, S. Haque
    Garg, Shobit
    Mandal, Sunandan
    Thakur, Kavita
    INDIAN JOURNAL OF PSYCHIATRY, 2020, 62 : S10 - S10
  • [29] An EEG-Based Attentiveness Recognition System Using Hilbert-Huang Transform and Support Vector Machine
    Peng, Chia-Ju
    Chen, Yi-Chun
    Chen, Chun-Chuan
    Chen, Shih-Jui
    Cagneau, Barthelemy
    Chassagne, Luc
    JOURNAL OF MEDICAL AND BIOLOGICAL ENGINEERING, 2020, 40 (02) : 230 - 238
  • [30] Caries Level Classification using K-Nearest Neighbor, Support Vector Machine, and Decision Tree using Zernike Moment Invariant Features
    Jusman, Yessi
    Nurkholid, Muhammad Ahdan Fawwaz
    Faiz, Muhammad Fajrul
    Puspita, Sartika
    Evellyne, Lady Olivia
    Muhammad, Kahfi
    2022 INTERNATIONAL CONFERENCE ON DATA SCIENCE AND ITS APPLICATIONS (ICODSA), 2022, : 7 - 11