An Epileptic EEG Detection Method Based on Data Augmentation and Lightweight Neural Network

IEEE J Transl Eng Health Med. 2023 Aug 24:12:22-31. doi: 10.1109/JTEHM.2023.3308196. eCollection 2024.

Abstract

Objective: Epilepsy, an enduring neurological disorder, afflicts approximately 65 million individuals globally, significantly impacting their physical and mental wellbeing. Traditional epilepsy detection methods are labor-intensive, leading to inefficiencies. Although deep learning techniques for brain signal detection have gained traction in recent years, their clinical application advancement is hindered by the significant requirement for high-quality data and computational resources during training.

Methods & results: The neural network training initially involved merging two datasets of different data quality, namely Bonn University datasets and CHB-MIT datasets, to bolster its generalization capabilities. To tackle the issues of dataset size and class imbalance, we employed small window segmentation and Synthetic Minority Over-sampling Technique (SMOTE). algorithms to augment and equalize the data. A streamlined neural network architecture was then proposed, drastically reducing the model's training parameters. Notably, a model trained with a mere 9,371 parameters yielded impressive results. The three-classification task on the combined dataset delivered an accuracy of 98.52%, sensitivity of 97.99%, specificity of 99.35%, and precision of 98.44%.

Conclusion: The experimental findings of this study underscore the superiority of the proposed method over existing approaches in both model size reduction and accuracy enhancement. As a result, it is more apt for deployment in low-cost, low computational hardware devices, including wearable technology, and various clinical applications. Clinical and Translational Impact Statement- This study is a Pre-Clinical Research. The lightweight neural network is easily deployed on hardware device for real-time epileptic EEG detection.

Keywords: Electroencephalography; data augmentation; deep learning; epilepsy detection; lightweight neural networks..

MeSH terms

  • Algorithms
  • Brain
  • Electroencephalography / methods
  • Epilepsy* / diagnosis
  • Humans
  • Neural Networks, Computer