An efficient feature selection and explainable classification method for EEG-based seizure detection

被引:14
作者
Ahmad, Ijaz [1 ,2 ,3 ]
Yao, Chen [4 ]
Li, Lin [5 ]
Chen, Yan [5 ]
Liu, Zhenzhen [5 ]
Ullah, Inam [6 ]
Shabaz, Mohammad [7 ]
Wang, Xin [1 ,2 ,3 ]
Huang, Kaiyang [8 ]
Li, Guanglin [1 ,3 ]
Zhao, Guoru [1 ,2 ,3 ]
Samuel, Oluwarotimi Williams [2 ,9 ,10 ]
Chen, Shixiong [1 ,2 ,3 ]
机构
[1] Shenzhen Inst Adv Technol, Chinese Acad Sci, CAS Key Lab Human Machine Intelligence Synergy Sys, Shenzhen 518055, Guangdong, Peoples R China
[2] Univ Chinese Acad Sci, Shenzhen Coll Adv Technol, Shenzhen 518055, Guangdong, Peoples R China
[3] Guangdong Hong Kong Macao Joint Lab Human Machine, Shenzhen 518055, Guangdong, Peoples R China
[4] Shenzhen Univ, Affiliated Hosp 1, Shenzhen Peoples Hosp 2, Dept Neurosurg,Natl Key Clin Specialty,Shenzhen Ke, 3002 Sungang Rd, Shenzhen 518055, Guangdong, Peoples R China
[5] Shenzhen Childrens Hosp, Epilepsy Ctr, Surg Div, Shenzhen 518038, Guangdong, Peoples R China
[6] Gachon Univ, Dept Comp Engn, Seongnam 13120, South Korea
[7] Model Inst Engn & Technol, Jammu, Jammu & Kashmir, India
[8] Guangzhou Univ Chinese Med, Med Informat Engn, Guangzhou 510405, Peoples R China
[9] Univ Derby, Sch Comp, Derby DE22 3AW, England
[10] Univ Derby, Data Sci Res Ctr, Derby DE22 3AW, England
基金
中国博士后科学基金; 国家重点研发计划; 中国国家自然科学基金;
关键词
Electroencephalogram; Machine learning; Coefficient correlation; Distance correlation; Biomedical signals; Explainable artificial intelligence; EPILEPTIC SEIZURE; SIGNALS; DIAGNOSIS;
D O I
10.1016/j.jisa.2023.103654
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Epilepsy is a prevalent neurological disorder that poses life-threatening emergencies. Early electroencephalogram (EEG) seizure detection can mitigate the risks and aid in the treatment of patients with epilepsy. EEG based epileptic seizure (ES) detection has significant applications in epilepsy treatment and medical diagnosis. Therefore, this paper presents an innovative framework for efficient ES detection, providing coefficient and distance correlation feature selection algorithms, a Bagged Tree-based classifer (BTBC), and Explainable Artificial Intelligence (XAI). Initially, the Butterworth filter is employed to eliminate various artifacts, and the discrete wavelet transform (DWT) is used to decompose the EEG signals and extract various eigenvalue features of the statistical time domain (STD) as linear and Fractal dimension-based non-linear (FD-NL). The optimal features are then identified through correlation coefficients with P-value and distance correlation analysis.These features are subsequently utilized by the Bagged Tree-based classifer (BTBC). The proposed model provides best performance in mitigating overfitting issues and improves the average accuracy by 2% using (CD, E), (AB, CD, E), and (A, B) experimental types as compared to other machine learning (ML) models using well-known Bonn and UCI-EEG benchmark datasets. Finally, SHapley additive exPlanation (SHAP) was used to interpret and explain the decision-making process of the proposed model. The results highlight the framework's capability to accurately classify ES, thereby improving the diagnosis process in patients with brain dysfunctions.
引用
收藏
页数:17
相关论文
共 42 条
[1]   A hybrid technique for EEG signals evaluation and classification as a step towards to neurological and cerebral disorders diagnosis [J].
Abdulbaqi, Azmi Shawkat ;
Younis, Muhanad Tahrir ;
Younus, Younus Tahreer ;
Obaid, Ahmed J. .
INTERNATIONAL JOURNAL OF NONLINEAR ANALYSIS AND APPLICATIONS, 2022, 13 (01) :773-781
[2]   RETRACTED: EEG-Based Epileptic Seizure Detection via Machine/Deep Learning Approaches: A Systematic Review (Retracted Article) [J].
Ahmad, Ijaz ;
Wang, Xin ;
Zhu, Mingxing ;
Wang, Cheng ;
Pi, Yao ;
Khan, Javed Ali ;
Khan, Siyab ;
Samuel, Oluwarotimi Williams ;
Chen, Shixiong ;
Li, Guanglin .
COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
[4]  
Al-Qazzaz N.K., 2023, ADV NONINVASIVE BIOM, P345
[5]   Extracting epileptic features in EEGs using a dual-tree complex wavelet transform coupled with a classification algorithm [J].
Al-Salman, Wessam ;
Li, Yan ;
Wen, Peng ;
Miften, Firas Sabar Y. ;
Oudah, Atheer ;
Al Ghayab, Hadi Ratham .
BRAIN RESEARCH, 2022, 1779
[6]   Selection of optimal wavelet features for epileptic EEG signal classification with LSTM [J].
Aliyu, Ibrahim ;
Lim, Chang Gyoon .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (02) :1077-1097
[7]  
Almustafa K.M., 2020, Inf Med, V21, P100444
[8]   Indications of nonlinear deterministic and finite-dimensional structures in time series of brain electrical activity: Dependence on recording region and brain state [J].
Andrzejak, RG ;
Lehnertz, K ;
Mormann, F ;
Rieke, C ;
David, P ;
Elger, CE .
PHYSICAL REVIEW E, 2001, 64 (06) :8-061907
[9]  
[Anonymous], 2008, Accelerometry based detection of epileptic seizures
[10]   Comparative analysis of ANN performance of four feature extraction methods used in the detection of epileptic seizures [J].
Demirci, Burcu Acar ;
Demirci, Osman ;
Engin, Mehmet .
COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 166