aifollow.news
返回 Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · 原发布时间 AI 评分54

斯坦福新论文追踪小型语言模型从示例中学习的机制

AI 导读

研究在 Qwen3-4B、SmolLM3-3B 和 Llama-3.2-3B 中追踪了模型如何利用提示中的示例:标签词元保存句子类别,后续约 90 个注意力头将新句子与已保存的类别匹配。实验中,每类给 1 个示例时准确率约为 20%,给 6 个时为 96%;研究指出,增加示例仍无法教会模型识别它原本未注意到的特征。

原帖引用 · 原文

@rohanpaul_ai

New Stanford paper finds that showing a small AI language model more examples cannot teach it a pattern it does not already notice, such as the third letter of a word.

finds that LLMs learn from few-shot examples by looking up stored labels, and only switch that lookup on when labels repeat.

An early signal opens this lookup only when some label repeats. With 1 example per class the model was right about 20% of the time, and with 6 it was right 96%.

Give a model a dozen sentences tagged with made-up labels like foo and bar, and it learns the mapping without weight updates. Earlier work found parts of this, but not how they fit together.

They traced the full path in Qwen3-4B, SmolLM3-3B and Llama-3.2-3B. Each label token stores the class of its sentence, and about 90 later heads match the new sentence against those stored classes.

For few-shot prompts, repeat every label several times and pick classes the model can already tell apart, since examples cannot teach a feature it lacks.

– arxiv. org/abs/2611.05344

Title: "A Circuit for In-Context Learning"

来源:Rohan Paul · x.com

论文
发现内容有误?提交纠错