arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.13835cs.CLcs.AI

A Tailored Pre-Training Model for Task-Oriented Dialog Generation

University of California, Davis(加州大学戴维斯分校)

Jing Gu, Qingyang Wu, Chongruo Wu, Weiyan Shi, Zhou Yu

英文摘要:

The recent success of large pre-trained language models such as BERT and GPT-2 has suggested the effectiveness of incorporating language priors in downstream dialog generation tasks. However, the performance of pre-trained models on the dialog task is not as optimal as expected. In this paper, we propose a Pre-trained Role Alternating Language model (PRAL), designed specifically for task-oriented conversational systems. We adopted (Wu et al., 2019) that models two speakers separately. We also design several techniques, such as start position randomization, knowledge distillation, and history discount to improve pre-training performance. We introduce a task-oriented dialog pretraining dataset by cleaning 13 existing data sets. We test PRAL on three different downstream tasks. The results show that PRAL performs better or on par with state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑