Tiny CNN for Seizure Prediction in Wearable Biomedical Devices

Epilepsy is a life-threatening disease affecting millions of people all over the world. Artificial intelligence epileptic predictors offer excellent potential to improve epilepsy therapy. Particularly, deep learning models such as convolutional neural networks (CNN) can be used to accurately detect...

Full description

Saved in:
Bibliographic Details
Main Authors: Zhang, Yang, Savaria, Yvon, Zhao, Shiqi, Mordido, Goncalo, Sawan, Mohamad, Leduc-Primeau, Francois
Format: Conference Proceeding
Language:English
Subjects:
Adaptation models
Biological system modeling
Deep learning
Epilepsy
Performance evaluation
Quantization (signal)
Sensitivity
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Epilepsy is a life-threatening disease affecting millions of people all over the world. Artificial intelligence epileptic predictors offer excellent potential to improve epilepsy therapy. Particularly, deep learning models such as convolutional neural networks (CNN) can be used to accurately detect ictogenesis through deep structured learning representations. In this work, a tiny one-dimensional stacked convolutional neural network (1DSCNN) is proposed based on short-time Fourier transform (STFT) to predict epileptic seizure. The results demonstrate that the proposed method obtains better performance compared to recent state-of-the-art methods, achieving an average sensitivity of 94.44%, average false prediction rate (FPR) of 0.011/h and average area under the curve (AUC) of 0.979 on the test set of the American Epilepsy Society Seizure Prediction Challenge dataset, while featuring a model size of only 21.32kB. Furthermore, after adapting the model to 4-bit quantization, its size is significantly decreased by 7.08x with only 0.51% AUC score precision loss, which shows excellent potential for hardware-friendly wearable implementation.
ISSN:2694-0604
DOI:10.1109/EMBC48229.2022.9872006