Trustworthy Sleep Staging From Eeg: Deep Ensembles, Mcdropout, And Predictive Calibration
BC, S.; Vrajitoru, D.
Show abstract
AO_SCPLOWBSTRACTC_SCPLOWReliable sleep stage classification from EEG signals is critical for the development of clinical decision support systems. However, many deep learning models lack mechanisms for estimating predictive uncertainty, which is important for trust and interpretability. This work explores the use of Monte Carlo Dropout and Deep Ensembles to estimate uncertainty in automatic sleep staging. We apply these methods to convolutional and recurrent neural architectures trained on a publicly available EEG dataset. Evaluation includes calibration analysis and visualization of predictive entropy across sleep stages. LSTM-based models demonstrated more consistent trustworthiness and calibration across runs, while EEGNet remained appealing for lightweight deployment.The results highlight the value of incorporating uncertainty estimates to improve model transparency and support safe use in medical contexts.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- An infection prediction model developed from inpatient data can predict out-of-hospital COVID-19 infections from wearable data when controlled for dataset shift 93%
- Event Driven Neural Network on a Mixed Signal Neuromorphic Processor for EEG Based Epileptic Seizure Detection 92%
- Recurrent Neural Network-based Acute Concussion Classifier using Raw Resting State EEG Data 92%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.