循环Transformer:仅循环中间层效果更优
"Towards Looped Models Done Right"
摘要
论文比较不同循环架构发现,仅循环Transformer中间层优于反复循环整个模型;Huginn式设计(在独立输入输出层之间放置循环层)整体表现更好,尤其在推理任务上。8B参数下Huginn MoE在10项基准中8项超过Ouro,存储参数比算力匹配的32B前馈MoE少75%,GSM8K准确率83.6%对80.8%。
摘要由模型根据原文撰写,以原文为准。
正文 · 原文
"Towards Looped Models Done Right"
Not all ways of looping Transformers work equally well. This paper finds that looping only the middle layers performs better than repeatedly looping the entire Transformer.
The researchers compare different loop architectures and find that the Huginn-style design, which places recurrent layers between separate input and output layers, generally performs better, especially on reasoning tasks.
At 8B parameters, Huginn MoE outperforms Ouro on 8 out of 10 benchmarks. It also uses 75% fewer stored parameters than a compute-matched 32B feedforward MoE, while achieving better GSM8K accuracy (83.6% vs. 80.8%)
来源:alphaXiv · x.com
引用原文
askalphaxiv. "Towards Looped Models Done Right"[EB/OL]. (2026-10-31)[访问日期]. https://x.com/askalphaxiv/status/2116438250730737844.