English

Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System

Computation and Language 2022-03-02 v2

Abstract

Pre-trained language models have been recently shown to benefit task-oriented dialogue (TOD) systems. Despite their success, existing methods often formulate this task as a cascaded generation problem which can lead to error accumulation across different sub-tasks and greater data annotation overhead. In this study, we present PPTOD, a unified plug-and-play model for task-oriented dialogue. In addition, we introduce a new dialogue multi-task pre-training strategy that allows the model to learn the primary TOD task completion skills from heterogeneous dialog corpora. We extensively test our model on three benchmark TOD tasks, including end-to-end dialogue modelling, dialogue state tracking, and intent classification. Experimental results show that PPTOD achieves new state of the art on all evaluated tasks in both high-resource and low-resource scenarios. Furthermore, comparisons against previous SOTA methods show that the responses generated by PPTOD are more factually correct and semantically coherent as judged by human annotators.

Keywords

Cite

@article{arxiv.2109.14739,
  title  = {Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System},
  author = {Yixuan Su and Lei Shu and Elman Mansimov and Arshit Gupta and Deng Cai and Yi-An Lai and Yi Zhang},
  journal= {arXiv preprint arXiv:2109.14739},
  year   = {2022}
}

Comments

Camera-ready for ACL2022 main conference