Conversational Topic Recommendation in Counseling and Psychotherapy with Decision Transformer and Large Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Gunal, Aylin, Lin, Baihan, Bouneffouf, Djallel
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929337761005568
author Gunal, Aylin
Lin, Baihan
Bouneffouf, Djallel
author_facet Gunal, Aylin
Lin, Baihan
Bouneffouf, Djallel
contents Given the increasing demand for mental health assistance, artificial intelligence (AI), particularly large language models (LLMs), may be valuable for integration into automated clinical support systems. In this work, we leverage a decision transformer architecture for topic recommendation in counseling conversations between patients and mental health professionals. The architecture is utilized for offline reinforcement learning, and we extract states (dialogue turn embeddings), actions (conversation topics), and rewards (scores measuring the alignment between patient and therapist) from previous turns within a conversation to train a decision transformer model. We demonstrate an improvement over baseline reinforcement learning methods, and propose a novel system of utilizing our model's output as synthetic labels for fine-tuning a large language model for the same task. Although our implementation based on LLaMA-2 7B has mixed results, future work can undoubtedly build on the design.
format Preprint
id arxiv_https___arxiv_org_abs_2405_05060
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Conversational Topic Recommendation in Counseling and Psychotherapy with Decision Transformer and Large Language Models
Gunal, Aylin
Lin, Baihan
Bouneffouf, Djallel
Computation and Language
Given the increasing demand for mental health assistance, artificial intelligence (AI), particularly large language models (LLMs), may be valuable for integration into automated clinical support systems. In this work, we leverage a decision transformer architecture for topic recommendation in counseling conversations between patients and mental health professionals. The architecture is utilized for offline reinforcement learning, and we extract states (dialogue turn embeddings), actions (conversation topics), and rewards (scores measuring the alignment between patient and therapist) from previous turns within a conversation to train a decision transformer model. We demonstrate an improvement over baseline reinforcement learning methods, and propose a novel system of utilizing our model's output as synthetic labels for fine-tuning a large language model for the same task. Although our implementation based on LLaMA-2 7B has mixed results, future work can undoubtedly build on the design.
title Conversational Topic Recommendation in Counseling and Psychotherapy with Decision Transformer and Large Language Models
topic Computation and Language
url https://arxiv.org/abs/2405.05060