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 条
  • [31] Heart Plaque Detection with Improved Accuracy using Decision tree in comparison with Least Squares Support Vector Machine
    Kumar, Vankamaddi Sunil
    Vidhya, K.
    CARDIOMETRY, 2022, (25): : 1584 - 1589
  • [32] Automatic Recognition of Village in Remote Sensing Images by Support Vector Machine Using Co-Occurrence Matrices
    Li, Li
    Chen, Yingyi
    Gao, Hongju
    Li, Daoliang
    SENSOR LETTERS, 2012, 10 (1-2) : 523 - 528
  • [33] Learning machines and sleeping brains: Automatic sleep stage classification using decision-tree multi-class support vector machines
    Lajnef, Tarek
    Chaibi, Sahbi
    Ruby, Perrine
    Aguera, Pierre-Emmanuel
    Eichenlaub, Jean-Baptiste
    Samet, Mounir
    Kachouri, Abdennaceur
    Jerbi, Karim
    JOURNAL OF NEUROSCIENCE METHODS, 2015, 250 : 94 - 105
  • [34] Features classification using geometrical deformation feature vector of support vector machine and active appearance algorithm for automatic facial expression recognition
    Rajesh A. Patil
    Vineet Sahula
    A. S. Mandal
    Machine Vision and Applications, 2014, 25 : 747 - 761
  • [35] Features classification using geometrical deformation feature vector of support vector machine and active appearance algorithm for automatic facial expression recognition
    Patil, Rajesh A.
    Sahula, Vineet
    Mandal, A. S.
    MACHINE VISION AND APPLICATIONS, 2014, 25 (03) : 747 - 761
  • [36] Support vector machine-based decision tree for snow cover extraction in mountain areas using high spatial resolution remote sensing image
    Zhu, Liujun
    Xiao, Pengfeng
    Feng, Xuezhi
    Zhang, Xueliang
    Wang, Zuo
    Jiang, Luyuan
    JOURNAL OF APPLIED REMOTE SENSING, 2014, 8
  • [37] High Resolution Water Table Modeling of the Shallow Groundwater Using a Knowledge-Guided Gradient Boosting Decision Tree Model
    Koch, Julian
    Gotfredsen, Jane
    Schneider, Raphael
    Troldborg, Lars
    Stisen, Simon
    Henriksen, Hans Jorgen
    FRONTIERS IN WATER, 2021, 3
  • [38] Evaluating the High Risk Groups for Suicide: A Comparison of Logistic Regression, Support Vector Machine, Decision Tree and Artificial Neural Network
    Amini, Payam
    Ahmadinia, Hasan
    Poorolajal, Jalal
    Moqaddasi Amiri, Mohammad
    IRANIAN JOURNAL OF PUBLIC HEALTH, 2016, 45 (09) : 1179 - 1187
  • [39] Machine learning for inference: using gradient boosting decision tree to assess non-linear effects of bus rapid transit on house prices
    Yang, Linchuan
    Liang, Yuan
    Zhu, Qing
    Chu, Xiaoling
    ANNALS OF GIS, 2021, 27 (03) : 273 - 284
  • [40] Modeling High Pan Evaporation Losses Using Support Vector Machine, Gaussian Processes, and Regression Tree Models
    Alsumaiei, Abdullah A.
    JOURNAL OF HYDROLOGIC ENGINEERING, 2024, 29 (05)