Improving Clinician Performance in Classifying EEG Patterns on the Ictal–Interictal Injury Continuum Using Interpretable Machine Learning
Authors:
- Alina Jade BarnettABAlina Jade Barnett
- Zhicheng GuoZGZhicheng Guo
- Jin JingJJJin Jing
- Wendong GeWGWendong Ge
- Peter W. KaplanPKPeter W. Kaplan
- Wan Yee KongWKWan Yee Kong
- Ioannis KarakisIKIoannis Karakis
- Aline HerlopianAHAline Herlopian
- Lakshman Arcot JayagopalLJLakshman Arcot Jayagopal
- Olga TaraschenkoOTOlga Taraschenko
- Olga SelioutskiOSOlga Selioutski
- Gamaleldin OsmanGOGamaleldin Osman
- 23 May 2024
- journal article
- research article
- Published by Massachusetts Medical Society in NEJM AI
- Vol. 1 (6)
Abstract
In intensive care units (ICUs), critically ill patients are monitored with electroencephalography (EEG) to prevent serious brain injury. EEG monitoring is constrained by clinician availability, and EEG interpretation can be subjective and prone to interobserver variability. Automated deep-learning systems for EEG could reduce human bias and accelerate the diagnostic process. However, existing uninterpretable (black-box) deep-learning models are untrustworthy, difficult to troubleshoot, and lack accountability in real-world applications, leading to a lack of both trust and adoption by clinicians. We developed an interpretable deep-learning system that accurately classifies six patterns of potentially harmful EEG activity - seizure, lateralized periodic discharges (LPDs), generalized periodic discharges (GPDs), lateralized rhythmic delta activity (LRDA), generalized rhythmic delta activity (GRDA), and other patterns - while providing faithful case-based explanations of its predictions. The model was trained on 50,697 total 50-second continuous EEG samples collected from 2711 patients in the ICU between July 2006 and March 2020 at Massachusetts General Hospital. EEG samples were labeled as one of the six EEG patterns by 124 domain experts and trained annotators. To evaluate the model, we asked eight medical professionals with relevant backgrounds to classify 100 EEG samples into the six pattern categories - once with and once without artificial intelligence (AI) assistance - and we assessed the assistive power of this interpretable system by comparing the diagnostic accuracy of the two methods. The model's discriminatory performance was evaluated with area under the receiver-operating characteristic curve (AUROC) and area under the precision-recall curve. The model's interpretability was measured with task-specific neighborhood agreement statistics that interrogated the similarities of samples and features. In a separate analysis, the latent space of the neural network was visualized by using dimension reduction techniques to examine whether the ictal-interictal injury continuum hypothesis, which asserts that seizures and seizure-like patterns of brain activity lie along a spectrum, is supported by data. The performance of all users significantly improved when provided with AI assistance. Mean user diagnostic accuracy improved from 47 to 71% (P<0.04). The model achieved AUROCs of 0.87, 0.93, 0.96, 0.92, 0.93, and 0.80 for the classes seizure, LPD, GPD, LRDA, GRDA, and other patterns, respectively. This performance was significantly higher than that of a corresponding uninterpretable black-box model (with P<0.0001). Videos traversing the ictal-interictal injury manifold from dimension reduction (a two-dimensional representation of the original high-dimensional feature space) give insight into the layout of EEG patterns within the network's latent space and illuminate relationships between EEG patterns that were previously hypothesized but had not yet been shown explicitly. These results indicate that the ictal-interictal injury continuum hypothesis is supported by data. Users showed significant pattern classification accuracy improvement with the assistance of this interpretable deep-learning model. The interpretable design facilitates effective human-AI collaboration; this system may improve diagnosis and patient care in clinical settings. The model may also provide a better understanding of how EEG patterns relate to each other along the ictal-interictal injury continuum. (Funded by the National Science Foundation, National Institutes of Health, and others.).Keywords
This publication has 33 references indexed in Scilit:
- Effects of epileptiform activity on discharge outcome in critically ill patients in the USA: a retrospective cross-sectional studyThe Lancet Digital Health, 2023
- A deep learning framework for epileptic seizure detection based on neonatal EEG signalsScientific Reports, 2022
- A case-based interpretable deep learning model for classification of mass lesions in digital mammographyNature Machine Intelligence, 2021
- Automated Annotation of Epileptiform Burden and Its Association with OutcomesAnnals of Neurology, 2021
- Interpreting deep learning models for epileptic seizure detection on EEG signalsArtificial Intelligence in Medicine, 2021
- Deep active learning for interictal ictal injury continuum EEG patternsJournal of Neuroscience Methods, 2021
- Stop explaining black box machine learning models for high stakes decisions and use interpretable models insteadNature Machine Intelligence, 2019
- Deep learning for electroencephalogram (EEG) classification tasks: a reviewJournal of Neural Engineering, 2019
- The Ictal–Interictal Continuum: To Treat or Not to Treat (and How)?Neurocritical Care, 2017
- The standardization debate: A conflation trap in critical care electroencephalographySeizure: European Journal of Epilepsy, 2015