Prompt-Guided Dialogue State Tracking with GPT-2 and Graph Attention
Source abstract
Abstract
Dialogue State Tracking (DST) is a critical component of task-oriented spoken dialogue systems (SDS), tasked with maintaining an accurate representation of the conversational state by predicting slots and their corresponding values. Recent advances leverage Large Language Models (LLMs) with prompt-based tuning to improve tracking accuracy and efficiency. However, these approaches often incur substantial computational and memory overheads and typically address slot extraction implicitly within prompts, without explicitly modeling the complex dependencies between slots and values. In this work, we propose PUGG, a novel DST framework that constructs schema-driven prompts to fine-tune GPT-2 and utilizes its tokenizer to implement a memory encoder. PUGG explicitly extracts slot values via GPT-2 and employs Graph Attention Networks (GATs) to model and reason over the intricate relationships between slots and their associated values. We evaluate PUGG on four publicly available datasets, where it achieves state-of-the-art performance across multiple evaluation metrics, highlighting its robustness and generalizability in diverse conversational scenarios. Our results indicate that the integration of GPT-2 substantially reduces model complexity and memory consumption by streamlining key processes. Moreover, prompt tuning enhances the model’s flexibility and precision in extracting relevant slot-value pairs, while the incorporation of GATs facilitates effective relational reasoning, leading to improved dialogue state representations.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Cognitive Functions and Memory · Context-Aware Activity Recognition Systems · Topic Modeling
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (56)
Show source-provided references
1. MultiWOZ 2.1: a consolidated multi-domain dialogue dataset with state corrections and state tracking baselines
2. Kim · Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics · 2020 · 56710.18653/v1/2020.acl-main.53 ↗
3. “Do you follow me?”: a survey of recent approaches in dialogue state tracking
4. Ouyang · Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics · 2020 · 3410.18653/v1/2020.acl-main.5 ↗
5. Joint intent detection and slot filling using weighted finite state transducer and BERT10.1007/s10489-022-03295-9 ↗
6. MetaASSIST: robust dialogue state tracking with meta learning
7. Cross-domain coreference modeling in dialogue state tracking with prompt learning10.1016/j.knosys.2023.111189 ↗
8. Enhanced multi-domain dialogue state tracker with second-order slot interactions10.1109/TASLP.2022.3221044 ↗
9. Global-locally self-attentive encoder for dialogue state tracking
10. ASSIST: towards label noise-robust dialogue state tracking
11. InstructTODS: large language models for end-to-end task-oriented dialogue systems
12. Towards LLM-driven dialogue state tracking
13. RSP-DST: revisable state prediction for dialogue state tracking10.3390/electronics12061494 ↗
14. Li Z, Chen ZZ, Ross M, Huber P, Moon S, Lin Z, et al. Large language models as zero-shot dialogue state tracker through function calling. arXiv:2402.10466. 2024.10.18653/v1/2024.acl-long.471 ↗
15. Niu C, Wang X, Cheng X, Song J, Zhang T. Enhancing dialogue state tracking models through LLM-backed user-agents simulation. arXiv:2405.13037. 2024.10.18653/v1/2024.acl-long.473 ↗
16. ECDG-DST: a dialogue state tracking model based on efficient context and domain guidance for smart dialogue systems10.1016/j.csl.2024.101741 ↗
17. Multi-domain dialogue state tracking via dual dynamic graph with hierarchical slot selector10.1016/j.knosys.2024.112754 ↗
18. Ma MD, Kao JY, Gao S, Gupta A, Jin D, Chung T, et al. Parameter-efficient low-resource dialogue state tracking by prompt tuning. arXiv:2301.10915. 2023.10.21437/Interspeech.2023-2238 ↗
19. UTMGAT: a unified transformer with memory encoder and graph attention networks for multidomain dialogue state tracking10.1007/s10489-024-05571-2 ↗
20. Saha D, Santra B, Goyal P. A study on prompt-based few-shot learning methods for belief state tracking in task-oriented dialog systems. arXiv:2204.08167. 2022.
21. A Dual prompt learning framework for few-shot dialogue state tracking
22. Continual prompt tuning for dialog state tracking
23. DSTEA: improving dialogue state tracking via entity adaptive pre-training10.1016/j.knosys.2024.111542 ↗
24. Slot self-attentive dialogue state tracking
25. A multi-attention approach using BERT and stacked bidirectional LSTM for improved dialogue state tracking10.3390/app13031775 ↗
26. Extrinsic evaluation of dialog state tracking and predictive metrics for dialog policy optimization
27. The SJTU system for dialog state tracking challenge 2
28. POMDP-based statistical spoken dialog systems: a review10.1109/JPROC.2012.2225812 ↗
29. Bayesian update of dialogue state: a POMDP framework for spoken dialogue systems10.1016/j.csl.2009.07.003 ↗
30. Dialog state tracking using conditional random fields
31. Noroozi V, Zhang Y, Bakhturina E, Kornuta T. A fast and robust BERT-based dialogue state tracker for schema-guided dialogue dataset. arXiv:2008.12335. 2020.
32. Knowledge-grounded dialogue modelling with dialogue-state tracking, domain tracking, and entity extraction10.1016/j.csl.2022.101460 ↗
33. Hu · Findings of the Association for Computational · 2022 · 2627
34. MA-DST: multi-attention-based scalable dialog state tracking
35. Mou X, Sigouin B, Steenstra I, Su H. Multimodal dialogue state tracking by QA approach with data augmentation. arXiv:2007.09903. 2020.
36. T2TD: text-3D generation model based on prior knowledge guidance10.1109/TPAMI.2024.3463753 ↗
37. Can foundation language models predict fluid dynamics?10.1016/j.engappai.2025.111427 ↗
38. Fg-T2M++: LLMs-augmented fine-grained text driven human motion generation10.1007/s11263-025-02392-9 ↗
39. Cross-lingual font style transfer with full-domain convolutional attention10.1016/j.patcog.2024.110709 ↗
40. Velickovic P, Cucurull G, Casanova A, Romero A, Lio P, Bengio Y, et al. Graph attention networks. arXiv:1710.10903. 2017.
41. Knowledge-aware graph-enhanced GPT-2 for dialogue state tracking
42. Enriching the dialogue state tracking model with a asyntactic discourse graph10.1016/j.patrec.2023.03.024 ↗
43. Efficient context and schema fusion networks for multi-domain dialogue state tracking
44. Multi-task learning with graph attention networks for multi-domain task-oriented dialogue systems10.1016/j.knosys.2022.110069 ↗
45. MGCRL: multi-view graph convolution and multi-agent reinforcement learning for dialogue state tracking10.1007/s00521-023-09328-9 ↗
46. Zang · Proceedings of the 2nd Workshop on Natural Language Processing for Conversational AI · 2020 · 10910.18653/v1/2020.nlp4convai-1.13 ↗
47. Rastogi A, Zang X, Sunkara S, Gupta R, Khaitan P. Schema-guided dialogue state tracking task at DSTC8. arXiv:2002.01359. 2020.
48. Language models are unsupervised multitask learners
49. Accuth+: accelerometer-based anti-spoofing voice authentication on wrist-worn wearables10.1109/TMC.2023.3314837 ↗
50. Fpa-GCN: enhancing aspect sentiment triplet extraction with feature-rich prediction-aware graph convolutional networks10.1007/s10489-025-06313-8 ↗
51. MultiWOZ—a large-scale multi-domain wizard-of-oz dataset for task-oriented dialogue modelling
52. MultiWOZ 2.4: a multi-domain task-oriented dialogue dataset with essential annotation corrections to improve state tracking evaluation
53. Kingma · 3rd International Conference on Learning Representations, ICLR · 2015 May 7–9
54. Dynamic multi-granularity translation system: dag-structured multi-granularity representation and self-attention10.3390/systems12100420 ↗
55. Dual-microphone active noise cancellation paved with Doppler assimilation for TADS10.1016/j.ymssp.2022.109727 ↗
56. AttriDiffuser: adversarially enhanced diffusion model for text-to-facial attribute image synthesis10.1016/j.patcog.2025.111447 ↗