학술논문

Ensemble Learning Using Individual Neonatal Data for Seizure Detection
Document Type
Periodical
Source
IEEE Journal of Translational Engineering in Health and Medicine IEEE J. Transl. Eng. Health Med. Translational Engineering in Health and Medicine, IEEE Journal of. 10:1-11 2022
Subject
Bioengineering
Communication, Networking and Broadcast Technologies
Components, Circuits, Devices and Systems
Computing and Processing
Signal Processing and Analysis
Robotics and Control Systems
General Topics for Engineers
Electroencephalography
Brain modeling
Pediatrics
Data models
Recording
Training
Predictive models
Convolutional neural network
distributed learning
ensemble learning
neonatal EEG
seizure detection algorithm
Language
ISSN
2168-2372
Abstract
Objective: Sharing medical data between institutions is difficult in practice due to data protection laws and official procedures within institutions. Therefore, most existing algorithms are trained on relatively small electroencephalogram (EEG) data sets which is likely to be detrimental to prediction accuracy. In this work, we simulate a case when the data can not be shared by splitting the publicly available data set into disjoint sets representing data in individual institutions. Methods and procedures: We propose to train a (local) detector in each institution and aggregate their individual predictions into one final prediction. Four aggregation schemes are compared, namely, the majority vote, the mean, the weighted mean and the Dawid-Skene method. The method was validated on an independent data set using only a subset of EEG channels. Results: The ensemble reaches accuracy comparable to a single detector trained on all the data when sufficient amount of data is available in each institution. Conclusion: The weighted mean aggregation scheme showed best performance, it was only marginally outperformed by the Dawid–Skene method when local detectors approach performance of a single detector trained on all available data. Clinical impact: Ensemble learning allows training of reliable algorithms for neonatal EEG analysis without a need to share the potentially sensitive EEG data between institutions.