Koushik Roy, Md. Easin, Saha Reno, Al Mahmud Sarker, Rahyan Shamsi
{"title":"利用堆叠集成模型对所有年龄段的抑郁症进行准确预测和诊断","authors":"Koushik Roy, Md. Easin, Saha Reno, Al Mahmud Sarker, 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, Md. Easin, Saha Reno, Al Mahmud Sarker, 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}
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.