跳到正文
Hacker News· smaddrellmander·· 2 小时前AI 评分38

用 Attention Sinks 实现高效流式语言模型

Efficient Streaming Language Models with Attention Sinks

AI 导读

2024 年一篇论文提出用 attention sinks 解释 LLM 长序列性能骤降:KV cache 一旦驱逐序列开头几个 token,GPT-2 small 在 wikitext-2 上的困惑度就从 10.7 飙到 3768。

来源:Hacker News · idlemachines.co.uk