Classification models combined with optimized features for mental stress prediction

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.

Authors

Publication Details

Journal
International Journal of Data and Network Science
Published
2025-01-01
DOI
https://doi.org/10.5267/j.ijdns.2025.8.010
Citations
2
Primary Topic
Emotion and Mood Recognition
Type
article
Field-Weighted Citation Impact
4.24

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Classification models combined with optimized features for mental stress prediction

Bundit Buddhahai, Dao Thi Thanh Loan, Tran Anh Tuan
2 citations
International Journal of Data and Network Science
Emotion and Mood Recognition
4.24
article

Classification models combined with optimized features for mental stress prediction

Bundit Buddhahai, Dao Thi Thanh Loan, Tran Anh Tuan
article en
2 citations

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.

International Journal of Data and Network ScienceVol. 9(4)
Walailak University
Openalex Percentile: Top 6%
Emotion and Mood Recognition
4.24
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.