arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2405.10992cs.LGcs.AI

通过示例选择克服任务型对话系统中的灾难性遗忘

Overcoming Catastrophic Forgetting by Exemplar Selection in Task-oriented Dialogue System

Nanyang Technological University(南洋理工大学);University of Aberdeen(阿伯丁大学);Zhejiang University(浙江大学)

Chen Chen, Ruizhe Li, Yuchen Hu, Yuanyuan Chen, Chengwei Qin, Qiang Zhang

AI总结:

针对任务型对话系统持续学习中的灾难性遗忘问题,提出基于超梯度的示例选择方法HESIT,通过追踪超梯度选有影响力示例重训,无需估计Hessian,适配大预训练模型,在最大ToD持续学习基准上全指标达SOTA。

摘要:

智能任务型对话系统(ToDs)需要持续获取新知识,即持续学习(Continual Learning, CL),这对适配不断变化的用户需求至关重要。然而,面对长序列的课程学习,灾难性遗忘会导致模型性能急剧下降。本文旨在解决任务型对话系统中的遗忘问题,提出一种基于超梯度的示例策略方法(HESIT),该策略采样有影响力的示例进行周期性再训练。HESIT并未单方面观测数据或模型,而是采用一种深度的示例选择策略,在为每个任务域选择示例时会考虑已训练模型的整体性能。具体而言,HESIT通过追踪优化过程中的超梯度来分析训练数据的影响力。此外,HESIT避免了海森矩阵(Hessian)估计,使其能够兼容搭载大型预训练模型的任务型对话系统。实验结果表明,HESIT通过示例选择有效缓解了灾难性遗忘,在最大的任务型对话系统持续学习基准上,所有指标均达到了当前最优性能。

英文摘要:

Intelligent task-oriented dialogue systems (ToDs) are expected to continuously acquire new knowledge, also known as Continual Learning (CL), which is crucial to fit ever-changing user needs. However, catastrophic forgetting dramatically degrades the model performance in face of a long streamed curriculum. In this paper, we aim to overcome the forgetting problem in ToDs and propose a method (HESIT) with hyper-gradient-based exemplar strategy, which samples influential exemplars for periodic retraining. Instead of unilaterally observing data or models, HESIT adopts a profound exemplar selection strategy that considers the general performance of the trained model when selecting exemplars for each task domain. Specifically, HESIT analyzes the training data influence by tracing their hyper-gradient in the optimization process. Furthermore, HESIT avoids estimating Hessian to make it compatible for ToDs with a large pre-trained model. Experimental results show that HESIT effectively alleviates catastrophic forgetting by exemplar selection, and achieves state-of-the-art performance on the largest CL benchmark of ToDs in terms of all metrics.

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

↑