UBAR: Towards Fully End-to-End Task-Oriented Dialog System with GPT-2


  • Yunyi Yang Sun Yat-sen University
  • Yunhao Li Sun Yat-sen University
  • Xiaojun Quan Sun Yat-sen University


Conversational AI/Dialog Systems


This paper presents our task-oriented dialog system UBAR which models task-oriented dialogs on a dialog session level. Specifically, UBAR is acquired by fine-tuning the large pre-trained unidirectional language model GPT-2 on the sequence of the entire dialog session which is composed of user utterance, belief state, database result, system act, and system response of every dialog turn. Additionally, UBAR is evaluated in a more realistic setting, where its dialog context has access to user utterances and all content it generated such as belief states, system acts, and system responses. Experimental results on the MultiWOZ datasets show that UBAR achieves state-of-the-art performances in multiple settings, improving the combined score of response generation, policy optimization, and end-to-end modeling by 4.7, 3.5, and 9.4 points respectively. Thorough analyses demonstrate that the session-level training sequence formulation and the generated dialog context are essential for UBAR to operate as a fully end-to-end task-oriented dialog system in real life. We also examine the transfer ability of UBAR to new domains with limited data and provide visualization and a case study to illustrate the advantages of UBAR in modeling on a dialog session level.




How to Cite

Yang, Y., Li, Y., & Quan, X. (2021). UBAR: Towards Fully End-to-End Task-Oriented Dialog System with GPT-2. Proceedings of the AAAI Conference on Artificial Intelligence, 35(16), 14230-14238. Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/17674



AAAI Technical Track on Speech and Natural Language Processing III