跳到正文
elvis· @omarsar0 · X·· 3 小时前AI 评分38
AI 导读

一项研究探讨循环与深度增长语言模型对上下文学习的影响:在 3B 参数规模下,深度增长和循环模型在 GSM8K 各演示数量下均高于同深度稠密模型,并能更准确预测未见形式语言。受控任务中它们更早习得上下文学习,且在上下文与权重记忆标签均可解题时持续依赖上下文。机制体现在注意力头上:复制层增长深度会复制早期归纳头,多数最强最终头源自少数早期祖先,循环模型在共享块多次传递中呈现相同重复模式。

正文 · 原文

Highly recommended read on looped and depth-grown models.

There is a lot of interest in looped and depth-grown language models for reasoning.

The question here is what reusing layers does to in-context learning.

At 3B parameters, both grown and looped models gain more from later context than a dense model of the same depth. They score higher on GSM8K at every number of demonstrations and predict unseen formal languages more accurately.

In controlled tasks, they pick up in-context learning earlier. When a task can be solved either from context or by memorizing fixed labels in the weights, they keep relying on context.

The mechanism shows up in the attention heads. Growing depth by copying layers also copies early induction heads, and most of the strongest final heads descend from a few early ancestors. Looped models show the same repeated pattern across passes of their shared block.

https://arxiv.org/abs/2610.31421

Chat with Paper: https://academy.dair.ai/papers/how-depth-growth-and-looping-shape-in-context-learning-in-language-models-2610.31421

来源:elvis · x.com