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

Meta 与 MIT 研究 Transformer 循环复用:以少量额外参数增加有效深度

AI 导读

Meta 与 MIT 的一篇论文研究让 Transformer 层重复运行,以增加模型深度而不同比例增加参数。研究在最多 10 亿参数、1550 亿训练 token 的稠密模型上测试了循环位置与次数:固定模型规模时,循环整个层堆栈的效果最好;训练算力紧张时,循环中间层更划算。

原帖引用 · 原文

@rohanpaul_ai

– https://t.co/ALfO6njve4

Title: "How to Loop Your Transformer: Loop Scaling, Specialization, and Parallel Inference"

回复Rohan Paul@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.
在 X 查看这条帖子 ↗

来源:Rohan Paul · x.com

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