- Research Article
- 10.54254/2753-8818/2026.ch30481
Prediction of Mental Health Risk Levels for Social Media Users Based on Multi-head Attention Mechanism Optimization of Bidirectional Gated Recurrent Unit Networks
- Dec 11, 2025
- Theoretical and Natural Science
- Xu Qihao + 5 more +5
In the field of social media mental health monitoring research, machine learning algorithms serve as the core support, establishing a crucial link between multi-source data and mental health status assessment. In view of the inherent flaws of traditional models, this paper constructs a mental health risk prediction model based on the Bidirectional gated Recurrent Unit (BiGRU) network. The research first conducts data distribution analysis and correlation analysis on the dataset, and then introduces multiple machine learning algorithms for classification and comparison. The results show that the proposed model demonstrates a comprehensive leading advantage in all evaluation indicators. Its accuracy rate, recall rate, precision rate and F1 value all reach 95.4%. Not only is it significantly higher than the 82.9%, 82.9%, 83.5%, 82.8% of Naive Bayes, 89.9%, 89.9%, 90.4%, 89.9% of ExtraTrees, and 87.7%, 87.7%, 87.9%, 87.7% of gradient boosting decision trees, It far exceeds support vector machines whose core classification indicators are only in the range of 54.3% to 57.9%, and also has a slight improvement compared to the suboptimal CatBoost. In terms of the AUC metric reflecting the ability to distinguish categories, this model achieved a high score of 99.6%, approaching the perfect classification level. It is slightly better than CatBoost's 99.2% and 99.0%, and significantly outperforms other comparison models, demonstrating superior classification accuracy and category discrimination ability. This model provides an efficient solution for the precise identification of mental health risks on social media, which is of great significance for promoting the practical development of mental health monitoring technology.
Read more