← 返回资源分享
Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System
论文
论文
发布时间2021-09-29
发表ACL 2022 5 · arXiv:2109.14739
作者:Deng Cai,Lei Shu,Arshit Gupta,Yi Zhang,Yi-An Lai,Yixuan Su,Elman Mansimov
详细介绍
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.
代码仓库 (2)
awslabs/pptod官方PyTorch
sogang-isds/TOATODPyTorch
