Constructing Depression Prediction Model using Machine Learning Algorithms

Depression is a prevalent mental health disorder with significant impacts on individuals and society. Given the increasing presence of depressive expressions on social media, this study constructs a predictive model to classify depression-related content using machine learning algorithms. Six algorithms for comparing a model, such as Decision Trees, Random Forest, K-nearest neighbors, Support Vector Machines (SVM), Logistic Regression, and Convolutional Neural Networks (CNNs), are used to identify the most effective model for accurately predicting depressive posts. The study utilized a dataset from Twitter, balancing depressive and non-depressive texts for unbiased training. Text preprocessing involved tokenization, stop word removal, and feature extraction using TF-IDF. Evaluation metrics such as accuracy, precision, recall, and F1-score were applied to assess performance. The results showed that Decision Trees and Logistic Regression performed best, with accuracy exceeding 99%. These models were implemented in a prototype for real-time depression risk prediction. Future work aims to expand the model's applicability to Thai language texts, enhancing support for mental health care

Authors

Institutions

Publication Details

Published
2024-12-13
DOI
https://doi.org/10.1145/3711542.3711573
Primary Topic
Maternal Mental Health During Pregnancy and Postpartum
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Constructing Depression Prediction Model using Machine Learning Algorithms

Vatinee Nuipian, Sorawit Hanumas, Kannika Plangklang
Maternal Mental Health During Pregnancy and Postpartum
article

Constructing Depression Prediction Model using Machine Learning Algorithms

Vatinee Nuipian, Sorawit Hanumas, Kannika Plangklang
article en

Abstract

Depression is a prevalent mental health disorder with significant impacts on individuals and society. Given the increasing presence of depressive expressions on social media, this study constructs a predictive model to classify depression-related content using machine learning algorithms. Six algorithms for comparing a model, such as Decision Trees, Random Forest, K-nearest neighbors, Support Vector Machines (SVM), Logistic Regression, and Convolutional Neural Networks (CNNs), are used to identify the most effective model for accurately predicting depressive posts. The study utilized a dataset from Twitter, balancing depressive and non-depressive texts for unbiased training. Text preprocessing involved tokenization, stop word removal, and feature extraction using TF-IDF. Evaluation metrics such as accuracy, precision, recall, and F1-score were applied to assess performance. The results showed that Decision Trees and Logistic Regression performed best, with accuracy exceeding 99%. These models were implemented in a prototype for real-time depression risk prediction. Future work aims to expand the model's applicability to Thai language texts, enhancing support for mental health care

King Mongkut's University of Technology North Bangkok (TH)
Openalex Percentile: Top 60%
Maternal Mental Health During Pregnancy and Postpartum
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.

Constructing Depression Prediction Model using Machine Learning Algorithms — Vatinee Nuipian, Sorawit Hanumas, et al. · (2024) | TGRS Research Map | TGRS