An Interpretable Machine Learning Method for the Detection of Schizophrenia Using EEG Signals

Thumbnail Image
Publication date
Defense date
Journal Title
Journal ISSN
Volume Title
Frontiers Media
Google Scholar
Research Projects
Organizational Units
Journal Issue
In this work we propose a machine learning (ML) method to aid in the diagnosis of schizophrenia using electroencephalograms (EEGs) as input data. The computational algorithm not only yields a proposal of diagnostic but, even more importantly, it provides additional information that admits clinical interpretation. It is based on an ML model called random forest that operates on connectivity metrics extracted from the EEG signals. Specifically, we use measures of generalized partial directed coherence (GPDC) and direct directed transfer function (dDTF) to construct the input features to the ML model. The latter allows the identification of the most performance-wise relevant features which, in turn, provide some insights about EEG signals and frequency bands that are associated with schizophrenia. Our preliminary results on real data show that signals associated with the occipital region seem to play a significant role in the diagnosis of the disease. Moreover, although every frequency band might yield useful information for the diagnosis, the beta and theta (frequency) bands provide features that are ultimately more relevant for the ML classifier that we have implemented.
Connectivity, Direct directed transfer function, Electroencephalography, Generalized partial directed coherence, Machine learning, Random forest, Schizophrenia
Bibliographic citation
Vázquez, M. A., Maghsoudi, A., & Mariño, I. P. (2021). An Interpretable Machine Learning Method for the Detection of Schizophrenia Using EEG Signals. Frontiers in Systems Neuroscience, 15.