This study explores data science and machine learning techniques to predict suicidal ideation in young individuals, aiming to identify the most effective subset of questions from a comprehensive questionnaire. We benchmarked various machine learning models, including ElasticNet, Ridge Regression, Support Vector Regressor (SVR), Gradient Boosting Regressor, Random Forest Regressor, and XGBoost Regressor. Ridge Regression emerged as the most suitable model, achieving a Mean Squared Error (MSE) of 146.77, Mean Absolute Error (MAE) of 8.68, and an R \(^{2}\) of 0.57. Utilizing SHAP (SHapley Additive exPlanations) analysis, we identified the 20 most influential questions from the dataset. This refined approach not only enhances the efficiency of the questionnaire but also aids targeted intervention strategies by facilitating the early detection of suicidal ideation. The findings provide mental health professionals with a streamlined and effective tool to assess and address suicidal tendencies among students and young people, thereby improving preventative measures and outcomes.

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

Improving Suicide Ideation Screening with Machine Learning and Questionnaire Optimization Through Feature Analysis

  • Ignacio Martínez,
  • César Astudillo,
  • Daniel Núñez

摘要

This study explores data science and machine learning techniques to predict suicidal ideation in young individuals, aiming to identify the most effective subset of questions from a comprehensive questionnaire. We benchmarked various machine learning models, including ElasticNet, Ridge Regression, Support Vector Regressor (SVR), Gradient Boosting Regressor, Random Forest Regressor, and XGBoost Regressor. Ridge Regression emerged as the most suitable model, achieving a Mean Squared Error (MSE) of 146.77, Mean Absolute Error (MAE) of 8.68, and an R \(^{2}\) of 0.57. Utilizing SHAP (SHapley Additive exPlanations) analysis, we identified the 20 most influential questions from the dataset. This refined approach not only enhances the efficiency of the questionnaire but also aids targeted intervention strategies by facilitating the early detection of suicidal ideation. The findings provide mental health professionals with a streamlined and effective tool to assess and address suicidal tendencies among students and young people, thereby improving preventative measures and outcomes.