Integrating Bert With CNN and BiLSTM for Explainable Detection of Depression in Social Media Contents

被引:2
作者
Xin, Cao [1 ]
Zakaria, Lailatul Qadri [1 ]
机构
[1] Univ Kebangsaan Malaysia, Fac Informat Sci & Technol, Ctr Artificial Intelligence Technol CAIT, Bangi 43600, Malaysia
关键词
Depression; Social networking (online); Data models; Adaptation models; Mental health; Training; Encoding; Bidirectional control; Predictive models; Transformers; Automatic depression detection; BERT; CNN; BiLSTM; MentalBERT; deep learning; Explainability AI;
D O I
10.1109/ACCESS.2024.3488081
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Depression is a prevalent mental health condition that significantly impacts individuals' lives. Early detection of depression is crucial for timely intervention and improved outcomes. However, traditional machine learning approaches are constrained by the limited amount of annotated data and lack of model transparency. This study aims to address these challenges by leveraging social media data and advanced natural language processing techniques to develop effective and explainable models for depression detection. The study focuses on two main objectives. The first objective is to develop and evaluate fine-tuned Bidirectional Encoder Representations from Transformers (BERT), BERT with Bidirectional Long Short-Term Memory (BERT-BiLSTM), and BERT with Convolutional Neural Network (BERT-CNN) models, and compare their performance with MentalBERT, a state-of-the-art model for mental health detection. The second objective is to observe the key features used by the BERT models to make the decision-making using Transformer Interpretability Beyond Attention Visualization and Average Attention Weight methods. The study utilizes three publicly available datasets: the Depression Reddit Dataset, the Sentiment Analysis for Tweets Dataset, and the Mental Health Corpus. The results demonstrate that the proposed models, especially BERT-BiLSTM and BERT-CNN, achieve superior performance compared to MentalBERT, particularly regarding accuracy and F1-score. Notably, BERT-CNN achieved exceptional accuracy scores of 0.982, 0.961, and 1.0 on the Depression Reddit Dataset, the Mental Health Corpus, and the Sentiment Analysis for Tweets Dataset, respectively, demonstrating its robust performance across different social media contexts. The attention map visualizations provide valuable insights into the language patterns and key features associated with depression in social media posts. This study contributes to the mental health field by presenting novel and explainable models for depression detection using social media data. The proposed approaches have the potential to assist mental health professionals in early identification and intervention, ultimately improving the lives of individuals affected by depression.
引用
收藏
页码:161203 / 161212
页数:10
相关论文
共 32 条
[1]  
Al Govindasamy Kuhaneswaran, 2021, 2021 5th International Conference on Intelligent Computing and Control Systems (ICICCS), P960, DOI 10.1109/ICICCS51141.2021.9432203
[2]   Deep Learning for Depression Detection from Textual Data [J].
Amanat, Amna ;
Rizwan, Muhammad ;
Javed, Abdul Rehman ;
Abdelhaq, Maha ;
Alsaqour, Raed ;
Pandya, Sharnil ;
Uddin, Mueen .
ELECTRONICS, 2022, 11 (05)
[3]  
[Anonymous], 2019, World Health Organization
[4]  
[Anonymous], 2023, Mental Health Corpus
[5]  
[Anonymous], 2021, Sentimental Analysis for Tweets
[6]  
[Anonymous], 2021, The National Institute of Mental Health
[7]   Explainable Artificial Intelligence (XAI): Concepts, taxonomies, opportunities and challenges toward responsible AI [J].
Barredo Arrieta, Alejandro ;
Diaz-Rodriguez, Natalia ;
Del Ser, Javier ;
Bennetot, Adrien ;
Tabik, Siham ;
Barbado, Alberto ;
Garcia, Salvador ;
Gil-Lopez, Sergio ;
Molina, Daniel ;
Benjamins, Richard ;
Chatila, Raja ;
Herrera, Francisco .
INFORMATION FUSION, 2020, 58 :82-115
[8]   A text classification framework for simple and effective early depression detection over social media streams [J].
Burdisso, Sergio G. ;
Errecalde, Marcelo ;
Montes-y-Gomez, Manuel .
EXPERT SYSTEMS WITH APPLICATIONS, 2019, 133 :182-197
[9]   Early Detection of Depression: Social Network Analysis and Random Forest Techniques [J].
Cacheda, Fidel ;
Fernandez, Diego ;
Novoa, Francisco J. ;
Carneiro, Victor .
JOURNAL OF MEDICAL INTERNET RESEARCH, 2019, 21 (06)
[10]  
Chang Y. Yao, 2023, Asia-Pacific J. Inf. Technol. Multimedia, V12, P39, DOI [10.17576/apjitm-2023-1201-03, DOI 10.17576/APJITM-2023-1201-03]