Integer convolutional neural network for seizure detection

Nhan Duy Truong, Anh Duy Nguyen, Levin Kuhlmann, Mohammad Reza Bonyadi, Jiawei Yang, Samuel Ippolito, Omid Kavehei

Research output: Contribution to journalArticleResearchpeer-review

26 Citations (Scopus)

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 data set specific. These issues greatly limit the applicability of such methods to hardware implementation, including in-silicon implementations such as application specific integrated circuits. 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 data sets consisting of intracranial and scalp electroencephalogram (EEG) signals. Integer-Net shows a consistent seizure detection performance across three data sets: Freiburg Hospital intracranial EEG data set, Children's Hospital of Boston-MIT scalp EEG data set, and UPenn and Mayo Clinic's seizure detection data set. Our experimental results show that a 4-bit Integer-Net leads to only 2% drop of accuracy compared with 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 the computational gain and reduce the inference time that are crucial for real-time application.

Original languageEnglish
Article number8370634
Pages (from-to)849-857
Number of pages9
JournalIEEE Journal on Emerging and Selected Topics in Circuits and Systems
Volume8
Issue number4
DOIs
Publication statusPublished - Dec 2018
Externally publishedYes

Keywords

  • Convolutional neural network
  • Integer convolution
  • Integer-Net
  • Machine learning
  • Seizure detection

Cite this