Enhancing Electroencephalogram-Based Prediction of Posttraumatic Stress Disorder Treatment Response Using Data Augmentation

Abstract

Objective This study aimed to improve the prediction of treatment response in patients with posttraumatic stress disorder (PTSD) by applying a variational autoencoder (VAE)-based data augmentation (DA) approach to electroencephalogram (EEG) data.Methods EEG spectrograms were collected from patients diagnosed with PTSD. A VAE model was pretrained on the original spectrograms and used to generate augmented data samples. These augmented spectrograms were then utilized to train a deep neural network (DNN) classifier. The performance of the model was evaluated by comparing the area under the receiver operating characteristic curve (AUC) between models trained with and without DA.Results The DNN trained with VAE-augmented EEG data achieved an AUC of 0.85 in predicting treatment response, which was 0.11 higher than the model trained without augmentation. This reflects a significant improvement in classification performance and model generalization.Conclusion VAE-based DA effectively addresses the challenge of limited EEG data in clinical settings and enhances the performance of DNN models for treatment response prediction in PTSD. This approach presents a promising direction for future EEG-based neuropsychiatric research involving small datasets.