A Deep Model for EEG Seizure Detection with Explainable AI using Connectivity Features

Abstract

During seizures, different types of communication between different parts of the brain are characterized by many state of the art connectivity measures. We propose to employ a set of undirected (spectral matrix, the inverse of the spectral matrix, coherence, partial coherence, and phase-locking value) and directed features (directed coherence, the partial directed coherence) to detect seizures using a deep neural network. Taking our data as a sequence of ten sub-windows, an optimal deep sequence learning architecture using attention, CNN, BiLstm, and fully connected neural networks is designed to output the detection label and the relevance of the features. The relevance is computed using the weights of the model in the activation values of the receptive fields at a particular layer. The best model resulted in 97.03% accuracy using balanced MIT-BIH data subset. Finally, an analysis of the relevance of the features is reported.