研究人员推出了一种新颖的混合架构 Dynamical Memory Transformer (DMT),旨在改进模型处理长序列的方式。与存储显式历史记录(导致上下文越长计算成本越高)的标准 Transformer 或将过去压缩为固定状态的状态空间模型 (SSM) 不同,DMT 将序列编码与信息检索分开。它采用多个内存单元,每个内存单元由具有输入相关泄漏率的动态系统管理,允许注意力机制读取和组合信息,而无需处理整个 token 历史记录。这种方法旨在实现推理过程中的恒定成本注意力和固定大小的循环状态,在认知任务和语言建模方面显示出潜力,尽管目前在纯语言建模性能方面落后于 Mamba。 AI
影响 引入了一种新的混合架构,可以提高 AI 模型处理长序列的效率和性能。
排序理由 该集群描述了一篇介绍新颖模型架构的最新研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- Dynamical Memory Transformer
- Hugging Face
- KV cache
- Mamba
- State Space Models
- transformer
- Yannis Bendi-Ouis
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →