A Novel Morlet Convolutional Neural Network

Automatic seizure detection holds significant importance for epilepsy diagnosis and treatment. Convolutional neural networks (CNNs) have shown immense potential in seizure detection. Though traditional CNN-based seizure detection models have achieved significant advancements, they often suffer from excessive parameters and limited interpretability, thus hindering their reliability and practical deployment on edge computing devices. Therefore, this study introduces an innovative Morlet convolutional neural network (Morlet-CNN) framework with its effectiveness demonstrated in seizure detection tasks. Unlike traditional CNNs, the convolutional kernels in the Morlet-CNN contain only two learnable parameters, allowing for a lightweight architecture. Additionally, we propose a frequency-domain-response-based kernel pruning algorithm for Morlet-CNN and implement an INT8 quantization algorithm by incorporating Kullback-Leibler (KL) divergence calibration with a Morlet lookup table (LUT). With the pruning and quantization algorithms, the model's parameter scale achieves over 90% reduction while maintaining minimal accuracy loss. Furthermore, the model exhibits enhanced interpretability from a signal processing perspective, distinguishing it from many previous CNN models. Extensive experimental validation on the Bonn and CHB-MIT datasets confirms the Morlet-CNN model's efficacy with a compact Kilobyte (KB)-level model size, making it highly suitable for real-world applications.