TOD-BERT: Pre-trained Natural Language Understanding for Task-Oriented Dialogue
Not on the map. Why?
Abstract (arXiv)
The underlying difference of linguistic patterns between general text and task-oriented dialogue makes existing pre-trained language models less useful in practice. In this work, we unify nine human-human and multi-turn task-oriented dialogue datasets for language modeling. To better model dialogue behavior during pre-training, we incorporate user and system tokens into the masked language modeling. We propose a contrastive objective function to simulate the response selection task. Our pre-trained task-oriented dialogue BERT (TOD-BERT) outperforms strong baselines like BERT on four downstream task-oriented dialogue applications, including intention recognition, dialogue state tracking, dialogue act prediction, and response selection. We also show that TOD-BERT has a stronger few-shot ability that can mitigate the data scarcity problem for task-oriented dialogue.
Key facts. Cited by 185 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Apr 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 48; 2022: 59; 2023: 39; 2024: 18; 2025: 12; 2026: 9.
185 papers.
Sort:
By area:
- 2021Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue Systemcited by 139
- 2021AMMUS : A Survey of Transformer-based Pretrained Models in Natural Language Processingcited by 131
- 2021Recent Advances in Deep Learning Based Dialogue Systems: A Systematic Surveycited by 130
- 2023Prompting and Evaluating Large Language Models for Proactive Dialogues: Clarification, Target-guided, and Non-collaborationcited by 115
- 2021GALAXY: A Generative Pre-trained Model for Task-Oriented Dialog with Semi-Supervised Learning and Explicit Policy Injectioncited by 102
- 2021Dialogue State Tracking with a Language Model using Schema-Driven Promptingcited by 94
- 2022META-GUI: Towards Multi-modal Conversational Agents on Mobile GUIcited by 81
- 2021DialogLM: Pre-trained Model for Long Dialogue Understanding and Summarizationcited by 72
- 2021Leveraging Slot Descriptions for Zero-Shot Cross-Domain Dialogue State Trackingcited by 61
- 2021Zero-Shot Dialogue State Tracking via Cross-Task Transfercited by 52
- 2021Few-Shot Intent Detection via Contrastive Pre-Training and Fine-Tuningcited by 46
- 2022NoisyTune: A Little Noise Can Help You Finetune Pretrained Language Models Bettercited by 43
38 more
- 2022Controllable Dialogue Simulation with In-Context Learningcited by 42
- 2022InstructDial: Improving Zero and Few-shot Generalization in Dialogue through Instruction Tuningcited by 39
- 2022SPACE-3: Unified Dialog Model Pre-training for Task-Oriented Dialog Understanding and Generationcited by 34
- 2021AuGPT: Auxiliary Tasks and Data Augmentation for End-To-End Dialogue with Pre-Trained Language Modelscited by 31
- 2021CINS: Comprehensive Instruction for Few-shot Learning in Task-oriented Dialog Systemscited by 30
- 2022Multi2WOZ: A Robust Multilingual Dataset and Conversational Pretraining for Task-Oriented Dialogcited by 30
- 2023Task-Optimized Adapters for an End-to-End Task-Oriented Dialogue Systemcited by 30
- 2022New Intent Discovery with Pre-training and Contrastive Learningcited by 28
- 2022SPACE-2: Tree-Structured Semi-Supervised Contrastive Pre-training for Task-Oriented Dialog Understandingcited by 28
- 2021Slot Self-Attentive Dialogue State Trackingcited by 27
- 2021Controllable Abstractive Dialogue Summarization with Sketch Supervisioncited by 27
- 2021Effectiveness of Pre-training for Few-shot Intent Classificationcited by 27
- 2021ConvFiT: Conversational Fine-Tuning of Pretrained Language Modelscited by 25
- 2021Pretraining the Noisy Channel Model for Task-Oriented Dialoguecited by 23
- 2021Are Pretrained Transformers Robust in Intent Classification? A Missing Ingredient in Evaluation of Out-of-Scope Intent Detectioncited by 22
- 2022"Do you follow me?": A Survey of Recent Approaches in Dialogue State Trackingcited by 22
- 2021DS-TOD: Efficient Domain Specialization for Task Oriented Dialogcited by 21
- 2021NER-BERT: A Pre-trained Model for Low-Resource Entity Taggingcited by 21
- 2022Dialogue Summaries as Dialogue States (DS2), Template-Guided Summarization for Few-shot Dialogue State Trackingcited by 21
- 2021Knowledge-Aware Graph-Enhanced GPT-2 for Dialogue State Trackingcited by 20
- 2021ReMeDi: Resources for Multi-domain, Multi-service, Medical Dialoguescited by 20
- 2022BORT: Back and Denoising Reconstruction for End-to-End Task-Oriented Dialogcited by 20
- 2022Fine-tuning Pre-trained Language Models for Few-shot Intent Detection: Supervised Pre-training and Isotropizationcited by 19
- 2021Self-training Improves Pre-training for Few-shot Learning in Task-oriented Dialog Systemscited by 18
- 2021QAConv: Question Answering on Informative Conversationscited by 17
- 2022Learning Dialogue Representations from Consecutive Utterancescited by 17
- 2023Towards Robust Personalized Dialogue Generation via Order-Insensitive Representation Regularizationcited by 17
- 2024SynthDST: Synthetic Data is All You Need for Few-Shot Dialog State Trackingcited by 17
- 2021Crossing the Conversational Chasm: A Primer on Natural Language Processing for Multilingual Task-Oriented Dialogue Systemscited by 16
- 2022Multimodal Conversational AI: A Survey of Datasets and Approachescited by 16
- 2022Robust Dialogue State Tracking with Weak Supervision and Sparse Datacited by 15
- 2022Alexa Teacher Model: Pretraining and Distilling Multi-Billion-Parameter Encoders for Natural Language Understanding Systemscited by 15
- 2022A Survey of Intent Classification and Slot-Filling Datasets for Task-Oriented Dialogcited by 14
- 2022Task Compass: Scaling Multi-task Pre-training with Task Prefixcited by 14
- 2021Automated Quality Assessment of Cognitive Behavioral Therapy Sessions Through Highly Contextualized Language Representationscited by 13
- 2021Advances in Multi-turn Dialogue Comprehension: A Surveycited by 12
- 2021Task-adaptive Pre-training and Self-training are Complementary for Natural Language Understandingcited by 12
- 2021Response Generation with Context-Aware Prompt Learningcited by 12
50 most cited of 185 All 185 (JSON)
- 2026Lost with a Map: Conversational State and Behavioral Reliability in Language Models27 Sep 2026
- 2026Leveraging Speech Acts for Low-Data and Cross-Domain Conversation Derailment Forecasting26 Aug 2026
- 2026Multi-turn Conversational AI from Text to Multimodal Interaction: Data, Models, Evaluation, and Open Challenges18 Aug 2026
- 2026Enabling Proactive Spoken Turns via a Generalized Style-Aware Full-Duplex Framework4 Aug 2026
- 2026TRACER: Early Failure Detection for Task-Oriented Dialogue4 Jul 2026
- 2026Stateful Guardrails for Multi-Turn LLM Systems: A Conversational Risk Accumulation Framework3 Jun 2026
- 2026CobSeg: Coherence Boundary Modeling for Dialogue Topic Segmentation29 May 2026
- 2026Dynamic Knowledge Fusion for Multi-Domain Dialogue State Tracking11 Mar 2026
- 2026ESAinsTOD: A Unified End-to-End Schema-Aware Instruction-Tuning Framework for Task-Oriented Dialog Modeling10 Mar 2026
- 2025"Mm, Wat?" Detecting Other-initiated Repair Requests in Dialogue28 Oct 2025
- 2025Multimodal Fine-grained Context Interaction Graph Modeling for Conversational Speech Synthesis7 Sep 2025
- 2025Few-Shot Query Intent Detection via Relation-Aware Prompt Learning6 Sep 2025
38 more
- 2025On the Effectiveness of Integration Methods for Multimodal Dialogue Response Retrieval13 Jun 2025
- 2025FlowKV: Enhancing Multi-Turn Conversational Coherence in LLMs via Isolated Key-Value Cache Management21 May 2025
- 2025A Multi-view Discourse Framework for Integrating Semantic and Syntactic Features in Dialog Agents12 Apr 2025
- 2025A-MESS: Anchor based Multimodal Embedding with Semantic Synchronization for Multimodal Intent Recognition25 Mar 2025
- 2025GRITHopper: Decomposition-Free Multi-Hop Dense Retrieval10 Mar 2025
- 2025PFDial: A Structured Dialogue Instruction Fine-tuning Method Based on UML Flowcharts9 Mar 2025
- 2025Learning LLM Preference over Intra-Dialogue Pairs: A Framework for Utterance-level Understandings7 Mar 2025
- 2025GuideLLM: Exploring LLM-Guided Conversation with Applications in Autobiography Interviewing10 Feb 2025
- 2025Wizard of Shopping: Target-Oriented E-commerce Dialogue Generation with Decision Tree Branching3 Feb 2025
- 2024Improved intent classification based on context information using a windows-based approach9 Nov 2024
- 2024Dialog2Flow: Pre-training Soft-Contrastive Action-Driven Sentence Embeddings for Automatic Dialog Flow Extraction24 Oct 2024
- 2024Emphasis Rendering for Conversational Text-to-Speech with Multi-modal Multi-scale Context Modeling12 Oct 2024
- 2024Survey: Transformer-based Models in Data Modality Conversion8 Aug 2024
- 2024TransferTOD: A Generalizable Chinese Multi-Domain Task-Oriented Dialogue System with Transfer Capabilities31 Jul 2024
- 2024Multi-turn Response Selection with Commonsense-enhanced Language Models26 Jul 2024
- 2024News Without Borders: Domain Adaptation of Multilingual Sentence Embeddings for Cross-lingual News Recommendation18 Jun 2024
- 2024Plan, Generate and Complicate: Improving Low-resource Dialogue State Tracking via Easy-to-Difficult Zero-shot Data Augmentation13 Jun 2024
- 2024Comparing Data Augmentation Methods for End-to-End Task-Oriented Dialog Systems10 Jun 2024
- 2024Scaling Laws for Discriminative Classification in Large Language Models24 May 2024
- 2024Red Teaming Language Models for Processing Contradictory Dialogues16 May 2024
- 2024Context Does Matter: Implications for Crowdsourced Evaluation Labels in Task-Oriented Dialogue Systems15 Apr 2024
- 2024DivTOD: Unleashing the Power of LLMs for Diversifying Task-Oriented Dialogue Representations31 Mar 2024
- 2024BootTOD: Bootstrap Task-oriented Dialogue Representations by Aligning Diverse Responses2 Mar 2024
- 2024UniRetriever: Multi-task Candidates Selection for Various Context-Adaptive Conversational Retrieval26 Feb 2024
- 2024SynthDST: Synthetic Data is All You Need for Few-Shot Dialog State Tracking3 Feb 2024
- 2024Local and Global Contexts for Conversation31 Jan 2024
- 2024State Value Generation with Prompt Learning and Self-Training for Low-Resource Dialogue State Tracking30 Jan 2024
- 2023OmniDialog: An Omnipotent Pre-training Model for Task-Oriented Dialogue System28 Dec 2023
- 2023TOD-Flow: Modeling the Structure of Task-Oriented Dialogues7 Dec 2023
- 2023Tracing Influence at Scale: A Contrastive Learning Approach to Linking Public Comments and Regulator Responses24 Nov 2023
- 2023Multi-Step Dialogue Workflow Action Prediction16 Nov 2023
- 2023URL-BERT: Training Webpage Representations via Social Media Engagements25 Oct 2023
- 2023Turn-Level Active Learning for Dialogue State Tracking23 Oct 2023
- 2023A Survey on Semantic Processing Techniques22 Oct 2023
- 2023UNO-DST: Leveraging Unlabelled Data in Zero-Shot Dialogue State Tracking16 Oct 2023
- 2023RSVP: Customer Intent Detection via Agent Response Contrastive and Generative Pre-Training15 Oct 2023
- 2023HierarchicalContrast: A Coarse-to-Fine Contrastive Learning Framework for Cross-Domain Zero-Shot Slot Filling13 Oct 2023
- 2023A Task-oriented Dialog Model with Task-progressive and Policy-aware Pre-training1 Oct 2023
50 newest of 185 All 185 (JSON)
- 2021Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue Systemcited by 139
- 2021AMMUS : A Survey of Transformer-based Pretrained Models in Natural Language Processingcited by 131
- 2021Recent Advances in Deep Learning Based Dialogue Systems: A Systematic Surveycited by 130
- 2023Prompting and Evaluating Large Language Models for Proactive Dialogues: Clarification, Target-guided, and Non-collaborationcited by 115
- 2021GALAXY: A Generative Pre-trained Model for Task-Oriented Dialog with Semi-Supervised Learning and Explicit Policy Injectioncited by 102
- 2021Dialogue State Tracking with a Language Model using Schema-Driven Promptingcited by 94
- 2022META-GUI: Towards Multi-modal Conversational Agents on Mobile GUIcited by 81
- 2021DialogLM: Pre-trained Model for Long Dialogue Understanding and Summarizationcited by 72
- 2021Leveraging Slot Descriptions for Zero-Shot Cross-Domain Dialogue State Trackingcited by 61
- 2021Zero-Shot Dialogue State Tracking via Cross-Task Transfercited by 52
- 2021Few-Shot Intent Detection via Contrastive Pre-Training and Fine-Tuningcited by 46
- 2022NoisyTune: A Little Noise Can Help You Finetune Pretrained Language Models Bettercited by 43
12 most cited of 177 All 177 (JSON)
- 2026Lost with a Map: Conversational State and Behavioral Reliability in Language Models27 Sep 2026
- 2026Leveraging Speech Acts for Low-Data and Cross-Domain Conversation Derailment Forecasting26 Aug 2026
- 2026Multi-turn Conversational AI from Text to Multimodal Interaction: Data, Models, Evaluation, and Open Challenges18 Aug 2026
- 2026Enabling Proactive Spoken Turns via a Generalized Style-Aware Full-Duplex Framework4 Aug 2026
- 2026TRACER: Early Failure Detection for Task-Oriented Dialogue4 Jul 2026
- 2026CobSeg: Coherence Boundary Modeling for Dialogue Topic Segmentation29 May 2026
- 2026Dynamic Knowledge Fusion for Multi-Domain Dialogue State Tracking11 Mar 2026
- 2026ESAinsTOD: A Unified End-to-End Schema-Aware Instruction-Tuning Framework for Task-Oriented Dialog Modeling10 Mar 2026
- 2025"Mm, Wat?" Detecting Other-initiated Repair Requests in Dialogue28 Oct 2025
- 2025Multimodal Fine-grained Context Interaction Graph Modeling for Conversational Speech Synthesis7 Sep 2025
- 2025Few-Shot Query Intent Detection via Relation-Aware Prompt Learning6 Sep 2025
- 2025On the Effectiveness of Integration Methods for Multimodal Dialogue Response Retrieval13 Jun 2025
12 newest of 177 All 177 (JSON)