arXiv:2501.09166cs.LGcs.AI2025-01被引 1

给Transformer加了可持久记忆模块,让模型能像人一样持续学习。

Attention is All You Need Until You Need Retention

  • 引入持久化记忆模块,实现动态存取与更新
  • 支持跨会话复用模式,提升增量学习能力
  • 适合需要持续适应的智能系统如客服、医疗诊断

本文提出一种新型的Retention Layer机制,用于解决基于Transformer架构模型固有的缺乏内在保留能力的问题。与人类能够编码并动态回忆符号模板不同,生成式预训练Transformer仅依赖固定预训练权重和短暂上下文窗口,限制了其适应性。所提的Retention Layer包含一个可实时填充数据、动态召回并引导输出生成的持久记忆模块。该增强使模型能够在不同会话间存储、更新并复用观察到的模式,实现增量学习,并弥合静态预训练与动态、上下文敏感适应之间的差距。该设计类比社会学习过程,涵盖注意力、保留、再现和动机阶段。技术上,它整合了记忆注意力机制和情景缓冲区,以管理记忆扩展性、缓解过拟合并确保高效召回。应用场景包括自适应个人助理、实时欺诈检测、自主机器人、内容审核和医疗诊断。在各领域中,该记忆机制使系统能实现增量学习、个性化输出,并有效应对不断变化的真实世界挑战。通过模拟人类学习的关键方面,这种增强型架构促进了更流畅、响应更快的AI范式,为动态、会话感知模型开辟了新路径,将传统Transformer的能力拓展至需持续适应的领域。

原文摘要 · Abstract (English)

This work introduces a novel Retention Layer mechanism for Transformer based architectures, addressing their inherent lack of intrinsic retention capabilities. Unlike human cognition, which can encode and dynamically recall symbolic templates, Generative Pretrained Transformers rely solely on fixed pretrained weights and ephemeral context windows, limiting their adaptability. The proposed Retention Layer incorporates a persistent memory module capable of real time data population, dynamic recall, and guided output generation. This enhancement allows models to store, update, and reuse observed patterns across sessions, enabling incremental learning and bridging the gap between static pretraining and dynamic, context sensitive adaptation. The Retention Layer design parallels social learning processes, encompassing attention, retention, reproduction, and motivation stages. Technically, it integrates a memory attention mechanism and episodic buffers to manage memory scalability, mitigate overfitting, and ensure efficient recall. Applications span adaptive personal assistants, real time fraud detection, autonomous robotics, content moderation, and healthcare diagnostics. In each domain, the retention mechanism enables systems to learn incrementally, personalize outputs, and respond to evolving real world challenges effectively. By emulating key aspects of human learning, this retention enhanced architecture fosters a more fluid and responsive AI paradigm, paving the way for dynamic, session aware models that extend the capabilities of traditional Transformers into domains requiring continual adaptation.

Transformer持续学习记忆模块自适应系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。