Looped transformers apply one block of weights repeatedly. Each iteration keeps a fraction of the previous hidden state, the carry, and adds the block's output, so information can pass between iterations through either path. We study a small, LayerNorm-based variant of the Parcae architecture and ask two questions: does the iterated state settle, and does learning to remember across iterations require the carry? On stability, pre-LayerNorm makes the block less sensitive at its fixed point as the carry grows, but not enough to guarantee contraction: with a scalar carry, a perturbation can grow in a single iteration even at a locally stable fixed point. These fixed points are stable because the spectral radius of the loop's Jacobian stays below $1$; its distance from $1$, the spectral margin, narrows as the carry approaches $1$ and, at high carry, as the width grows, and some initializations do not settle within the iteration budget. On memory, a multi-hop copy task trained at a fixed loop depth is learned as well with the carry frozen at $0.5$, $0.02$, or $0$ as with a learned carry, which changes little during training. When the depth is sampled during training, a carry of $0.5$ still trains but $0.02$ and $0$ do not. All experiments are small ($d \le 32$ for training).


翻译:暂无翻译

0
下载
关闭预览

相关内容

【ICML2024】揭示Graph Transformers 中的过全局化问题
专知会员服务
21+阅读 · 2024年5月27日
Transformer模型-深度学习自然语言处理,17页ppt
专知会员服务
108+阅读 · 2020年8月30日
Stabilizing Transformers for Reinforcement Learning
专知会员服务
61+阅读 · 2019年10月17日
WWW 2020 开源论文 | 异构图Transformer
PaperWeekly
13+阅读 · 2020年4月3日
从头开始了解Transformer
AI科技评论
25+阅读 · 2019年8月28日
Transferring Knowledge across Learning Processes
CreateAMind
29+阅读 · 2019年5月18日
多图带你读懂 Transformers 的工作原理
AI研习社
10+阅读 · 2019年3月18日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Arxiv
11+阅读 · 2021年10月26日
VIP会员
最新内容
人工智能与战争的未来
专知会员服务
3+阅读 · 11月8日
《Agent Mosaic:识别智能体LLM系统中的机密泄露》
人工智能在军事教育中的应用:若干视角与经验
“星链”减速与不断适应的乌克兰
专知会员服务
7+阅读 · 11月6日
美国中期选举之后:帕兰提尔面临不确定性?
专知会员服务
4+阅读 · 11月6日
相关基金
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员