arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.01775cs.CLcs.AIcs.LG

DialogBERT: Discourse-Aware Response Generation via Learning to Recover and Rank Utterances

School of Software, Shanghai Jiao Tong University(上海交通大学软件学院);NAVER AI Lab(NAVER人工智能实验室)

Xiaodong Gu, Kang Min Yoo, Jung-Woo Ha

英文摘要:

Recent advances in pre-trained language models have significantly improved neural response generation. However, existing methods usually view the dialogue context as a linear sequence of tokens and learn to generate the next word through token-level self-attention. Such token-level encoding hinders the exploration of discourse-level coherence among utterances. This paper presents DialogBERT, a novel conversational response generation model that enhances previous PLM-based dialogue models. DialogBERT employs a hierarchical Transformer architecture. To efficiently capture the discourse-level coherence among utterances, we propose two training objectives, including masked utterance regression and distributed utterance order ranking in analogy to the original BERT training. Experiments on three multi-turn conversation datasets show that our approach remarkably outperforms the baselines, such as BART and DialoGPT, in terms of quantitative evaluation. The human evaluation suggests that DialogBERT generates more coherent, informative, and human-like responses than the baselines with significant margins.

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

↑