利用堆叠集成模型对所有年龄段的抑郁症进行准确预测和诊断

IF 2 Q3 COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS
Koushik Roy, Md. Easin, Saha Reno, Al Mahmud Sarker, Rahyan Shamsi
{"title":"利用堆叠集成模型对所有年龄段的抑郁症进行准确预测和诊断","authors":"Koushik Roy,&nbsp;Md. Easin,&nbsp;Saha Reno,&nbsp;Al Mahmud Sarker,&nbsp;Rahyan Shamsi","doi":"10.1002/eng2.70416","DOIUrl":null,"url":null,"abstract":"<p>Depression is a serious mental health issue affecting people of all ages, with early detection being crucial for timely treatment. In this study, we developed a highly accurate machine-learning model using a stacking ensemble technique to predict depression. The proposed model integrates several base learners, including XGBoost, extra trees, and gradient boosting, with Random Forest as the meta-learner. By applying feature engineering, hyperparameter tuning, and balancing techniques like SMOTE, we optimized the model's performance. The final model achieved impressive performance, with accuracy, precision, recall, and F1-score all reaching 96.8%, and an AUC of 0.988. The model's average precision was also notably high at 0.990, demonstrating its effectiveness in balancing precision and recall. These results show the model's potential to greatly enhance early diagnosis and intervention for depression, offering hope for improved mental health outcomes across various age groups.</p>","PeriodicalId":72922,"journal":{"name":"Engineering reports : open access","volume":"7 10","pages":""},"PeriodicalIF":2.0000,"publicationDate":"2025-09-28","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"https://onlinelibrary.wiley.com/doi/epdf/10.1002/eng2.70416","citationCount":"0","resultStr":"{\"title\":\"Leveraging A Stacking Ensemble Model for Accurate Depression Prediction and Diagnosis Across All Ages\",\"authors\":\"Koushik Roy,&nbsp;Md. Easin,&nbsp;Saha Reno,&nbsp;Al Mahmud Sarker,&nbsp;Rahyan Shamsi\",\"doi\":\"10.1002/eng2.70416\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"<p>Depression is a serious mental health issue affecting people of all ages, with early detection being crucial for timely treatment. In this study, we developed a highly accurate machine-learning model using a stacking ensemble technique to predict depression. The proposed model integrates several base learners, including XGBoost, extra trees, and gradient boosting, with Random Forest as the meta-learner. By applying feature engineering, hyperparameter tuning, and balancing techniques like SMOTE, we optimized the model's performance. The final model achieved impressive performance, with accuracy, precision, recall, and F1-score all reaching 96.8%, and an AUC of 0.988. The model's average precision was also notably high at 0.990, demonstrating its effectiveness in balancing precision and recall. These results show the model's potential to greatly enhance early diagnosis and intervention for depression, offering hope for improved mental health outcomes across various age groups.</p>\",\"PeriodicalId\":72922,\"journal\":{\"name\":\"Engineering reports : open access\",\"volume\":\"7 10\",\"pages\":\"\"},\"PeriodicalIF\":2.0000,\"publicationDate\":\"2025-09-28\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"https://onlinelibrary.wiley.com/doi/epdf/10.1002/eng2.70416\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Engineering reports : open access\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://onlinelibrary.wiley.com/doi/10.1002/eng2.70416\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q3\",\"JCRName\":\"COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Engineering reports : open access","FirstCategoryId":"1085","ListUrlMain":"https://onlinelibrary.wiley.com/doi/10.1002/eng2.70416","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q3","JCRName":"COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS","Score":null,"Total":0}
引用次数: 0

摘要

抑郁症是一种严重的心理健康问题,影响着所有年龄段的人,早期发现对于及时治疗至关重要。在这项研究中,我们开发了一个高度精确的机器学习模型,使用堆叠集成技术来预测抑郁症。该模型集成了几种基本学习器,包括XGBoost、额外树和梯度增强,并以随机森林作为元学习器。通过应用特征工程、超参数调优和SMOTE等平衡技术,我们优化了模型的性能。最终模型取得了令人印象深刻的性能,准确率、精密度、召回率和f1得分均达到96.8%,AUC为0.988。模型的平均精度也非常高,达到0.990,证明了其在平衡精度和召回率方面的有效性。这些结果表明,该模型有潜力极大地提高抑郁症的早期诊断和干预,为改善不同年龄组的心理健康状况带来了希望。
本文章由计算机程序翻译,如有差异,请以英文原文为准。

Leveraging A Stacking Ensemble Model for Accurate Depression Prediction and Diagnosis Across All Ages

Leveraging A Stacking Ensemble Model for Accurate Depression Prediction and Diagnosis Across All Ages

Depression is a serious mental health issue affecting people of all ages, with early detection being crucial for timely treatment. In this study, we developed a highly accurate machine-learning model using a stacking ensemble technique to predict depression. The proposed model integrates several base learners, including XGBoost, extra trees, and gradient boosting, with Random Forest as the meta-learner. By applying feature engineering, hyperparameter tuning, and balancing techniques like SMOTE, we optimized the model's performance. The final model achieved impressive performance, with accuracy, precision, recall, and F1-score all reaching 96.8%, and an AUC of 0.988. The model's average precision was also notably high at 0.990, demonstrating its effectiveness in balancing precision and recall. These results show the model's potential to greatly enhance early diagnosis and intervention for depression, offering hope for improved mental health outcomes across various age groups.

求助全文
通过发布文献求助,成功后即可免费获取论文全文。 去求助
来源期刊
CiteScore
5.10
自引率
0.00%
发文量
0
审稿时长
19 weeks
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
确定
请完成安全验证×
copy
已复制链接
快去分享给好友吧!
我知道了
右上角分享
点击右上角分享
0
联系我们:info@booksci.cn Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。 Copyright © 2023 布克学术 All rights reserved.
京ICP备2023020795号-1
ghs 京公网安备 11010802042870号
Book学术文献互助
Book学术文献互助群
群 号:604180095
Book学术官方微信