Proposal of a Model to Detect Depression in Social Media User Posts Using Machine Learning Techniques
摘要
The detection of mental health problems, such as depression, through user-generated content on social networks has become a key research topic. This study proposes a machine learning-based model to identify signs of depression in social network posts by comparing several algorithms: Random Forest (RF), Logistic Regression (LR), Support Vector Machines (SVM), K-Nearest Neighbors (KNN), Long Short-Term Memory networks (LSTM), and Multilayer Perceptrons (MLP) through three feature extraction techniques: Bag of Words (BoW), Term Frequency-Inverse Document Frequency (TF-IDF), and Word2Vec (W2V). Initial results obtained before hyperparameter tuning indicated that LR with BoW obtained the highest accuracy (95.86%) and highest F1 score (95.68%), making it ideal for scenarios in which accuracy is prioritized. After applying careful hyperparameter optimization, performance improved significantly: SVM with BoW obtained the highest F1 score (96.09%), which was closely followed by LSTM and logistic regression with TF-IDF, both above 95.73%. These results underscore the substantial impact of hyperparameter tuning and highlight the potential of carefully aligning machine learning models with BoW and TF-IDF feature extraction techniques to advance automated mental health monitoring in social network contexts.