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

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

AI 导读

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

原帖引用 · AI 翻译

@rohanpaul_ai
译文 · 以原文为准

Meta 和 MIT 的一篇新论文发现,在循环中重复使用 Transformer 的层,效果可以超过普通模型。

循环运行让相同的层执行多次,使模型无需增加参数就能获得更大的深度。但每次循环都会消耗算力,而且通常必须针对每个词元依次运行。

他们训练了参数量最高达 10 亿、训练数据最多达 1550 亿个词元的稠密模型,测试了在哪些位置循环以及循环多少次。他们还添加了微型的逐循环适配器,并修正了一种并行循环方法,使多个循环能在解码时同时运行。

在模型大小固定时,循环运行整个层堆栈带来了最佳质量;而在训练算力紧张时,循环运行中间层的收益最高。逐循环适配器只增加了 1.1% 至 4.3% 的参数,而将深度翻倍会增加 51.4%;随着循环次数增加,适配器带来的帮助也更大。

来源:Rohan Paul · x.com

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