TensorX
返回文献探索

Paper · arXiv 2305.06555

Domain Incremental Lifelong Learning in an Open World

Yi Dai, Hao Lang, Yinhe Zheng, Bowen Yu, Fei Huang, Yongbin Li

1 upvotesMay 11, 2023arXiv 预印本
AI 摘要

Diana is a dynamic architecture-based lifelong learning model using prompt-enhanced language models to handle a sequence of tasks, including unseen tasks, with improved generalization and performance.

lifelong learningarchitecture-basedprompt-enhanced language modeltask-level promptsinstance-level promptsprompt key vectorsunseen tasks

Abstract

Lifelong learning (LL) is an important ability for NLP models to learn new tasks continuously. Architecture-based approaches are reported to be effective implementations for LL models. However, it is non-trivial to extend previous approaches to domain incremental LL scenarios since they either require access to task identities in the testing phase or cannot handle samples from unseen tasks. In this paper, we propose Diana: a dynamic architecture-based lifelong learning model that tries to learn a sequence of tasks with a prompt-enhanced language model. Four types of hierarchically organized prompts are used in Diana to capture knowledge from different granularities. Specifically, we dedicate task-level prompts to capture task-specific knowledge to retain high LL performances and maintain instance-level prompts to learn knowledge shared across input samples to improve the model's generalization performance. Moreover, we dedicate separate prompts to explicitly model unseen tasks and introduce a set of prompt key vectors to facilitate knowledge sharing between tasks. Extensive experiments demonstrate that Diana outperforms state-of-the-art LL models, especially in handling unseen tasks. We release the code and data at https://github.com/AlibabaResearch/DAMO-ConvAI/tree/main/diana.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Domain Incremental Lifelong Learning in an Open World | TensorX