Bootstrapping LLM-based Task-Oriented Dialogue Agents via Self-Talk

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ulmer, Dennis, Mansimov, Elman, Lin, Kaixiang, Sun, Justin, Gao, Xibin, Zhang, Yi
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916086842130432
author Ulmer, Dennis
Mansimov, Elman
Lin, Kaixiang
Sun, Justin
Gao, Xibin
Zhang, Yi
author_facet Ulmer, Dennis
Mansimov, Elman
Lin, Kaixiang
Sun, Justin
Gao, Xibin
Zhang, Yi
contents Large language models (LLMs) are powerful dialogue agents, but specializing them towards fulfilling a specific function can be challenging. Instructing tuning, i.e. tuning models on instruction and sample responses generated by humans (Ouyang et al., 2022), has proven as an effective method to do so, yet requires a number of data samples that a) might not be available or b) costly to generate. Furthermore, this cost increases when the goal is to make the LLM follow a specific workflow within a dialogue instead of single instructions. Inspired by the self-play technique in reinforcement learning and the use of LLMs to simulate human agents, we propose a more effective method for data collection through LLMs engaging in a conversation in various roles. This approach generates a training data via "self-talk" of LLMs that can be refined and utilized for supervised fine-tuning. We introduce an automated way to measure the (partial) success of a dialogue. This metric is used to filter the generated conversational data that is fed back in LLM for training. Based on our automated and human evaluations of conversation quality, we demonstrate that such self-talk data improves results. In addition, we examine the various characteristics that showcase the quality of generated dialogues and how they can be connected to their potential utility as training data.
format Preprint
id arxiv_https___arxiv_org_abs_2401_05033
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bootstrapping LLM-based Task-Oriented Dialogue Agents via Self-Talk
Ulmer, Dennis
Mansimov, Elman
Lin, Kaixiang
Sun, Justin
Gao, Xibin
Zhang, Yi
Computation and Language
Artificial Intelligence
Large language models (LLMs) are powerful dialogue agents, but specializing them towards fulfilling a specific function can be challenging. Instructing tuning, i.e. tuning models on instruction and sample responses generated by humans (Ouyang et al., 2022), has proven as an effective method to do so, yet requires a number of data samples that a) might not be available or b) costly to generate. Furthermore, this cost increases when the goal is to make the LLM follow a specific workflow within a dialogue instead of single instructions. Inspired by the self-play technique in reinforcement learning and the use of LLMs to simulate human agents, we propose a more effective method for data collection through LLMs engaging in a conversation in various roles. This approach generates a training data via "self-talk" of LLMs that can be refined and utilized for supervised fine-tuning. We introduce an automated way to measure the (partial) success of a dialogue. This metric is used to filter the generated conversational data that is fed back in LLM for training. Based on our automated and human evaluations of conversation quality, we demonstrate that such self-talk data improves results. In addition, we examine the various characteristics that showcase the quality of generated dialogues and how they can be connected to their potential utility as training data.
title Bootstrapping LLM-based Task-Oriented Dialogue Agents via Self-Talk
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2401.05033