An Epileptic EEG Detection Method Based on Data Augmentation and Lightweight Neural Network
C.Y. Wang, Lei Liu, Wenhai Zhuo, Yun Xie
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.