Machine-Learning-Based Diagnostics of EEG Pathology

Abstract

Machine learning (ML) methods have the potential to automate clinical EEG analysis. They can be categorized into feature-based (with handcrafted features), and end-to-end approaches (with learned features). Previous studies on EEG pathology decoding have typically analyzed a limited number of features, decoders, or both. For a I) more elaborate feature-based EEG analysis, and II) in-depth comparisons of both approaches, here we first develop a comprehensive feature-based framework, and then compare this framework to state-of-the-art end-to-end methods. To this aim, we apply the proposed feature-based framework and deep neural networks including an EEG-optimized temporal convolutional network (TCN) to the task of pathological versus non-pathological EEG classification. For a robust comparison, we chose the Temple University Hospital (TUH) Abnormal EEG Corpus (v2.0.0), which contains approximately 3000 EEG recordings. The results demonstrate that the proposed feature-based decoding framework can achieve accuracies on the same level as state-of-the-art deep neural networks. We find accuracies across both approaches in an astonishingly narrow range from 81--86\%. Moreover, visualizations and analyses indicated that both approaches used similar aspects of the data, e.g., delta and theta band power at temporal electrode locations. We argue that the accuracies of current binary EEG pathology decoders could saturate near 90\% due to the imperfect inter-rater agreement of the clinical labels, and that such decoders are already clinically useful, such as in areas where clinical EEG experts are rare. We make the proposed feature-based framework available open source and thus offer a new tool for EEG machine learning research.

References in corpus (18)

• Adam: A Method for Stochastic Optimization
• Scikit-learn: Machine Learning in Python
• EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfaces
• An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling
• Deep learning with convolutional neural networks for EEG decoding and visualization
• Methods for Interpreting and Understanding Deep Neural Networks
• Scalable and accurate deep learning for electronic health records
• Neural Architecture Search: A Survey
• SGDR: Stochastic Gradient Descent with Warm Restarts
• Tensor Decomposition for Signal Processing and Machine Learning
• AutoAugment: Learning Augmentation Policies from Data
• Visualizing the Loss Landscape of Neural Nets
• SLEEPNET: Automated Sleep Staging System via Deep Learning
• Early Seizure Detection with an Energy-Efficient Convolutional Neural Network on an Implantable Microcontroller
• Manifold-regression to predict from MEG/EEG brain signals without source modeling
• A large-scale evaluation framework for EEG deep learning architectures
• Data Augmentation for Brain-Computer Interfaces: Analysis on Event-Related Potentials Data
• Fast AutoAugment

Cited by in corpus (10)

• EEG-ITNet: An Explainable Inception Temporal Convolutional Network for Motor Imagery Classification
• Automatic detection of abnormal EEG signals using wavelet feature extraction and gradient boosting decision tree
• MP-SeizNet: A Multi-Path CNN Bi-LSTM Network for Seizure-Type Classification Using EEG
• Amplifying Pathological Detection in EEG Signaling Pathways through Cross-Dataset Transfer Learning
• Multi-center validation study of automated classification of pathological slowing in adult scalp electroencephalograms via frequency features
• Data augmentation for learning predictive models on EEG: a systematic comparison
• Evaluating the structure of cognitive tasks with transfer learning
• Scope and Arbitration in Machine Learning Clinical EEG Classification
• A dual-system approach for epilepsy diagnosis: integrating mamba-Bi-LSTM architecture with SHAP-based verification
• Hierarchical Variable Importance with Statistical Control for Medical Data-Based Prediction