This is the PyTorch implementation of the paper: Non-Autoregressive Dialog State Tracking. Hung Le, Richard Socher, Steven C.H. Hoi. ICLR 2020. (arXiv)
This code has been written using PyTorch 1.0.1. If you find the paper or the source code useful to your projects, please cite the following bibtex:
@inproceedings{
Le2020Non-Autoregressive,
title={Non-Autoregressive Dialog State Tracking},
author={Hung Le and Richard Socher and Steven C.H. Hoi},
booktitle={International Conference on Learning Representations},
year={2020},
url={https://openreview.net/forum?id=H1e_cC4twS}
}
Recent efforts in Dialogue State Tracking (DST) for task-oriented dialogues have progressed toward open-vocabulary or generation-based approaches where the models can generate slot value candidates from the dialogue history itself. These approaches have shown good performance gain, especially in complicated dialogue domains with dynamic slot values. However, they fall short in two aspects: (1) they do not allow models to explicitly learn signals across domains and slots to detect potential dependencies among \textit{(domain, slot)} pairs; and (2) existing models follow auto-regressive approaches which incur high time cost when the dialogue evolves over multiple domains and multiple turns. In this paper, we propose a novel framework of Non-Autoregressive Dialog State Tracking (NADST) which can factor in potential dependencies among domains and slots to optimize the models towards better prediction of dialogue states as a complete set rather than separate slots. In particular, the non-autoregressive nature of our method not only enables decoding in parallel to significantly reduce the latency of DST for real-time dialogue response generation, but also detect dependencies among slots at token level in addition to slot and domain level. Our empirical results show that our model achieves the state-of-the-art joint accuracy across all domains on the MultiWOZ 2.1 corpus, and the latency of our model is an order of magnitude lower than the previous state of the art as the dialogue history extends over time.
A sample task-oriented dialogue with annotated dialogue states after each user turn. The dialogue states in red and blue denote slots from the attraction domain and train domain respectively. Slot values are expressed in user and system utterances (highlighted by underlined text).
Our NADST has 3 key components: encoders (red), fertility decoder (blue), and state decoder (green).
(i) Encoders encode sequences of dialogue history, delexicalized dialogue history, and domain and slot tokens into continuous representations; (ii) Fertility Decoder has 3 attention mechanisms to learn potential dependencies across \textit{(domain, slot)} pairs in combination with dialogue history. The output is used to generate fertilities and slot gates; and (iii) State Decoder receives the input sequence including sub-sequences of `(domain, slot) x fertility` to decode a complete dialogue state sequence as concatenation of component slot values. For simplicity, we do not show feedforward, residual connection, and layer-normalization layers in the figure. Best viewed in color.
We use the MultiWOZ benchmark, including both version 2.0 (Link) and 2.1 (Link). Download the data and unzip into the root directory of the repo e.g. NADST/data2.0 and NADST/data2.1.
The data includes the original and pre-processed MultiWOZ dialogues with delexicalized system utterances and user utterances. The pre-processed dialogues are prefixed with nadst_.
We created scripts/run.sh to prepare evaluation code, train models, generate dialogue states, and evaluating the generated states with automatic metrics. You can directly run this file which includes example parameter setting:
| Parameter | Description | Values |
|---|---|---|
| path | path where the model and training logs will be saved | e.g. save_temp/nadst_model |
| d | the embedding dimension to embed word tokens into continuous representations | e.g. 256 |
| h_attn | the number of attention heads applied in all attention blocks | e.g. 16 |
| fert_dec_N | Number of layers/attention steps in fertility decoder | e.g. 3 |
| state_dec_N | Number of layers/attention steps in state decoder | e.g. 3 |
| bsz | Number of instances in each training batch | e.g. 32 |
| wu | Number of warmup steps to increase the learning rate | 20000 |
| dr | Dropout rate during training | e.g. 0.2 |
| dv | The data version of the MultiWOZ benchmark | either 2.0 or 2.1 |
While training, the model with the best validation is saved. The model is evaluated by using the losses from gate prediction, fertility prediction, and state generation.
The model output, parameters, vocabulary, and training and validation logs will be save into folder determined in the path parameter.
Other parameters, including data-related options, model parameters, training and generating settings, are defined in the file utils/config.py.
An example of pretrained NADST model using the sampled script in scripts/run.sh can be downloaded here. Unzip the download file and update the path parameter in the test command in the scripts/run.sh to the corresponding unzip directory e.g. save/pretrained_nadst. Using the pretrained model, the test script provides the following results:
| Joint Acc | Slot Acc | F1 | |
|---|---|---|---|
| Use predicted fertility/gate | 48.25% | 97.24% | 0.8858 |
| Use oracle fertility/gate | 70.64% | 98.86% | 0.9459 |
Heatmap visualization of self-attention scores of 5 heads between Z_{ds×fert} representations in the state decoder. The corresponding prediction output for each representation is presented on the right side. The examples are for the 6th turn in dialogue ID MUL0536 (upper row) and PMUL3759 (lower row) in MultiWOZ2.1.
Example predicted dialogue states for dialogue ID MUL0536 in MultiWOZ2.1