Generation of synthetic EEG data for training algorithms supporting the diagnosis of major depressive disorder

被引:12
作者
Carrle, Friedrich Philipp [1 ,2 ]
Hollenbenders, Yasmin [1 ,2 ]
Reichenbach, Alexandra [1 ,2 ]
机构
[1] Heilbronn Univ, Ctr Machine Learning, Heilbronn, Germany
[2] Heidelberg Univ, Med Fac Heidelberg, Heidelberg, Germany
关键词
major depressive disorder; electroencephalography; generative adversarial network; deep learning; data augmentation; synthetic data; biomarker; diagnosis; DATA AUGMENTATION;
D O I
10.3389/fnins.2023.1219133
中图分类号
Q189 [神经科学];
学科分类号
071006 ;
摘要
Introduction: Major depressive disorder (MDD) is the most common mental disorder worldwide, leading to impairment in quality and independence of life. Electroencephalography (EEG) biomarkers processed with machine learning (ML) algorithms have been explored for objective diagnoses with promising results. However, the generalizability of those models, a prerequisite for clinical application, is restricted by small datasets. One approach to train ML models with good generalizability is complementing the original with synthetic data produced by generative algorithms. Another advantage of synthetic data is the possibility of publishing the data for other researchers without risking patient data privacy. Synthetic EEG time-series have not yet been generated for two clinical populations like MDD patients and healthy controls.Methods: We first reviewed 27 studies presenting EEG data augmentation with generative algorithms for classification tasks, like diagnosis, for the possibilities and shortcomings of recent methods. The subsequent empirical study generated EEG time-series based on two public datasets with 30/28 and 24/29 subjects (MDD/controls). To obtain baseline diagnostic accuracies, convolutional neural networks (CNN) were trained with time-series from each dataset. The data were synthesized with generative adversarial networks (GAN) consisting of CNNs. We evaluated the synthetic data qualitatively and quantitatively and finally used it for re-training the diagnostic model.Results: The reviewed studies improved their classification accuracies by between 1 and 40% with the synthetic data. Our own diagnostic accuracy improved up to 10% for one dataset but not significantly for the other. We found a rich repertoire of generative models in the reviewed literature, solving various technical issues. A major shortcoming in the field is the lack of meaningful evaluation metrics for synthetic data. The few studies analyzing the data in the frequency domain, including our own, show that only some features can be produced truthfully.Discussion: The systematic review combined with our own investigation provides an overview of the available methods for generating EEG data for a classification task, their possibilities, and shortcomings. The approach is promising and the technical basis is set. For a broad application of these techniques in neuroscience research or clinical application, the methods need fine-tuning facilitated by domain expertise in (clinical) EEG research.
引用
收藏
页数:17
相关论文
共 91 条
[1]  
Abdelfattah SM, 2018, IEEE IJCNN
[2]  
Nguyen A, 2015, PROC CVPR IEEE, P427, DOI 10.1109/CVPR.2015.7298640
[3]  
[Anonymous], 2018, EEG-GAN: Generative adversarial networks for electroencephalographic (EEG) brain signals
[4]  
Arjovsky M, 2017, PR MACH LEARN RES, V70
[5]  
Bellman R., 1959, P IRE, V4, P1, DOI 10.1109/TAC.1959.1104847
[6]   Synthetic Biological Signals Machine-Generated by GPT-2 Improve the Classification of EEG and EMG Through Data Augmentation [J].
Bird, Jordan J. ;
Pritchard, Michael ;
Fratini, Antonio ;
Ekart, Aniko ;
Faria, Diego R. .
IEEE ROBOTICS AND AUTOMATION LETTERS, 2021, 6 (02) :3498-3504
[7]   The non-invasive Berlin Brain-Computer Interface:: Fast acquisition of effective performance in untrained subjects [J].
Blankertz, Benjamin ;
Dornhege, Guido ;
Krauledat, Matthias ;
Mueller, Klaus-Robert ;
Curio, Gabriel .
NEUROIMAGE, 2007, 37 (02) :539-550
[8]   Pros and cons of GAN evaluation measures [J].
Borji, Ali .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2019, 179 :41-65
[9]  
Bundesarztekammer (BAK) Kassenarztliche Bundesvereinigung (KBV) Arbeitsgemeinschaft der Wissenschaftlichen Medizinischen Fachgesellschaften (AWMF), 2022, Nationale VersorgungsLeitlinie Unipolare Depression-Langfassung,Version 3.2
[10]  
Cai H., 2020, ar Xiv preprint ar Xiv: 2002.09283