The final, formatted version of the article will be published soon.
ORIGINAL RESEARCH article
Front. Psychiatry
Sec. Digital Mental Health
Volume 15 - 2024 |
doi: 10.3389/fpsyt.2024.1504190
Machine Learning Prediction of Anxiety Symptoms in Social Anxiety Disorder: Utilizing Multimodal Data from Virtual Reality Sessions
Provisionally accepted- 1 Department of Biomedical Informatics, College of Medicine, Korea University, Seoul, Republic of Korea
- 2 Department of Psychiatry, College of Medicine, Korea University, Seoul, Republic of Korea
- 3 Graduate School of Health Science and Technology, Department of Biomedical Engineering, Ulsan National Institute of Science and Technology, Eonyang, Ulsan, Republic of Korea
- 4 School of Psychiatry, Korea University, Seoul, Republic of Korea
- 5 Department of Biotechnology and Bioinformatics, Korea University, Sejong, Republic of Korea
- 6 Department of Biomedical Informatics, Korea University, Seoul, Republic of Korea
Introduction: Machine learning (ML) is an effective tool for predicting mental states and is a key technology in digital psychiatry. This study aimed to develop ML algorithms to predict the upper tertile group of various anxiety symptoms based on multimodal data from virtual reality (VR) therapy sessions for social anxiety disorder (SAD) patients and to evaluate their predictive performance across each data type.Methods: This study included 32 SAD-diagnosed individuals, and finalized a dataset of 132 samples from 25 participants. It utilized multimodal (physiological and acoustic) data from VR sessions to simulate social anxiety scenarios. This study employed extended Geneva minimalistic acoustic parameter set for acoustic feature extraction and extracted statistical attributes from time series-based physiological responses. We developed ML models that predict the upper tertile group for various anxiety symptoms in SAD using Random Forest, extreme gradient boosting (XGBoost), light gradient boosting machine (LightGBM), and categorical boosting (CatBoost) models. The best parameters were explored through grid search or random search, and the models were validated using stratified crossvalidation and leave-one-out cross-validation.The CatBoost, using multimodal features, exhibited high performance, particularly for the Social Phobia Scale with an area under the receiver operating characteristics curve (AUROC) of 0.852. It also showed strong performance in predicting cognitive symptoms, with the highest AUROC of 0.866 for the Post-Event Rumination Scale. For generalized anxiety, the LightGBM's prediction for the State-Trait Anxiety Inventory-trait led to an AUROC of 0.819. In the same analysis, models using only physiological features had AUROCs of 0.626, 0.744, and 0.671, whereas models using only acoustic features had AUROCs of 0.788, 0.823, and 0.754.This study showed that a ML algorithm using integrated multimodal data can predict upper tertile anxiety symptoms in patients with SAD with higher performance than acoustic or physiological data obtained during a VR session. The results of this study can be used as evidence for personalized VR sessions and to demonstrate the strength of the clinical use of multimodal data.
Keywords: machine learning, Multimodal data, Digital phenotyping, Digital psychiatry, Social Anxiety Disorder, Virtual reality intervention, anxiety prediction
Received: 30 Sep 2024; Accepted: 09 Dec 2024.
Copyright: © 2024 Park, Shin, Jung, Hur, Pil Pack, Lee, Lee and Cho. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.
* Correspondence:
Hwamin Lee, Department of Biomedical Informatics, Korea University, Seoul, 136-701, Republic of Korea
Chul-Hyun Cho, Department of Psychiatry, College of Medicine, Korea University, Seoul, Republic of Korea
Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.