Abstract
AbstractMultimodal classification is increasingly common in biomedical informatics studies. Many such studies use deep learning classifiers with raw data, which makes explainability difficult. As such, only a few studies have applied explainability methods, and new methods are needed. In this study, we propose sleep stage classification as a testbed for method development and train a convolutional neural network with electroencephalogram (EEG), electrooculogram, and electromyogram data. We then present a global approach that is uniquely adapted for electrophysiology analysis. We further present two local approaches that can identify subject-level differences in explanations that would be obscured by global methods and that can provide insight into the effects of clinical and demographic variables upon the patterns learned by the classifier. We find that EEG is globally the most important modality for all sleep stages, except non-rapid eye movement stage 1 and that local subject-level differences in importance arise. We further show that sex, followed by medication and age had significant effects upon the patterns learned by the classifier. Our novel methods enhance explainability for the growing field of multimodal classification, provide avenues for the advancement of personalized medicine, and yield novel insights into the effects of demographic and clinical variables upon classifiers.
Publisher
Cold Spring Harbor Laboratory
Reference81 articles.
1. Zhai B , Perez-Pozuelo I , Clifton EAD , Palotti J , Guan Y. Making Sense of Sleep: Multimodal Sleep Stage Classification in a Large, Diverse Population Using Movement and Cardiac Sensing. Proc ACM Interactive, Mobile, Wearable Ubiquitous Technol. 2020;4(2).
2. Lin J , Pan S , Lee CS , Oviatt S. An Explainable Deep Fusion Network for Affect Recognition Using Physiological Signals. In: Proceedings of the 28th ACM International Conference on Information and Knowledge Management. 2019. p. 2069–72.
3. Machine Learning Models Identify Multimodal Measurements Highly Predictive of Transdiagnostic Symptom Severity for Mood, Anhedonia, and Anxiety;Biol Psychiatry Cogn Neurosci Neuroimaging,2020
4. Wang IN , Lee CH , Kim HJ , Kim H , Kim DJ . An Ensemble Deep Learning Approach for Sleep Stage Classification via Single-channel EEG and EOG. Int Conf ICT Converg. 2020;2020-Octob:394–8.
5. Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification;IEEE Trans Biomed Eng,2019