Stabilized In-Context Learning with Pre-trained Language Models for Few Shot Dialogue State Tracking

Derek Chen, Kun Qian, Zhou Yu


Abstract
Prompt-based methods with large pre-trained language models (PLMs) have shown impressive unaided performance across many NLP tasks. These models improve even further with the addition of a few labeled in-context exemplars to guide output generation. However, for more complex tasks such as dialogue state tracking (DST), designing prompts that reliably convey the desired intent is nontrivial, leading to unstable results. Furthermore, building in-context exemplars for dialogue tasks is difficult because conversational contexts are long while model input lengths are relatively short. To overcome these issues we first adapt a meta-learning scheme to the dialogue domain which stabilizes the ability of the model to perform well under various prompts. We additionally design a novel training method to improve upon vanilla retrieval mechanisms to find ideal in-context examples. Finally, we introduce a saliency model to limit dialogue text length, allowing us to include more exemplars per query. In effect, we are able to achieve highly competitive results for few-shot DST on MultiWOZ.
Anthology ID:
2023.findings-eacl.115
Volume:
Findings of the Association for Computational Linguistics: EACL 2023
Month:
May
Year:
2023
Address:
Dubrovnik, Croatia
Editors:
Andreas Vlachos, Isabelle Augenstein
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1551–1564
Language:
URL:
https://aclanthology.org/2023.findings-eacl.115
DOI:
10.18653/v1/2023.findings-eacl.115
Bibkey:
Cite (ACL):
Derek Chen, Kun Qian, and Zhou Yu. 2023. Stabilized In-Context Learning with Pre-trained Language Models for Few Shot Dialogue State Tracking. In Findings of the Association for Computational Linguistics: EACL 2023, pages 1551–1564, Dubrovnik, Croatia. Association for Computational Linguistics.
Cite (Informal):
Stabilized In-Context Learning with Pre-trained Language Models for Few Shot Dialogue State Tracking (Chen et al., Findings 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.findings-eacl.115.pdf
Video:
 https://aclanthology.org/2023.findings-eacl.115.mp4