International Journal of Data and Network Science, Volume 9, Issue 4, Pages 737-750 , 01/09/2025

Classification models combined with optimized features for mental stress prediction

Tran Anh Tuan, Dao Thi Thanh Loan, Bundit Buddhahai

Abstract

Mental stress is a growing global health concern, closely linked to psychological, behavioral, and physiological disorders. Accurate and early prediction of mental stress is crucial for timely interventions and improved health outcomes. Despite numerous studies leveraging machine learning (ML) techniques for stress classification, many have overlooked the integration of systematic feature selection and comprehensive model evaluation, limiting generalizability and interpretability. To address these gaps, this study proposes a robust ML-based framework that combines optimized feature selection methods-Recursive Feature Elimination (RFE), Extra Trees (ET), and Boruta-with various classification algorithms including Random Forest (RF), K-Nearest Neighbors (KNN), Decision Tree (DT), Multilayer Perceptron (MLP), Support Vector Machine (SVM), Gradient Boosting, and voting classifier. The models were evaluated using 10-fold cross-validation and ranked using the TOPSIS multi-criteria decision-making approach. The experimental results demonstrate high predictive performance across models (accuracy ≥ 0.98), with RF, DT, MLP, and Gradient Boosting achieving perfect accuracy (1.00). Among all configurations, the RF-Boruta model emerged as the most optimal (TOPSIS score: 0.914558). These findings highlight the effectiveness of combining systematic feature optimization with ML classification for accurate and interpretable stress prediction, offering valuable insights for data-driven mental health interventions.

Document Type

Article

Source Type

Journal

Keywords

Classification modelMachine learningMental stressOptimized feature

ASJC Subject Area

Computer Science : Computer Networks and CommunicationsComputer Science : Information SystemsComputer Science : SoftwareComputer Science : Computer Science ApplicationsComputer Science : Artificial IntelligenceSocial Sciences : Communication

Funding Agency

Weatherhead Center for International Affairs, Harvard University


Bibliography


Tuan, T., Loan, D., & Buddhahai, B. (2025). Classification models combined with optimized features for mental stress prediction. International Journal of Data and Network Science, 9(4) 737-750. doi:10.5267/j.ijdns.2025.8.010

Copy | Save