跳到正文
HF Daily Papers·· 9 天前精选AI 评分41

循环 Transformer 缩放研究显示完整 BPTT 在匹配计算量下胜过截断方案

Truncate or Not? A Scaling Study of Looped Transformers with Truncated Backpropagation

AI 导读

研究团队通过近 600 次训练运行比较循环 Transformer 的完整 BPTT(按时间反向传播)与截断方案,发现匹配计算量时完整 BPTT 始终更优。统一缩放律显示,模型大小可调时,完整训练两轮循环是最佳设置;其验证损失低于计算最优的非循环模型,非嵌入参数约少 27%,训练计算量仅多 8%。论文地址为 https://arxiv.org/abs/2610.26030。

推荐理由

近 600 次训练与统一缩放律为循环 Transformer 的截断取舍提供了匹配计算量比较,并给出参数效率与训练成本的具体结果。

来源:HF Daily Papers · huggingface.co