Unsupervised Learning of KB Queries in Task-Oriented Dialogs

Dinesh Raghu, Nikhil Gupta, Mausam


Abstract
Task-oriented dialog (TOD) systems often need to formulate knowledge base (KB) queries corresponding to the user intent and use the query results to generate system responses. Existing approaches require dialog datasets to explicitly annotate these KB queries—these annotations can be time consuming, and expensive. In response, we define the novel problems of predicting the KB query and training the dialog agent, without explicit KB query annotation. For query prediction, we propose a reinforcement learning (RL) baseline, which rewards the generation of those queries whose KB results cover the entities mentioned in subsequent dialog. Further analysis reveals that correlation among query attributes in KB can significantly confuse memory augmented policy optimization (MAPO), an existing state of the art RL agent. To address this, we improve the MAPO baseline with simple but important modifications suited to our task. To train the full TOD system for our setting, we propose a pipelined approach: it independently predicts when to make a KB query (query position predictor), then predicts a KB query at the predicted position (query predictor), and uses the results of predicted query in subsequent dialog (next response predictor). Overall, our work proposes first solutions to our novel problem, and our analysis highlights the research challenges in training TOD systems without query annotation.
Anthology ID:
2021.tacl-1.23
Volume:
Transactions of the Association for Computational Linguistics, Volume 9
Month:
Year:
2021
Address:
Cambridge, MA
Editors:
Brian Roark, Ani Nenkova
Venue:
TACL
SIG:
Publisher:
MIT Press
Note:
Pages:
374–390
Language:
URL:
https://aclanthology.org/2021.tacl-1.23
DOI:
10.1162/tacl_a_00372
Bibkey:
Cite (ACL):
Dinesh Raghu, Nikhil Gupta, and Mausam. 2021. Unsupervised Learning of KB Queries in Task-Oriented Dialogs. Transactions of the Association for Computational Linguistics, 9:374–390.
Cite (Informal):
Unsupervised Learning of KB Queries in Task-Oriented Dialogs (Raghu et al., TACL 2021)
Copy Citation:
PDF:
https://aclanthology.org/2021.tacl-1.23.pdf
Video:
 https://aclanthology.org/2021.tacl-1.23.mp4