LLM2D
MentalArena:用于心理健康障碍诊断和治疗的语言模型的自游戏训练
MentalArena: Self-play Training of Language Models for Diagnosis and Treatment of Mental Health Disorders
作者: Cheng Li, May Fung, Qingyun Wang, Chi Han, Manling Li, Jindong Wang, Heng Ji
发布日期: 10/10/2024
arXiv ID: oai:arXiv.org:2410.06845v1

摘要

arXiv:2410.06845v1 类别:交叉学科 摘要:精神健康障碍是全世界最主要的严重疾病之一。患有此类疾病的人群往往缺乏适当的医疗服务,突显了训练用于精神健康障碍诊断和治疗的模型的重要性。然而,在精神健康领域,隐私问题限制了个性化治疗数据的获取能力,使得构建强大的模型变得颇具挑战性。本文我们引入了MentalArena,这是一种自我博弈框架,通过生成特定领域的个性化数据来训练语言模型,从而获得一个能够进行个性化诊断和治疗(如治疗师)以及提供信息(如患者)的更好模型。为了准确模拟人类精神健康病人,我们设计了Symptom Encoder,从认知和行为的角度模拟了一个真实的病人。为了应对患者-治疗师交互中意图偏见的问题,我们提出了Symptom Decoder来比较诊断症状和编码症状,并根据识别到的偏差动态管理患者和治疗师之间的对话。我们对MentalArena进行了评估,与包括生物医学QA和精神健康任务在内的6个基准进行比较,并与6个先进模型进行对比。我们的模型在GPT-3.5和Llama-3-8b的基础上进行微调,显著优于其同类模型,包括GPT-4o。我们希望我们的工作能够激励未来个性化护理的研究。代码可在https://github.com/Scarelette/MentalArena/tree/main获取