Hostname: page-component-74656f9884-j7cmm Total loading time: 0 Render date: 2026-11-07T10:35:35.241Z Has data issue: false hasContentIssue false

Deep learning-based classification of posttraumatic stress disorder and depression following trauma utilizing visual and auditory markers of arousal and mood

Published online by Cambridge University Press:  03 August 2020

Katharina Schultebraucks*
Affiliation:
Department of Emergency Medicine, Vagelos School of Physicians and Surgeons, Columbia University Irving Medical Center, New York, New York, USA Department of Psychiatry, New York University Grossman School of Medicine, New York, New York, USA Data Science Institute, Columbia University, New York, New York, USA
Vijay Yadav
Affiliation:
AiCure, New York, New York, USA
Arieh Y. Shalev
Affiliation:
Department of Psychiatry, New York University Grossman School of Medicine, New York, New York, USA
George A. Bonanno
Affiliation:
Department of Counseling and Clinical Psychology, Teachers College, Columbia University, New York, New York, USA
Isaac R. Galatzer-Levy
Affiliation:
Department of Psychiatry, New York University Grossman School of Medicine, New York, New York, USA AiCure, New York, New York, USA
*
Author for correspondence: Katharina Schultebraucks, E-mail: ks3796@cumc.columbia.edu
Rights & Permissions [Opens in a new window]

Abstract

Background

Visual and auditory signs of patient functioning have long been used for clinical diagnosis, treatment selection, and prognosis. Direct measurement and quantification of these signals can aim to improve the consistency, sensitivity, and scalability of clinical assessment. Currently, we investigate if machine learning-based computer vision (CV), semantic, and acoustic analysis can capture clinical features from free speech responses to a brief interview 1 month post-trauma that accurately classify major depressive disorder (MDD) and posttraumatic stress disorder (PTSD).

Methods

N = 81 patients admitted to an emergency department (ED) of a Level-1 Trauma Unit following a life-threatening traumatic event participated in an open-ended qualitative interview with a para-professional about their experience 1 month following admission. A deep neural network was utilized to extract facial features of emotion and their intensity, movement parameters, speech prosody, and natural language content. These features were utilized as inputs to classify PTSD and MDD cross-sectionally.

Results

Both video- and audio-based markers contributed to good discriminatory classification accuracy. The algorithm discriminates PTSD status at 1 month after ED admission with an AUC of 0.90 (weighted average precision = 0.83, recall = 0.84, and f1-score = 0.83) as well as depression status at 1 month after ED admission with an AUC of 0.86 (weighted average precision = 0.83, recall = 0.82, and f1-score = 0.82).

Conclusions

Direct clinical observation during post-trauma free speech using deep learning identifies digital markers that can be utilized to classify MDD and PTSD status.

Information

Type
Original Article
Copyright
Copyright © The Author(s), 2020. Published by Cambridge University Press
Figure 0

Fig. 1. The pipeline of data analysis.

Figure 1

Table 1. Sample characteristics

Figure 2

Fig. 2. Receiver operating characteristic (ROC) curve of the internal test set for predicting (a) PCL-5 cut-off ⩾33 (AUC = 0.90) and (b) CES-D cut-off ⩾23 (AUC = 0.86).

Figure 3

Fig. 3. SHAP (SHapley Additive exPlanations) variable importance (Lundberg & Lee, 2017) of the Neural Network for the internal test set for predicting (a) PCL-5 cut-off ⩾33, (b) CES-D cut-off ⩾23, (c) PCL-5 symptom severity, and (d) CES-D symptom severity. The mean absolute SHAP value per feature is presented in the bar plot with larger bar plots displaying higher importance of the feature in discriminating between the ‘provisional PTSD diagnosis’ and ‘no PTSD’/‘provisional depression diagnosis’ and ‘no depression’. The variable importance based on SHAP values is calculated by evaluating the model performance with and without each feature included in the model in every possible order.

Figure 4

Fig. 4. SHAP (Lundberg & Lee, 2017) summary dot plot of the Neural Network for the internal test set for predicting (a) PCL-5 cut-off ⩾33 and (b) CES-D cut-off ⩾23, (c) PCL-5 symptom severity and (d) CES-D symptom severity. The higher the SHAP value of a feature, the higher the log odds of the ‘provisional PTSD diagnosis’/‘provisional depression diagnosis’. On the y-axis, the features are sorted by their general feature importance (see Fig. 4). The dots represent, for each variable value of each participant in the sample, how the variable value influences the attribution of the participant to one of the two outcome classes. Dots that are on the left side shift the classification of participants to the class ‘no PTSD/no MDD’, whereas dots on the right side of the x-axis shift the classification of participants to the class ‘PTSD/MDD’. The color represents the range of the feature values from low (blue) to high (red). For instance, the lower the score for the feature ‘positive emotion’ (NLP), the higher the odds for ‘provisional depression diagnosis’ (Fig. 4b).

Supplementary material: File

Schultebraucks et al. supplementary material

Schultebraucks et al. supplementary material

Download Schultebraucks et al. supplementary material(File)
File 69 KB