arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.02468cs.CL
收藏

面向端到端任务导向对话系统的任务优化适配器

Task-Optimized Adapters for an End-to-End Task-Oriented Dialogue System

Sogang University(西江大学)

Namo Bang, Jeehyun Lee, Myoung-Wan Koo

AI总结:

本文提出一种带有任务优化适配器的端到端任务导向对话系统,通过每任务独立学习少量参数并结合强化学习增强DST与NLG模块,在MultiWOZ基准上取得竞争性性能,并在DST任务上达到最先进水平。

摘要:

任务导向对话(TOD)系统旨在通过跟踪对话状态并生成适当的回复来执行特定任务,以帮助用户实现既定目标。近年来,基于大规模数据集预训练的端到端对话模型在对话系统中展现出了良好的性能。然而,这些模型在训练对话系统的各个任务(NLU、DST、NLG)时共享相同的参数,因此对每个任务进行调试十分困难。此外,为了创建任务导向的聊天机器人,需要对大量参数进行微调,这需要耗费大量精力,使得非专家用户难以处理。因此,我们打算训练与预训练语言模型(PLM)相比相对轻量且快速的模型。在本文中,我们提出了一种带有任务优化适配器(Task-Optimized Adapters)的端到端TOD系统,这些适配器针对每个任务独立学习,仅在预训练网络的固定层之后添加少量参数。我们还通过强化学习增强了DST和NLG模块的性能,克服了适配器学习中缺乏的学习曲线问题,并实现了适合目标的自然且一致的回复生成。我们的方法是一种与模型无关的方法,并且不需要提示调优,因为仅使用不带提示的输入数据。实验结果表明,与现有的端到端模型相比,我们的方法在MultiWOZ基准上表现出具有竞争力的性能。特别是,我们在2.2数据集的DST任务上取得了最先进的性能。

英文摘要:

Task-Oriented Dialogue (TOD) systems are designed to carry out specific tasks by tracking dialogue states and generating appropriate responses to help users achieve defined goals. Recently, end-to-end dialogue models pre-trained based on large datasets have shown promising performance in the conversational system. However, they share the same parameters to train tasks of the dialogue system (NLU, DST, NLG), so debugging each task is challenging. Also, they require a lot of effort to fine-tune large parameters to create a task-oriented chatbot, making it difficult for non-experts to handle. Therefore, we intend to train relatively lightweight and fast models compared to PLM. In this paper, we propose an End-to-end TOD system with Task-Optimized Adapters which learn independently per task, adding only small number of parameters after fixed layers of pre-trained network. We also enhance the performance of the DST and NLG modules through reinforcement learning, overcoming the learning curve that has lacked at the adapter learning and enabling the natural and consistent response generation that is appropriate for the goal. Our method is a model-agnostic approach and does not require prompt-tuning as only input data without a prompt. As results of the experiment, our method shows competitive performance on the MultiWOZ benchmark compared to the existing end-to-end models. In particular, we attain state-of-the-art performance on the DST task of 2.2 dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑