Scaling to Tens of Thousands of Test-Time Iterations with Loop-Native Attention Residuals
测试时迭代扩展至数万步的循环原生注意力残差
🧭 循环 Transformer 测试时迭代扩展
The Hong Kong Polytechnic University · Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences · Peng Cheng Laboratory · University of Chinese Academy of Sciences · ELLIS Institute Tübingen
提出 InfiLoop 循环原生残差连接,使 7M 模型在数万次测试时迭代中保持推理提升,Sudoku 97.9%,ARC-AGI-2 13.6% pass@2
📌 工作参考价值(4 点)
- 对 Agent 多轮推理/工具调用场景:借鉴 InfiLoop 的循环原生残差,在 agent 状态更新中引入 content-based weighting + learned temporal decay,保留关键中间推理、抑制噪声 proposal,避免长链路中正确中间状态被覆盖。
- 工程实现上:exact streaming recurrence 使聚合内存恒定,适合在线推理和流式数据处理;在电商推荐中可用于实时会话状态维护或长期用户行为序列建模,避免循环步数增加带来的记忆膨胀和误差累积。
- 测试时扩展以小参数(7M)获得显著推理提升,对资源受限的线上推理服务有参考价值:可以设计轻量循环模块,通过增加循环步数而非模型参数来提升复杂 query 理解或多步策略决策能力。
- 普通 residual connection 在 loop 中可能不够,需要为循环结构单独设计门控残差;具体 trick 是学习每个新更新接受多少、保留多少,而非简单加权平均。