HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

A Simple Language Model for Task-Oriented Dialogue

Ehsan Hosseini-Asl; Bryan McCann; Chien-Sheng Wu; Semih Yavuz; Richard Socher

A Simple Language Model for Task-Oriented Dialogue

Abstract

Task-oriented dialogue is often decomposed into three tasks: understanding user input, deciding actions, and generating a response. While such decomposition might suggest a dedicated model for each sub-task, we find a simple, unified approach leads to state-of-the-art performance on the MultiWOZ dataset. SimpleTOD is a simple approach to task-oriented dialogue that uses a single, causal language model trained on all sub-tasks recast as a single sequence prediction problem. This allows SimpleTOD to fully leverage transfer learning from pre-trained, open domain, causal language models such as GPT-2. SimpleTOD improves over the prior state-of-the-art in joint goal accuracy for dialogue state tracking, and our analysis reveals robustness to noisy annotations in this setting. SimpleTOD also improves the main metrics used to evaluate action decisions and response generation in an end-to-end setting: inform rate by 8.1 points, success rate by 9.7 points, and combined score by 7.2 points.

Code Repositories

salesforce/simpletod
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
end-to-end-dialogue-modelling-on-multiwoz-2-0SimpleTOD
BLEU: 15.0
MultiWOZ (Inform): 84.4
MultiWOZ (Success): 70.1
end-to-end-dialogue-modelling-on-multiwoz-2-1SimpleTOD
BLEU: 15.2
MultiWOZ (Inform): 85.0
MultiWOZ (Success): 70.5
response-generation-on-mmconvSimpleTOD
BLEU: 20.3
Comb.: 32.2
Inform: 14.6
Success: 9.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp