Current browse context:
cs.AI
Change to browse by:
References & Citations
Computer Science > Computation and Language
Title: DialogBERT: Discourse-Aware Response Generation via Learning to Recover and Rank Utterances
(Submitted on 3 Dec 2020 (v1), last revised 13 Dec 2021 (this version, v2))
Abstract: 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.
Submission history
From: Xiaodong Gu [view email][v1] Thu, 3 Dec 2020 09:06:23 GMT (586kb,D)
[v2] Mon, 13 Dec 2021 06:48:34 GMT (197kb,D)
Link back to: arXiv, form interface, contact.