Knowledge-Aware Graph-Enhanced GPT-2 for Dialogue State Tracking

Weizhe Lin, Bo-Hsiang Tseng, Bill Byrne


Abstract
Dialogue State Tracking is central to multi-domain task-oriented dialogue systems, responsible for extracting information from user utterances. We present a novel hybrid architecture that augments GPT-2 with representations derived from Graph Attention Networks in such a way to allow causal, sequential prediction of slot values. The model architecture captures inter-slot relationships and dependencies across domains that otherwise can be lost in sequential prediction. We report improvements in state tracking performance in MultiWOZ 2.0 against a strong GPT-2 baseline and investigate a simplified sparse training scenario in which DST models are trained only on session-level annotations but evaluated at the turn level. We further report detailed analyses to demonstrate the effectiveness of graph models in DST by showing that the proposed graph modules capture inter-slot dependencies and improve the predictions of values that are common to multiple domains.
Anthology ID:
2021.emnlp-main.620
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:
7871–7881
Language:
URL:
https://preview.aclanthology.org/build-pipeline-with-new-library/2021.emnlp-main.620/
DOI:
10.18653/v1/2021.emnlp-main.620
Bibkey:
Cite (ACL):
Weizhe Lin, Bo-Hsiang Tseng, and Bill Byrne. 2021. Knowledge-Aware Graph-Enhanced GPT-2 for Dialogue State Tracking. In Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, pages 7871–7881, Online and Punta Cana, Dominican Republic. Association for Computational Linguistics.
Cite (Informal):
Knowledge-Aware Graph-Enhanced GPT-2 for Dialogue State Tracking (Lin et al., EMNLP 2021)
Copy Citation:
PDF:
https://preview.aclanthology.org/build-pipeline-with-new-library/2021.emnlp-main.620.pdf
Video:
 https://preview.aclanthology.org/build-pipeline-with-new-library/2021.emnlp-main.620.mp4
Code
 linweizhedragon/knowledge-aware-graph-enhanced-gpt-2-for-dialogue-state-tracking
Data
MultiWOZ