Identifying posttraumatic stress disorder staging from clinical and sociodemographic features: a proof-of-concept study using a machine learning approach

•Machine learning algorithms can be used to develop predictive models in psychiatry•Machine learning models may be suitable to posttraumatic stress disorder staging•A model with four-class staging were proposed to predict PTSD staging This proof-of-concept study aimed to investigate the viability of a predictive model to support posttraumatic stress disorder (PTSD) staging. We performed a naturalistic, cross-sectional study at two Brazilian centers: the Psychological Trauma Research and Treatment (NET-Trauma) Program at Universidade Federal of Rio Grande do Sul, and the Program for Research and Care on Violence and PTSD (PROVE), at Universidade Federal of São Paulo. Five supervised machine-learning algorithms were tested: Elastic Net, Gradient Boosting Machine, Random Forest, Support Vector Machine, and C5.0, using clinical (Clinician-Administered PTSD Scale version 5) and sociodemographic features. A hundred and twelve patients were enrolled (61 from NET-Trauma and 51 from PROVE). We found a model with four classes suitable for the PTSD staging, with best performance metrics using the C5.0 algorithm to CAPS-5 15-items plus sociodemographic features, with an accuracy of 65.6% for the train dataset and 52.9% for the test dataset (both significant). The number of symptoms, CAPS-5 total score, global severity score, and presence of current/previous trauma events appear as main features to predict PTSD staging. This is the first study to evaluate staging in PTSD with machine learning algorithms using accessible clinical and sociodemographic features, which may be used in future research.