Research on epilepsy detection and recognition based on the combination of time frequency transform and deep learning model

Abstract

To improve the detection performance of epileptic electroencephalogram (EEG) signals and address their non-stationary characteristics,this paper compares the combined effects of continuous wavelet transform (CWT) and short-time Fourier transform (STFT) with three neural network models—EEGNet,AlexNet,and Shallow ConvNet—and incorporates targeted optimization designs. Specifically,Focal Loss,dynamic data augmentation,and an early stopping mechanism are introduced in the training phase to enhance model robustness. For EEGNet,optimizations are implemented by integrating a Squeeze-and-Excitation (SE) attention module,improving depthwise separable convolution,and dynamically adapting dimensions to reduce classification errors. For Shallow ConvNet,improvements include layered convolution for extracting “time-frequency” features and average pooling to adapt to long-duration data blocks. Experiments are conducted based on subject-independent validation,and the results show that the CWT-based feature extraction method outperforms STFT comprehensively. Among all combinations,the CWT+Shallow ConvNet pair exhibits the optimal overall performance,while the CWT+EEGNet combination follows closely with excellent precision. These findings verify the effectiveness of combining precise time-frequency features (extracted by CWT) with optimized neural network models,providing reliable technical support for clinical epileptic EEG signal detection.