Mem2Seq: Effectively Incorporating Knowledge Bases into End-to-End Task-Oriented Dialog Systems
About
End-to-end task-oriented dialog systems usually suffer from the challenge of incorporating knowledge bases. In this paper, we propose a novel yet simple end-to-end differentiable model called memory-to-sequence (Mem2Seq) to address this issue. Mem2Seq is the first neural generative model that combines the multi-hop attention over memories with the idea of pointer network. We empirically show how Mem2Seq controls each generation step, and how its multi-hop attention mechanism helps in learning correlations between memories. In addition, our model is quite general without complicated task-specific designs. As a result, we show that Mem2Seq can be trained faster and attain the state-of-the-art performance on three different task-oriented dialog datasets.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Task-oriented Dialogue | Stanford Multi-Domain Dialogue (SMD) (test) | BLEU12.6 | 29 | |
| Task-oriented Dialogue Response Generation | Multi-WOZ 2.1 (test) | BLEU6.6 | 22 | |
| Dialog Generation | DSTC2 (test) | Accuracy (Response)45 | 10 | |
| Dialogue Response Generation | bAbI Dialogue Task 4 | Per-response Accuracy100 | 9 | |
| Dialogue Response Generation | bAbI Dialogue Task 4 OOV | Per-response Accuracy100 | 9 | |
| Task-oriented Dialog Generation | In-Car Assistant (test) | BLEU12.6 | 9 | |
| Dialogue Response Generation | bAbI Dialogue Task 1 | Per-response Accuracy100 | 9 | |
| Dialogue Response Generation | bAbI Dialogue Task 3 | Accuracy (Per-response)94.7 | 9 | |
| Dialogue Response Generation | bAbI Dialogue Task 1 OOV | Per-response Accuracy0.94 | 9 | |
| Dialogue Response Generation | bAbI Dialogue Task 2 OOV | Accuracy (Per-response)86.5 | 9 |