Energy-Efficient Neural Network for Epileptic Seizure Prediction
Shiqi Zhao, Jie Yang, Mohamad Sawan
Abstract
Seizure prediction for drug-refractory epilepsy patients can improve their quality of life, reduce their anxiety, and help them take the necessary precautions. Nowadays, numerous deep learning algorithms have been proposed to predict seizure onset and obtain better performance than traditional machine learning methods. However, these methods require a large set of parameters and large hardware resources; they also have high energy consumption. Therefore, these methods cannot be implemented on compact, low-power wearable, or implantable medical devices. The devices should operate on a real-time basis to continually inform the epileptic patients. In this paper, we describe energy-efficient and hardware-friendly methods to predict the epileptic seizures. A model of only 45 kB was obtained by the neural architecture search and was evaluated across three datasets. The overall sensitivity, false prediction rate, and area under receiver operating characteristic curve were 99.81%, 0.005/h, 1 and 93.48%, 0.063/h, 0.977 and 85.19%, 0.116/h, 0.933, respectively, for three public datasets. This model was further reduced with several methods for model compression. The performances for the model sizes less than 50 kB for scalp EEG data and less than 10 kB for intracranial EEG data outperformed all the other models of similar model sizes. In particular, the energy consumption estimation was less than 10 mJ per inference for scalp EEG signal and less than 0.5 mJ per inference for intracranial EEG signal, which meet the requirements for low-power wearable and implantable devices, respectively.