arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.06871cs.CL

TOD-BERT: Pre-trained Natural Language Understanding for Task-Oriented Dialogue

Salesforce Research(Salesforce研究院)

Chien-Sheng Wu, Steven Hoi, Richard Socher, Caiming Xiong

英文摘要:

The underlying difference of linguistic patterns between general text and task-oriented dialogue makes existing pre-trained language models less useful in practice. In this work, we unify nine human-human and multi-turn task-oriented dialogue datasets for language modeling. To better model dialogue behavior during pre-training, we incorporate user and system tokens into the masked language modeling. We propose a contrastive objective function to simulate the response selection task. Our pre-trained task-oriented dialogue BERT (TOD-BERT) outperforms strong baselines like BERT on four downstream task-oriented dialogue applications, including intention recognition, dialogue state tracking, dialogue act prediction, and response selection. We also show that TOD-BERT has a stronger few-shot ability that can mitigate the data scarcity problem for task-oriented dialogue.

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

↑