Two-Stream Attention 3-D Deep Network-Based Childhood Epilepsy Syndrome Classification

Abstract

Epilepsy syndromes are typical childhood nervous system diseases, usually containing several different seizure types commonly seen. There exist around 20 known childhood epilepsy syndromes and accurate classifying these epilepsy syndromes plays a vital guidance to diagnosis and treatment. Current research mainly focused on seizure detection, while few have been presented on epilepsy syndrome classification. In this article, a novel two-stream 3-D attention module-based deep network (TSA3-D) is developed to exploit multichannel time-frequency and frequency-space features of interictal EEGs for epilepsy syndrome classification. Two multichannel montage transforms are applied for EEG feature optimization to reduce artifact affection. In TSA3-D, both the channelwise attention (CA) and dual spatial attention (DSA)-based 3-D attention modules are developed to enhance the EEG feature learning capability. The study is conducted on the long-term interictal EEGs of 115 subjects from the Children’s Hospital, Zhejiang University School of Medicine (CHZU). These data belong to seven typical childhood epilepsy syndromes and one normal control group. Experimental results show that TSA3-D can achieve an overall accuracy of 99.52%, the highest among when compared with the state-of-the-art (SOTA) methods.