{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/augpt-dialogue-with-pre-trained-language","title":"AuGPT: Auxiliary Tasks and Data Augmentation for End-To-End Dialogue with Pre-Trained Language Models","arxiv_id":"2102.05126","date":"2021-02-09","proceeding":"EMNLP (NLP4ConvAI) 2021 11","authors":["Jonáš Kulhánek","Vojtěch Hudeček","Tomáš Nekvinda","Ondřej Dušek"],"abstract":"Attention-based pre-trained language models such as GPT-2 brought considerable progress to end-to-end dialogue modelling. However, they also present considerable risks for task-oriented dialogue, such as lack of knowledge grounding or diversity. To address these issues, we introduce modified training objectives for language model finetuning, and we employ massive data augmentation via back-translation to increase the diversity of the training data. We further examine the possibilities of combining data from multiples sources to improve performance on the target dataset. We carefully evaluate our contributions with both human and automatic methods. Our model substantially outperforms the baseline on the MultiWOZ data and shows competitive performance with state of the art in both automatic and human evaluation.","url_abs":"https://arxiv.org/abs/2102.05126v3","url_pdf":"https://arxiv.org/pdf/2102.05126v3.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"augpt-dialogue-with-pre-trained-language","repo_url":"https://github.com/ufal/augpt","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":{"status":"unanswered"}}],"tasks":[{"task_slug":"diversity","task_name":"Diversity"},{"task_slug":"end-to-end-dialogue-modelling","task_name":"End-To-End Dialogue Modelling"},{"task_slug":"language-modeling","task_name":"Language Modeling"},{"task_slug":"translation","task_name":"Translation"}],"methods":[{"method_slug":"adam","method_name":"Adam"},{"method_slug":"attention","method_name":"Attention"},{"method_slug":"attention-dropout","method_name":"Attention Dropout"},{"method_slug":"bpe","method_name":"BPE"},{"method_slug":"cosine-annealing","method_name":"Cosine Annealing"},{"method_slug":"dense-connections","method_name":"Dense Connections"},{"method_slug":"discriminative-fine-tuning","method_name":"Discriminative Fine-Tuning"},{"method_slug":"dropout","method_name":"Dropout"},{"method_slug":"gpt-2","method_name":"GPT-2"},{"method_slug":"layer-normalization","method_name":"Layer Normalization"},{"method_slug":"linear-layer","method_name":"Linear Layer"},{"method_slug":"linear-warmup-with-cosine-annealing","method_name":"Linear Warmup With Cosine Annealing"},{"method_slug":"multi-head-attention","method_name":"Multi-Head Attention"},{"method_slug":"residual-connection","method_name":"Residual Connection"},{"method_slug":"softmax","method_name":"Softmax"},{"method_slug":"weight-decay","method_name":"Weight Decay"}],"datasets_introduced":[],"methods_introduced":[],"results":[{"leaderboard":"/sota/end-to-end-dialogue-modelling-on-multiwoz-2-0","task":"End-To-End Dialogue Modelling","dataset":"MULTIWOZ 2.0","model":"AuGPT","rank_in_archive_order":3,"of":6,"metrics":{"BLEU":"17.2","MultiWOZ (Inform)":"90.2","MultiWOZ (Success)":"75.5"},"uses_additional_data":true},{"leaderboard":"/sota/end-to-end-dialogue-modelling-on-multiwoz-2-1","task":"End-To-End Dialogue Modelling","dataset":"MULTIWOZ 2.1","model":"AuGPT","rank_in_archive_order":3,"of":4,"metrics":{"BLEU":"17.2","MultiWOZ (Inform)":"91.4","MultiWOZ (Success)":"72.9"},"uses_additional_data":true}],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2102.05126","mcp":null,"developers":"https://syntology.ai/developers"},"arxiv_metadata":null,"syntology_extracted_results":null}