一项研究探讨了循环层与深度增长语言模型对上下文学习的影响:在 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