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

Abstract

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. 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%. 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.