用只保留查询的注意力机制,解决持续学习中的遗忘与僵化问题
Continual Learning with Query-Only Attention
- 摒弃键值对,仅用查询构建注意力,简化模型结构
- 在多个任务上显著降低灾难性遗忘,优于选择性重初始化等基线
- 为持续学习中保持适应力提供了新思路,适合研究元学习与模型压缩者
持续学习需在不重复数据点的流式数据中持续学习,因任务间分布变化而极具挑战。本文提出一种仅保留查询的注意力机制,舍去键和值,但仍保留Transformer的核心归纳偏置。该简化机制在持续学习场景中显著缓解了可塑性丧失与灾难性遗忘,优于选择性重初始化等基线方法。我们建立了查询仅注意力、全注意力与模型无关元学习之间的概念关联,将其统一为元学习的实例。通过初步的海瑟矩阵谱分析发现,跨任务保持更高曲率秩的模型更易维持可塑性。结果表明,完整注意力并非实现元学习优势的必要条件。
原文摘要 · Abstract (English)
Continual learning involves learning from a stream of data without repetition of data points, a scenario that is inherently complex due to distributional shift across tasks. We propose a query-only attention mechanism that discards keys and values, yet preserves the core inductive bias of transformer architectures. In continual learning scenarios, this simplified mechanism significantly mitigates both loss of plasticity and catastrophic forgetting, outperforming baselines such as selective re-initialization. We establish a conceptual link between query-only attention, full transformer attention, and model agnostic meta-learning, framing them as instances of meta-learning. We further provide intuition for why query-based models and attention networks help preserve plasticity in continual settings. Finally, through preliminary Hessian spectrum analysis, we observe that models maintaining higher curvature rank across tasks tend to retain plasticity. Our findings suggest that full attention may not be essential for capturing the benefits of meta-learning in continual learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。