arXiv:2604.01577cs.LGcs.AI2026-04被引 2

用持续隐状态提升模型在长序列上的泛化能力

Exploration of Fast-Slow Latent Recurrence for Train-Short, Test-Long Generalization

  • 引入可复用的隐状态,跨时间步保持并更新信息
  • 在符号序列、导航和强化学习任务中显著优于基线模型
  • 适合需要长程记忆且资源受限的实时推理场景

我们研究流式任务中的分布外泛化问题:模型在短序列上训练,却需在未知长时序上运行且内存受限。提出一种持续快慢隐状态递归结构,隐状态在观测间不重置,而是通过共享权重的递归核心对每一步输入进行多次更新,并将结果状态传递至下一步。该方法可在不重新处理不断增长上下文的前提下,维持并优化紧凑的全局状态。在符号序列预测、监督导航和部分可观测强化学习任务上验证,该结构在分布外泛化性能上优于递归、状态空间和Transformer基线模型。通过递归核心消融实验,发现状态相关转移和特征级非线性混合是强泛化性能的关键架构要素。结果表明,重新审视持续递归作为更通用序列预测的架构先验具有重要意义。

原文摘要 · Abstract (English)

We study out of distribution generalization in streaming tasks where models are trained on short sequences but must operate over much longer, unknown horizons under bounded memory. Our focus is on a persistent fast slow recurrent formulation in which a latent state is maintained across observations rather than reset at each stream step. For each incoming observation, the model performs multiple weight-shared latent updates with a recurrent core and then carries the resulting state forward to the next observation. This allows the model to maintain and refine a compact stream-level state without reprocessing a growing context. We evaluate this formulation across symbolic sequence prediction, supervised navigation, and partially observable reinforcement learning tasks. Across these settings, persistent latent recurrence improves OOD generalization over recurrent, state-space, and Transformer baselines. Through recurrent-core ablations, we identify architectural ingredients that are consistently associated with strong OOD performance, including state-dependent transitions and feature-wise nonlinear mixing. Together, these results highlight the value of revisiting persistent recurrence as an architectural bias for more generalizable sequence prediction.

序列建模泛化能力隐状态流式任务

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。