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

BoB: BERT Over BERT for Training Persona-based Dialogue Models from Limited Personalized Data

Harbin Institute of Technology(哈尔滨工业大学)

Haoyu Song, Yan Wang, Kaiyan Zhang, Wei-Nan Zhang, Ting Liu

英文摘要:

Maintaining consistent personas is essential for dialogue agents. Although tremendous advancements have been brought, the limited-scale of annotated persona-dense data are still barriers towards training robust and consistent persona-based dialogue models. In this work, we show how the challenges can be addressed by disentangling persona-based dialogue generation into two sub-tasks with a novel BERT-over-BERT (BoB) model. Specifically, the model consists of a BERT-based encoder and two BERT-based decoders, where one decoder is for response generation, and another is for consistency understanding. In particular, to learn the ability of consistency understanding from large-scale non-dialogue inference data, we train the second decoder in an unlikelihood manner. Under different limited data settings, both automatic and human evaluations demonstrate that the proposed model outperforms strong baselines in response quality and persona consistency.

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

↑