Generalized Anxiety Disorder (GAD) is a mental disorder characterized by persistent and excessive worry about everyday situations. In the area of mental health, machine learning has proven useful for classifying psychological disorders based on sociodemographic, personality and physiological variables. This article presents a model for the classification of GAD based on the Random Forest algorithm, complemented with the SHAP (SHapley Additive exPlanations) explanatory technique. A total of 280 records were collected using four questionnaires that assessed life history strategies, sociodemographic characteristics, and levels of worry. After preprocessing, feature selection and class balancing, different models were assessed. Random Forest model combined with the ADASYN technique achieved the best results, with an accuracy of 82% and sensitivity of 80%. The analysis of variable importance using SHAP identified gender, impulsivity, visuospatial ability, resilience to adverse situations and the K factor as the most relevant factors. These findings support the feasibility of integrating life history variables into machine learning models for the detection of GAD and provide valuable information for its understanding from a developmental and clinical perspective.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Classification of Generalized Anxiety Through Life History Strategies with Random Forest and Explainable Machine Learning

  • Ana Luisa Islas-Avila,
  • Alicia Martinez-Rebollar,
  • Ricardo Castrejon-Salgado,
  • Laura Avila-Jimenez,
  • Hugo Estrada-Esquivel,
  • Angélica Fabiola Sanchez-Gutierrez

摘要

Generalized Anxiety Disorder (GAD) is a mental disorder characterized by persistent and excessive worry about everyday situations. In the area of mental health, machine learning has proven useful for classifying psychological disorders based on sociodemographic, personality and physiological variables. This article presents a model for the classification of GAD based on the Random Forest algorithm, complemented with the SHAP (SHapley Additive exPlanations) explanatory technique. A total of 280 records were collected using four questionnaires that assessed life history strategies, sociodemographic characteristics, and levels of worry. After preprocessing, feature selection and class balancing, different models were assessed. Random Forest model combined with the ADASYN technique achieved the best results, with an accuracy of 82% and sensitivity of 80%. The analysis of variable importance using SHAP identified gender, impulsivity, visuospatial ability, resilience to adverse situations and the K factor as the most relevant factors. These findings support the feasibility of integrating life history variables into machine learning models for the detection of GAD and provide valuable information for its understanding from a developmental and clinical perspective.