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

Meta 与 MIT 研究发现,循环复用模型层可改善固定规模下的表现

AI 导读

Meta 与 MIT 的论文研究让同一组 Transformer 层多次运行,以增加有效深度而不增加基础模型参数。研究在最高 10 亿参数、最多 1550 亿训练 token 的模型上测试发现:固定模型规模时,循环整个层堆栈的效果最好;训练算力紧张时,循环中间层更划算。循环通常会增加计算量,且每个 token 的循环需依次执行。

原帖引用 · 原文

@rohanpaul_ai

New Meta and MIT paper finds reusing a transformer's layers in loops can beat a plain model.

Looping runs the same layers several times, giving a model more depth without more parameters. But each loop costs compute and normally runs in sequence for every token.

They trained dense models up to 1B parameters on up to 155B tokens, testing where to loop and how many times. They also added tiny per-loop adapters and repaired a parallel looping method so loops can run at once during decoding.

Looping the whole stack gave the best quality at a fixed size, while looping middle layers paid off best when training compute was tight. Per-loop adapters added only 1.1 to 4.3% more parameters, versus 51.4% for doubling depth, and helped more as loops increased.

来源:Rohan Paul · x.com

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