Leveraging A Stacking Ensemble Model for Accurate Depression Prediction and Diagnosis Across All Ages
ABSTRACT 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.
Authors
- Reza Shamsi
- Abeed Sarker (ORCID: https://orcid.org/0000-0001-7358-544X)
- Saha Reno (ORCID: https://orcid.org/0000-0003-1897-9002)
- Koushik Roy (ORCID: https://orcid.org/0000-0002-5624-5723)
- Md. Easin
Institutions
- Ahsanullah University of Science and Technology (BD)
- Nara University (JP)
- United International University (BD)
Publication Details
- Journal
- Engineering Reports
- Published
- 2025-09-28
- DOI
- https://doi.org/10.1002/eng2.70416
- Primary Topic
- Mental Health Research Topics
- Type
- article
- Field-Weighted Citation Impact
- 0.00