Suicidal ideation detection is crucial for preventing suicides, a leading cause of death worldwide. Many individuals express suicidal thoughts on social media, offering a vital opportunity for early detection through advanced machine learning techniques. The identification of suicidal ideation in social media text is improved by utilising a hybrid framework that integrates Convolutional Neural Networks (CNN) and Bidirectional Long Short-Term Memory (BiLSTM), enhanced with an attention mechanism. To enhance the interpretability of the model's predictions, Explainable AI (XAI) methods are applied, with a particular focus on SHapley Additive exPlanations (SHAP), are incorporated. At first, the model managed to reach an accuracy of 92.81%. By applying fine-tuning and early stopping techniques, the accuracy improved to 94.29%. The SHAP analysis revealed key features influencing the model's predictions, such as terms related to mental health struggles. This level of transparency boosts the model's credibility while helping mental health professionals understand and trust the predictions. This work highlights the potential for improving the accuracy and interpretability of detecting suicidal tendencies, making a valuable contribution to the progress of mental health monitoring systems. It emphasizes the significance of blending powerful machine learning methods with explainability to develop reliable and impactful mental health solutions.


翻译:自杀意念检测对于预防自杀这一全球主要死因至关重要。许多人在社交媒体上表达自杀念头,这为通过先进机器学习技术进行早期检测提供了重要机会。本研究通过整合卷积神经网络(CNN)和双向长短期记忆网络(BiLSTM)的混合框架,并结合注意力机制增强,提升了社交媒体文本中自杀意念的识别效果。为提高模型预测的可解释性,引入了可解释人工智能(XAI)方法,特别聚焦于SHapley可加性解释(SHAP)。初始模型准确率达到92.81%,通过微调和早停技术优化后,准确率提升至94.29%。SHAP分析揭示了影响模型预测的关键特征,如与心理健康困扰相关的术语。这种透明度增强了模型的可信度,同时帮助心理健康专业人员理解和信任预测结果。本研究突显了在提高自杀倾向检测准确性和可解释性方面的潜力,为心理健康监测系统的发展做出了重要贡献。它强调了将强大机器学习方法与可解释性相结合,以开发可靠且具有影响力的心理健康解决方案的重要意义。

0
下载
关闭预览

相关内容

AAAI 2022 | ProtGNN:自解释图神经网络
专知
10+阅读 · 2022年2月28日
【CVPR 2020 Oral】小样本类增量学习
专知
20+阅读 · 2020年6月26日
论文浅尝 | GEOM-GCN: Geometric Graph Convolutional Networks
开放知识图谱
14+阅读 · 2020年4月8日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
VIP会员
相关基金
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
Top
微信扫码咨询专知VIP会员