论文精选

HOLA:为线性注意力引入海马体互补存储,提升长程召回

NEW paper worth reading. (bookmark it) The basic idea is to pair a compressive recurrent state wit...

精选理由

线性注意力一直有长程遗忘问题,HOLA用一个小缓存就解决了,在32k长度上效果惊人。

AI 摘要

HOLA论文提出为线性注意力添加海马体互补存储,保留delta-rule状态的压缩记忆并增加一个有限精确KV缓存,形成半参数测试时记忆。在340M参数、15B SlimPajama tokens训练下,WikiText困惑度从27.32降至22.92,低于全注意力Transformer++的26.88。在RULER needle recall测试中,HOLA在32k tokens(16倍训练长度)上保持稳健。

AI 翻译 · 中文

HOLA论文提出为线性注意力添加海马体互补存储,保留delta-rule状态的压缩记忆并增加一个有限精确KV缓存,形成半参数测试时记忆。在340M参数、15B SlimPajama tokens训练下,WikiText困惑度从27.32降至22.92,低于全注意力Transformer++的26.88。在RULER needle recall测试中,HOLA在32k tokens(16倍训练长度)上保持稳健。

elvisNEW paper worth reading. (bookmark it) The basic idea is to pair a compressive recurrent state with a small exact memory, which helps to recover long-range recall without giving up the efficiency of linear attention. Mor