查看: 7|回复: 0

Transformer 模型要么需要耗费大量精力来保存完整的历史信息,要么可以使用滑动窗口,虽然效率高但会遗忘窗口外的信息。

[复制链接]

18

主题

1

回帖

56

积分

注册会员

积分
56
发表于 昨天 23:41 | 显示全部楼层 |阅读模式
“磁悬浮:滑动循环记忆”

Transformer 模型要么需要耗费大量精力来保存完整的历史信息,要么可以使用滑动窗口,虽然效率高但会遗忘窗口外的信息。

本文旨在将遗忘的历史信息循环利用。为此,他们使用了一个更强大的预填充器,可以并行创建记忆目标;以及一个滑动窗口 Transformer 模型,该模型能够学习如何重现这些记忆,并通过其键值缓存将这些记忆传递下去。

预填充器在推理阶段被移除,因此模型获得了具有固定大小注意力机制和键值缓存的持久非线性记忆,同时仍然使用并行 Transformer 迭代进行训练。

https://alphaxiv.org/abs/2608.02870


本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

关注公众号

相关侵权、举报、投诉及建议等,请发 E-mail:2776601884@qq.com

Powered by Discuz! X5.0 © 2001-2026 Discuz! Team.|青ICP备2025004122号-1

在本版发帖
关注公众号
返回顶部