用任务专属标记避免遗忘,不存旧数据也能持续学习。
Reducing catastrophic forgetting of incremental learning in the absence of rehearsal memory with task-specific token
- 用视觉变换器思想设计任务专用标记,压缩存储过往知识
- 在多个增量学习场景中准确率最高,后向迁移最小
- 适合隐私敏感场景,可拓展至多种视觉变换器架构
深度学习模型在连续学习新数据时通常出现灾难性遗忘。许多增量学习方法通过重用旧任务数据来缓解此问题,但直接访问历史数据引发隐私与安全担忧。为此,本文提出一种无需存储旧数据即可保留旧知识的新方法。该方法受视觉变换器架构启发,引入可封装各任务压缩知识的独特标记。通过根据不同任务调整注意力机制,生成任务特定嵌入,从而以标记模拟多模型效果。方法还包含蒸馏过程,确保多次学习后仍保持高效交互,有效抑制遗忘。我们在基准数据集上评估了不同任务增量学习场景下的准确率与后向迁移表现。结果表明,本方法在对比方法中取得最高准确率和最低后向迁移,性能最优。除提出新模型外,该方法也为视觉变换器架构的多种扩展奠定了基础。
原文摘要 · Abstract (English)
Deep learning models generally display catastrophic forgetting when learning new data continuously. Many incremental learning approaches address this problem by reusing data from previous tasks while learning new tasks. However, the direct access to past data generates privacy and security concerns. To address these issues, we present a novel method that preserves previous knowledge without storing previous data. This method is inspired by the architecture of a vision transformer and employs a unique token capable of encapsulating the compressed knowledge of each task. This approach generates task-specific embeddings by directing attention differently based on the task associated with the data, thereby effectively mimicking the impact of having multiple models through tokens. Our method incorporates a distillation process that ensures efficient interactions even after multiple additional learning steps, thereby optimizing the model against forgetting. We measured the performance of our model in terms of accuracy and backward transfer using a benchmark dataset for different task-incremental learning scenarios. Our results demonstrate the superiority of our approach, which achieved the highest accuracy and lowest backward transfer among the compared methods. In addition to presenting a new model, our approach lays the foundation for various extensions within the spectrum of vision-transformer architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。