Learning Knowledge Bases with Parameters for Task-Oriented Dialogue Systems
This addresses the problem of scaling end-to-end dialogue systems for developers and researchers, though it is incremental as it builds on existing embedding methods.
The authors tackled the scalability and annotation inefficiency of task-oriented dialogue systems by embedding knowledge bases directly into model parameters, achieving competitive performance across five datasets with varying KB sizes.
Task-oriented dialogue systems are either modularized with separate dialogue state tracking (DST) and management steps or end-to-end trainable. In either case, the knowledge base (KB) plays an essential role in fulfilling user requests. Modularized systems rely on DST to interact with the KB, which is expensive in terms of annotation and inference time. End-to-end systems use the KB directly as input, but they cannot scale when the KB is larger than a few hundred entries. In this paper, we propose a method to embed the KB, of any size, directly into the model parameters. The resulting model does not require any DST or template responses, nor the KB as input, and it can dynamically update its KB via fine-tuning. We evaluate our solution in five task-oriented dialogue datasets with small, medium, and large KB size. Our experiments show that end-to-end models can effectively embed knowledge bases in their parameters and achieve competitive performance in all evaluated datasets.