arXiv:2411.10661cs.LGcs.CV2024-11被引 4

用集成模型提升灾后创伤后应激障碍预测准确率

Enhancing PTSD Outcome Prediction with Ensemble Models in Disaster Contexts

  • 融合六类模型的投票机制,提升预测鲁棒性
  • 在基准数据集上达到96.76%准确率,显著优于单个模型
  • 适合灾害心理干预决策者与医疗人员参考使用

创伤后应激障碍(PTSD)是暴露于创伤事件后的重要心理健康挑战。早期检测与有效干预至关重要,否则可能导致长期心理困扰。在灾难影响人群中,准确识别PTSD对及时开展针对性心理干预尤为关键。现有机器学习方法虽用于分类诊断,但普遍存在性能与泛化能力不足的问题。为此,本文构建了完整的预处理流程:包括数据清洗、缺失值处理(SimpleImputer)、类别变量标签编码、通过SMOTE进行数据增强以平衡样本分布,并采用StandardScaler进行特征缩放。数据按80%训练、20%测试划分。设计了基于多数投票的集成模型,结合逻辑回归、支持向量机(SVM)、随机森林、XGBoost、LightGBM及定制人工神经网络(ANN)。该集成模型在基准数据集上实现96.76%的准确率,显著优于单一模型。该方法优势在于多模型融合带来的更强鲁棒性、更优的跨数据点泛化能力以及更高的检测精度。同时,利用SMOTE增强数据缓解类别不平衡问题,提升预测可靠性。研究为政策制定者与医疗提供者在脆弱群体中应用预测分析应对心理危机提供了重要参考。

原文摘要 · Abstract (English)

Post-traumatic stress disorder (PTSD) is a significant mental health challenge that affects individuals exposed to traumatic events. Early detection and effective intervention for PTSD are crucial, as it can lead to long-term psychological distress if untreated. Accurate detection of PTSD is essential for timely and targeted mental health interventions, especially in disaster-affected populations. Existing research has explored machine learning approaches for classifying PTSD, but many face limitations in terms of model performance and generalizability. To address these issues, we implemented a comprehensive preprocessing pipeline. This included data cleaning, missing value treatment using the SimpleImputer, label encoding of categorical variables, data augmentation using SMOTE to balance the dataset, and feature scaling with StandardScaler. The dataset was split into 80\% training and 20\% testing. We developed an ensemble model using a majority voting technique among several classifiers, including Logistic Regression, Support Vector Machines (SVM), Random Forest, XGBoost, LightGBM, and a customized Artificial Neural Network (ANN). The ensemble model achieved an accuracy of 96.76\% with a benchmark dataset, significantly outperforming individual models. The proposed method's advantages include improved robustness through the combination of multiple models, enhanced ability to generalize across diverse data points, and increased accuracy in detecting PTSD. Additionally, the use of SMOTE for data augmentation ensured better handling of imbalanced datasets, leading to more reliable predictions. The proposed approach offers valuable insights for policymakers and healthcare providers by leveraging predictive analytics to address mental health issues in vulnerable populations, particularly those affected by disasters.

PTSD预测集成学习灾后心理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。