A Class-Imbalance Aware and Explainable Spatio-Temporal Graph Attention Network for Neonatal Seizure Detection.

Autor: Raeisi K; Department of Neuroscience, Imaging and Clinical Sciences, Universita Gabriele d'Annunzio, Chieti 66100, Italy., Khazaei M; Department of Neuroscience, Imaging and Clinical Sciences, Universita Gabriele d'Annunzio, Chieti 66100, Italy., Tamburro G; Department of Neuroscience, Imaging and Clinical Sciences-Behavioral Imaging and Neural Dynamics Center, Universita Gabriele d'Annunzio, Chieti 66100, Italy., Croce P; Department of Neuroscience, Imaging and Clinical Sciences-Behavioral Imaging and Neural Dynamics Center, Universita Gabriele d'Annunzio, Chieti 66100, Italy., Comani S; Department of Neuroscience, Imaging and Clinical Sciences-Behavioral Imaging and Neural Dynamics Center, Universita Gabriele d'Annunzio, Chieti 66100, Italy., Zappasodi F; Department of Neuroscience, Imaging and Clinical Sciences-Behavioral, Imaging and Neural Dynamics Center-Institute for, Advanced Biomedical Technologies, Universita Gabriele d'Annunzio, Chieti 66100, Italy.
Jazyk: angličtina
Zdroj: International journal of neural systems [Int J Neural Syst] 2023 Sep; Vol. 33 (9), pp. 2350046. Date of Electronic Publication: 2023 Jul 28.
DOI: 10.1142/S0129065723500466
Abstrakt: Seizures are the most prevalent clinical indication of neurological disorders in neonates. In this study, a class-imbalance aware and explainable deep learning approach based on Convolutional Neural Networks (CNNs) and Graph Attention Networks (GATs) is proposed for the accurate automated detection of neonatal seizures. The proposed model integrates the temporal information of EEG signals with the spatial information on the EEG channels through the graph representation of the multi-channel EEG segments. One-dimensional CNNs are used to automatically develop a feature set that accurately represents the differences between seizure and nonseizure epochs in the time domain. By employing GAT, the attention mechanism is utilized to emphasize the critical channel pairs and information flow among brain regions. GAT coefficients were then used to empirically visualize the important regions during the seizure and nonseizure epochs, which can provide valuable insight into the location of seizures in the neonatal brain. Additionally, to tackle the severe class imbalance in the neonatal seizure dataset using under-sampling and focal loss techniques are used. Overall, the final Spatio-Temporal Graph Attention Network (ST-GAT) outperformed previous benchmarked methods with a mean AUC of 96.6% and Kappa of 0.88, demonstrating its high accuracy and potential for clinical applications.
Databáze: MEDLINE