Riemannian classifier enhances the accuracy of machine-learning-based diagnosis of PTSD using resting EEG

Prog Neuropsychopharmacol Biol Psychiatry. 2020 Aug 30:102:109960. doi: 10.1016/j.pnpbp.2020.109960. Epub 2020 May 3.

Abstract

Recently, objective and automated methods for the diagnosis of post-traumatic stress disorder (PTSD) have attracted increasing attention. However, previous studies on machine-learning-based diagnosis of PTSD with resting-state electroencephalogram (EEG) have reported poor accuracies of as low as 60%. Here, a Riemannian geometry-based classifier, the Fisher geodesic minimum distance to the mean (FgMDM), was employed for PTSD classification for the first time. Eyes-closed resting-state EEG data of 39 healthy individuals and 42 PTSD patients were used for the analysis. EEG source activities in 148 cortical regions were parcellated based on the Destrieux atlas, and their covariances were evaluated for each individual. Thirty epochs of preprocessed EEG were employed to calculate source activities. In addition, the FgMDM approach was applied to each EEG source covariance to construct the classifier. For a comparison, linear discriminant analysis (LDA), support vector machine (SVM), and random forest (RF) classifiers employing source band powers and network features as feature candidates were also tested. The FgMDM classifier showed an average classification accuracy of 75.240.80%. In contrast, the maximum accuracies of LDA, SVM, and RF classifiers were 66.54 ± 2.99%, 61.11 ± 2.98%, and 60.99 ± 2.19%, respectively. Our study demonstrated that the diagnostic accuracy of PTSD with resting-state EEG could be significantly improved by employing the FgMDM framework, which is a type of Riemannian geometry-based classifier.

Keywords: Classification; Computer-aided diagnosis; Machine learning; Post-traumatic stress disorder; Riemannian geometry.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Adult
  • Algorithms
  • Discriminant Analysis
  • Electroencephalography / methods*
  • Electroencephalography / statistics & numerical data
  • Female
  • Humans
  • Machine Learning
  • Male
  • Middle Aged
  • Reproducibility of Results
  • Rest
  • Stress Disorders, Post-Traumatic / classification
  • Stress Disorders, Post-Traumatic / diagnosis*
  • Support Vector Machine