SeizFt: Interpretable Machine Learning for Seizure Detection Using Wearables

被引:12
作者
Al-Hussaini, Irfan [1 ]
Mitchell, Cassie S. [2 ,3 ,4 ]
机构
[1] Georgia Inst Technol, Sch Elect & Comp Engn, Atlanta, GA 30332 USA
[2] Georgia Inst Technol, Dept Biomed Engn, Atlanta, GA 30332 USA
[3] Emory Univ, Atlanta, GA 30332 USA
[4] Georgia Inst Technol, Georgia Tech, Machine Learning Ctr, Atlanta, GA 30332 USA
来源
BIOENGINEERING-BASEL | 2023年 / 10卷 / 08期
基金
美国国家科学基金会;
关键词
seizure; EEG; augmentation; xai; interpretability; imbalanced classes; electroencephalogram; artificial intelligence; machine learning;
D O I
10.3390/bioengineering10080918
中图分类号
Q81 [生物工程学(生物技术)]; Q93 [微生物学];
学科分类号
071005 ; 0836 ; 090102 ; 100705 ;
摘要
This work presents SeizFt-a novel seizure detection framework that utilizes machine learning to automatically detect seizures using wearable SensorDot EEG data. Inspired by interpretable sleep staging, our novel approach employs a unique combination of data augmentation, meaningful feature extraction, and an ensemble of decision trees to improve resilience to variations in EEG and to increase the capacity to generalize to unseen data. Fourier Transform (FT) Surrogates were utilized to increase sample size and improve the class balance between labeled non-seizure and seizure epochs. To enhance model stability and accuracy, SeizFt utilizes an ensemble of decision trees through the CatBoost classifier to classify each second of EEG recording as seizure or non-seizure. The SeizIt1 dataset was used for training, and the SeizIt2 dataset for validation and testing. Model performance for seizure detection was evaluated using two primary metrics: sensitivity using the any-overlap method (OVLP) and False Alarm (FA) rate using epoch-based scoring (EPOCH). Notably, SeizFt placed first among an array of state-of-the-art seizure detection algorithms as part of the Seizure Detection Grand Challenge at the 2023 International Conference on Acoustics, Speech, and Signal Processing (ICASSP). SeizFt outperformed state-of-the-art black-box models in accurate seizure detection and minimized false alarms, obtaining a total score of 40.15, combining OVLP and EPOCH across two tasks and representing an improvement of similar to 30% from the next best approach. The interpretability of SeizFt is a key advantage, as it fosters trust and accountability among healthcare professionals. The most predictive seizure detection features extracted from SeizFt were: delta wave, interquartile range, standard deviation, total absolute power, theta wave, the ratio of delta to theta, binned entropy, Hjorth complexity, delta + theta, and Higuchi fractal dimension. In conclusion, the successful application of SeizFt to wearable SensorDot data suggests its potential for real-time, continuous monitoring to improve personalized medicine for epilepsy.
引用
收藏
页数:16
相关论文
共 66 条
[1]  
Abadi M., 2015, TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems
[2]   SERF: Interpretable Sleep Staging using Embeddings, Rules, and Features [J].
Al-Hussaini, Irfan ;
Mitchell, Cassie S. .
PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2022, 2022, :3791-3795
[3]  
Al-Hussaini I., 2022, P NEURIPS 2022 WORKS
[4]  
Al-Hussaini I, 2019, PR MACH LEARN RES, V106
[5]  
Al-Hussaini Irfan, 2023, Proc IEEE Int Conf Acoust Speech Signal Process, V2023, DOI 10.1109/icassp49357.2023.10097091
[6]   Trade-off between Gradual Set and On/Off Ratio in HfO x -Based Analog Memory with a Thin SiO x Barrier Layer [J].
Athena, Fabia F. ;
West, Matthew P. ;
Hah, Jinho ;
Graham, Samuel ;
Vogel, Eric M. .
ACS APPLIED ELECTRONIC MATERIALS, 2023, 5 (06) :3048-3058
[7]   Towards a better understanding of the forming and resistive switching behavior of Ti-doped HfOx RRAM [J].
Athena, Fabia F. ;
West, Matthew P. ;
Hah, Jinho ;
Hanus, Riley ;
Graham, Samuel ;
Vogel, Eric M. .
JOURNAL OF MATERIALS CHEMISTRY C, 2022, 10 (15) :5896-5904
[8]   Asymmetric Resistive Switching of Bilayer HfOx/AlOy and AlOy/HfOx Memristors: The Oxide Layer Characteristics and Performance Optimization for Digital Set and Analog Reset Switching [J].
Basnet, Pradip ;
Anderson, Erik C. ;
Athena, Fabia Farlin ;
Chakrabarti, Bhaswar ;
West, Matthew P. ;
Vogel, Eric M. .
ACS APPLIED ELECTRONIC MATERIALS, 2023, 5 (03) :1859-1865
[9]  
Bedeeuzzaman M., 2012, Int. J. Comput. Appl., V44, P1, DOI 10.5120/6304-8614
[10]   Seizure Diaries and Forecasting With Wearables: Epilepsy Monitoring Outside the Clinic [J].
Brinkmann, Benjamin H. ;
Karoly, Philippa J. ;
Nurse, Ewan S. ;
Dumanis, Sonya B. ;
Nasseri, Mona ;
Viana, Pedro F. ;
Schulze-Bonhage, Andreas ;
Freestone, Dean R. ;
Worrell, Greg ;
Richardson, Mark P. ;
Cook, Mark J. .
FRONTIERS IN NEUROLOGY, 2021, 12