Raurale SA, Boylan GB, Mathieson S, Marnane WP, Lightbody G, O'Toole JM. Grading hypoxic-ischemic encephalopathy in neonatal EEG with convolutional neural networks and quadratic time-frequency distributions.
J Neural Eng 2021;
18. [PMID:
33618337 PMCID:
PMC8208632 DOI:
10.1088/1741-2552/abe8ae]
[Citation(s) in RCA: 13] [Impact Index Per Article: 4.3] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 10/12/2020] [Accepted: 02/22/2021] [Indexed: 12/12/2022]
Abstract
OBJECTIVE
To develop an automated system to classify the severity of hypoxic-ischaemic encephalopathy injury (HIE) in neonates from the background electroencephalogram (EEG).
METHOD
By combining a quadratic time{frequency distribution (TFD) with a convolutional neural network, we develop a system that classifies 4 EEG grades of HIE. The network learns directly from the two- dimensional TFD through 3 independent layers with convolution in the time, frequency, and time{frequency directions. Computationally efficient algorithms make it feasible to transform each 5 minute epoch to the time-frequency domain by controlling for oversampling to reduce both computation and computer memory. The system is developed on EEG recordings from 54 neonates. Then the system is validated on a large unseen dataset of 338 hours of EEG recordings from 91 neonates obtained across multiple international centres.
RESULTS
The proposed EEG HIE-grading system achieves a leave-one-subject-out testing accuracy of 88.9% and kappa of 0.84 on the development dataset. Accuracy for the large unseen test dataset is 69.5% (95% confidence interval, CI: 65.3 to 73.6%) and kappa of 0.54, which is a significant (P < 0.001) improvement over a state-of-the-art feature-based method with an accuracy of 56.8% (95% CI: 51.4 to 61.7%) and kappa of 0.39. Performance of the proposed system was unaffected when the number of channels in testing was reduced from 8 to 2|accuracy for large validation dataset remained at 69.5% (95% CI: 65.5 to 74.0%).
SIGNIFICANCE
The proposed system outperforms the state-of-the-art machine learning algorithms for EEG grade classification on a large multi-centre unseen dataset, indicating the potential to assist clinical decision making for neonates with HIE.
Collapse