用教师模型指导循环变压器,让其高效压缩视觉历史
Compressing Observation History into Agent Memory: Distilling Transformers into Recurrent Transformers

- 用教师模型显式压缩观察历史为固定大小瓶颈
- 学生模型通过监督学习,使记忆压缩策略对齐
- 在机器人任务中实现线性时间复杂度,性能接近全历史模型
Transformer 在建模序列数据方面表现优异,但处理长序列时计算成本过高。针对地图无关位姿估计等长时序视觉与机器人应用,存储和维护完整观察历史不切实际。循环Transformer通过维持固定大小的内存来缓解此问题,但性能仍落后于基于完整历史的Transformer。我们指出,这一差距并非源于架构限制,而是学习压缩过去信息的方式不同所致。循环模型在无历史访问的情况下,必须在每一步显式决定保留什么信息,学习难度更大。本文提出一种蒸馏方法,将传统全历史Transformer的压缩策略迁移到循环变体上。通过设计一个显式将观察历史压缩为固定大小瓶颈表示的教师模型,并直接以该瓶颈监督学生模型的记忆,实现了两种压缩机制的对齐。实验表明,该方法可训练出具有线性时间复杂度的循环潜在机器人记忆,显著缩小与全历史Transformer的性能差距。
原文摘要 · Abstract (English)
Transformers are AI's workhorse with strong performance in modeling sequential data, but their computational cost becomes prohibitive when processing long sequences. We target long-horizon streaming vision and robotics applications like map-free pose estimation, where it is particularly impractical to store and maintain a history of observations. Recurrent Transformers address this limitation by maintaining fixed-size memory but their performance lags behind that of transformers operating over the full observation history. We argue that this gap does not stem from architectural limitations, but from differences in how these models learn to compress past information. Without access to an observation history, recurrent models must explicitly decide what to retain in memory at each step, a significantly harder learning problem. In this work, we propose a distillation approach that transfers the compression strategy of a classical full-history transformer to a recurrent variant. We enable this by designing a teacher model that explicitly compresses its observation history into a fixed-size bottleneck representation. By directly supervising the student's memory with this bottleneck representation, we align the two compression mechanisms. We show that this approach allows to train a recurrent latent robotic memory with linear-time complexity while substantially narrowing the performance gap to full-history transformers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。