Contextualize Knowledge Bases with Transformer for End-to-end Task-Oriented Dialogue Systems

Yanjie Gou, Yinjie Lei, Lingqiao Liu, Yong Dai, Chunxu Shen


Abstract
Incorporating knowledge bases (KB) into end-to-end task-oriented dialogue systems is challenging, since it requires to properly represent the entity of KB, which is associated with its KB context and dialogue context. The existing works represent the entity with only perceiving a part of its KB context, which can lead to the less effective representation due to the information loss, and adversely favor KB reasoning and response generation. To tackle this issue, we explore to fully contextualize the entity representation by dynamically perceiving all the relevant entities and dialogue history. To achieve this, we propose a COntext-aware Memory Enhanced Transformer framework (COMET), which treats the KB as a sequence and leverages a novel Memory Mask to enforce the entity to only focus on its relevant entities and dialogue history, while avoiding the distraction from the irrelevant entities. Through extensive experiments, we show that our COMET framework can achieve superior performance over the state of the arts.
Anthology ID:
2021.emnlp-main.353
Volume:
Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing
Month:
November
Year:
2021
Address:
Online and Punta Cana, Dominican Republic
Editors:
Marie-Francine Moens, Xuanjing Huang, Lucia Specia, Scott Wen-tau Yih
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
4300–4310
Language:
URL:
https://aclanthology.org/2021.emnlp-main.353
DOI:
10.18653/v1/2021.emnlp-main.353
Bibkey:
Cite (ACL):
Yanjie Gou, Yinjie Lei, Lingqiao Liu, Yong Dai, and Chunxu Shen. 2021. Contextualize Knowledge Bases with Transformer for End-to-end Task-Oriented Dialogue Systems. In Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, pages 4300–4310, Online and Punta Cana, Dominican Republic. Association for Computational Linguistics.
Cite (Informal):
Contextualize Knowledge Bases with Transformer for End-to-end Task-Oriented Dialogue Systems (Gou et al., EMNLP 2021)
Copy Citation:
PDF:
https://aclanthology.org/2021.emnlp-main.353.pdf
Data
MultiWOZ