A Novel Morlet Convolutional Neural Network.

Journal: International journal of neural systems
Published Date:

Abstract

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.

Authors

  • Peilin Zhu
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Zirong Li
    Department of Orthopedics, Beijing Key Laboratory for Immune-Mediated Inflammatory Diseases, China-Japan Friendship Hospital, Peking Union Medical College, Beijing 100029, China.
  • Chao Cao
    Department of Computer Science and Engineering, The University of Texas at Arlington, Arlington, TX, United States.
  • Zhida Shang
    Shenzhen Institute of Shandong University, Shenzhen 518057, P. R. China.
  • Guoyang Liu
    School of Microelectronics, Shandong University, Jinan 250100, P. R. China.
  • Weidong Zhou

Keywords

No keywords available for this article.