用大模型同时识抑郁并生成医学解释,让算法更可信。
Generating Medically-Informed Explanations for Depression Detection using LLMs
- 用大模型多任务学习,一边判抑郁一边写医学依据。
- 在Reddit数据集上AUPRC领先,比BERT等模型更准。
- 人工评估显示解释专业完整,适合临床辅助场景。
从社交媒体数据中早期发现抑郁症为及时干预提供了宝贵机会。然而,该任务面临巨大挑战,既需专业医学知识,又要求模型具备高精度与可解释性。本文提出LLM-MTD(大语言模型用于多任务抑郁症检测),利用预训练大语言模型同时对社交媒体文本进行抑郁症分类,并生成基于医学诊断标准的文本解释。模型采用多任务学习框架,结合损失函数优化分类准确率与解释质量。我们在基准数据集Reddit自报告抑郁症数据集(RSDD)上评估了LLM-MTD,对比了多种基线方法,包括传统机器学习和微调的BERT。实验结果表明,LLM-MTD在抑郁症检测上达到当前最优性能,显著提升了AUPRC及其他关键指标。此外,人工评估显示生成的解释具有相关性、完整性和医学准确性,凸显了本方法的可解释性优势。本研究提出了一种将大语言模型能力与可解释性相结合的新范式,推动抑郁症检测向临床实用迈进。
原文摘要 · Abstract (English)
Early detection of depression from social media data offers a valuable opportunity for timely intervention. However, this task poses significant challenges, requiring both professional medical knowledge and the development of accurate and explainable models. In this paper, we propose LLM-MTD (Large Language Model for Multi-Task Depression Detection), a novel approach that leverages a pre-trained large language model to simultaneously classify social media posts for depression and generate textual explanations grounded in medical diagnostic criteria. We train our model using a multi-task learning framework with a combined loss function that optimizes both classification accuracy and explanation quality. We evaluate LLM-MTD on the benchmark Reddit Self-Reported Depression Dataset (RSDD) and compare its performance against several competitive baseline methods, including traditional machine learning and fine-tuned BERT. Our experimental results demonstrate that LLM-MTD achieves state-of-the-art performance in depression detection, showing significant improvements in AUPRC and other key metrics. Furthermore, human evaluation of the generated explanations reveals their relevance, completeness, and medical accuracy, highlighting the enhanced interpretability of our approach. This work contributes a novel methodology for depression detection that combines the power of large language models with the crucial aspect of explainability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。