Efficient or Powerful? Trade-offs Between Machine Learning and Deep Learning for Mental Illness Detection on Social Media
By: Zhanyi Ding , Zhongyan Wang , Yeyubei Zhang and more
Potential Business Impact:
Helps computers find mental health issues in online posts.
Social media platforms provide valuable insights into mental health trends by capturing user-generated discussions on conditions such as depression, anxiety, and suicidal ideation. Machine learning (ML) and deep learning (DL) models have been increasingly applied to classify mental health conditions from textual data, but selecting the most effective model involves trade-offs in accuracy, interpretability, and computational efficiency. This study evaluates multiple ML models, including logistic regression, random forest, and LightGBM, alongside deep learning architectures such as ALBERT and Gated Recurrent Units (GRUs), for both binary and multi-class classification of mental health conditions. Our findings indicate that ML and DL models achieve comparable classification performance on medium-sized datasets, with ML models offering greater interpretability through variable importance scores, while DL models are more robust to complex linguistic patterns. Additionally, ML models require explicit feature engineering, whereas DL models learn hierarchical representations directly from text. Logistic regression provides the advantage of capturing both positive and negative associations between features and mental health conditions, whereas tree-based models prioritize decision-making power through split-based feature selection. This study offers empirical insights into the advantages and limitations of different modeling approaches and provides recommendations for selecting appropriate methods based on dataset size, interpretability needs, and computational constraints.
Similar Papers
Understanding Mental Health Content on Social Media and Its Effect Towards Suicidal Ideation
Computers and Society
Helps find people needing mental health help online.
AI-Powered Early Diagnosis of Mental Health Disorders from Real-World Clinical Conversations
Computation and Language
Helps doctors find mental health problems faster.
Classification of Psychiatry Clinical Notes by Diagnosis: A Deep Learning and Machine Learning Approach
Machine Learning (CS)
Helps computers find mental health problems in notes.