AI & ML
Truncate or Not? A Scaling Study of Looped Transformers with Truncated Backpropagation
arXiv stat.ML0 views
arXiv:2610.26030v2 Announce Type: replace-cross Abstract: Looped transformers reuse a shared block of layers to increase computational depth without increasing the number of trainable parameters. They are trained with backpropagation through time (BPTT). To save training compute, BPTT is often truncated to the last few loops, but truncation yields biased gradients that may weaken the benefit of looping. In this work, we study the value of truncation in looped transformer training in a systematic way: we jointly vary model size, training tokens, and forward and backward loop counts across nearly 600 training runs and fit a unified scaling law that captures the effects of looping and truncation through effective parameters. Despite the training-compute savings of truncation, our fitted law favors full BPTT when forward and backward loop counts are jointly optimized. In matched-compute runs, full BPTT through fewer loops consistently outperforms truncated settings of the same compute. The law further shows that at fixed model size, the compute-optimal loop count and training-token budget both grow with compute. When model size is also free, the law identifies two fully trained loops as the best looped setting. Experimentally, this setting achieves lower validation loss than the compute-optimal non-looped model with about 27% fewer non-embedding parameters and only 8% more training compute.
Source:https://arxiv.org/abs/2610.26030PreviousCertifiably Correct Fr\'echet Means in Orbit SpacesNextA Simple and General Approach to top-$k$ Selection
Related posts
Preparing our next security sprint to harden OpenAI
0OpenAI Blog
Rewarding defenders with bigger bug bounties
0OpenAI Blog
삼중 진화 시대: 암호화폐의 정치적 성숙, 비트코인의 거시경제 민감성, 그리고 AI 융합
0dev.to
My most popular app? Surprisingly, YuGPT! It backs up ChatGPT and Codex history locally and lets you search both with flexible options. I built it because the official apps couldn't do what I needed
0dev.to
Top Dev Tools and Tutorials Trending This Week (2026-11-03)
0dev.to GitHub

DeepSeek Super-Agent คืออะไร? แผนที่เปิดให้อ่าน กับโค้ดที่ตามแผนยังไม่ครบ
0dev.to GitHub