XGBoost Classification of Epileptic EEG Using Nonlinear Dynamical Features and SHAP
Xiaojie Lu1,2,3, Hui Lou1, Xiaoyang Jin1
1School of Medical Information, Wannan Medical University, Wuhu 241002, China.
Abstract:
To evaluate whether nonlinear descriptors of electroencephalogram (EEG) signals support interpretable XGBoost classification and to determine how analysis window duration affects performance. A secondary analysis of the public Bonn EEG dataset was performed. Nine nonlinear features were extracted from non-overlapping 1, 5, 10, and 20 s windows after an original-recording-level train/validation/test split, and a multiclass XGBoost model was interpreted with class-specific SHAP values. The model achieved 93.3% overall accuracy; the class-specific AUC values were 0.978 for Z/O, 0.978 for N/F, and 0.984 for S. Across the four fixed-split duration conditions, the 1 s condition had the lowest descriptive performance, whereas the 5, 10, and 20 s conditions were broadly comparable; no uniquely optimal duration was established. The nonlinear-feature/XGBoost framework provides interpretable benchmark segment classification evidence. Because EEG is modeled as a stochastic process and the dataset is small and heterogeneous, the SHAP attributions do not establish physiological causality or clinical diagnostic validity.
