On-Device Learning with Raspberry Pi for GCN-Based Epilepsy EEG Classification

Epilepsy is a chronic brain disease characterized by recurrent and transient seizures, which is accompanied by super-synchronous abnormal discharge of electroencephalogram (EEG) signals. As a non-invasive auxiliary diagnostic technique, EEG is currently important means of seizure detection. However, due to ignoring the spatial topological relationship between electrodes, existing data-driver methods fail to fully reflect the interaction between signals. Meanwhile, their models usually be designed with a large number of redundant parameters, making it difficult to deploy to micro-embedded devices with limited-resources. In this paper, we propose a on-device learning with edge device for epilepsy EEG classification network based on GCN (oDLGCN-EEG). Specifically, to analyze the spatial relationships between various electrodes and their temporal dependencies, we design a Brain Topology Network (BTN) for the spatiotemporal dependency feature map construction. To capture the internal activity during epilepsy seizures, we design a Neural Feature Extraction Module (NFEM) for the neural activity feature map construction. Besides, we propose a pruning scheme to optimize the model, which successfully deploys the optimized oDLGCN-EEG on the embedded Raspberry Pi device for efficient and low-power consumption intelligent epilepsy classification. Experiments in comparison with state-of-the-art methods show that oDLGCN-EEG achieves the best classification accuracy and with the smallest parameter number on baseline EEG dataset. The code is available at https://github.com/cathnat/Epileptic_Classification.