Integer Convolutional Neural Network for Seizure Detection

Abstract

Outstanding seizure detection algorithms have been developed over past two decades. Despite this success, their implementations as part of implantable or wearable devices are still limited. These works are mainly based on heavily handcrafted feature extraction, which is computationally expensive and is shown to be dataset specific. These issues greatly limit the applicability of such methods to hardware implementation, including in-silicon implementations such as application specific integrated circuits (ASIC). In this paper, we propose an integer convolutional neural network (CNN) implementation, Integer-Net, as a memory-efficient unified hardware-friendly CNN framework. The performance of Integer-Net is evaluated with multiple time-series datasets consisting of intracranial and scalp electroencephalogram (EEG) signals. Integer-Net shows a consistent seizure detection performance across three datasets: Freiburg Hospital intracranial EEG (iEEG) dataset, Children’s Hospital of Boston-MIT scalp EEG (sEEG) dataset, and UPenn and Mayo Clinic’s seizure detection dataset. Our experimental results show that a 4-bit Integer-Net leads to only 2% drop of accuracy compared to a 32-bit real-value resolution CNN model, while offering more than 7 times improvement in memory efficiency. We discuss the structure of the integer convolution to improve computational gain and reduce inference time that are crucial for real-time application.