arXiv:2506.14459cs.LG2025-06被引 3

用集成学习提升职场抑郁预测准确率

A Model-Mediated Stacked Ensemble Approach for Depression Prediction Among Professionals

  • 多模型堆叠+逻辑回归融合,捕捉复杂心理因素
  • 测试集准确率达98.75%,各项指标超98%
  • 适合心理健康研究与早期干预系统开发

抑郁症是职业环境中重要的心理健康问题,工作压力、经济压力和生活方式失衡会加剧身心状况恶化。尽管关注度提升,但构建准确且可泛化的心理疾病预测模型仍面临挑战。传统分类方法难以应对抑郁症受多重相互关联因素影响的复杂性,如职业压力、睡眠模式和工作满意度。本研究提出一种基于堆叠的集成学习方法,以提升专业人士抑郁症分类的预测准确性。数据来自Kaggle的Depression Professional Dataset,包含人口统计、职业及生活方式等影响心理健康的特征。所提堆叠模型融合多个基学习器,并通过逻辑回归进行整合,有效捕捉多样化的学习模式。实验结果显示,该模型在训练集上达到99.64%的准确率,在测试集上为98.75%,精确率、召回率和F1分数均超过98%。结果表明集成学习在心理健康分析中具有显著效果,具备早期检测与干预的应用潜力。

原文摘要 · Abstract (English)

Depression is a significant mental health concern, particularly in professional environments where work-related stress, financial pressure, and lifestyle imbalances contribute to deteriorating well-being. Despite increasing awareness, researchers and practitioners face critical challenges in developing accurate and generalizable predictive models for mental health disorders. Traditional classification approaches often struggle with the complexity of depression, as it is influenced by multifaceted, interdependent factors, including occupational stress, sleep patterns, and job satisfaction. This study addresses these challenges by proposing a stacking-based ensemble learning approach to improve the predictive accuracy of depression classification among professionals. The Depression Professional Dataset has been collected from Kaggle. The dataset comprises demographic, occupational, and lifestyle attributes that influence mental well-being. Our stacking model integrates multiple base learners with a logistic regression-mediated model, effectively capturing diverse learning patterns. The experimental results demonstrate that the proposed model achieves high predictive performance, with an accuracy of 99.64% on training data and 98.75% on testing data, with precision, recall, and F1-score all exceeding 98%. These findings highlight the effectiveness of ensemble learning in mental health analytics and underscore its potential for early detection and intervention strategies.

抑郁症预测集成学习心理健康

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