arXiv:2503.23478cs.LGcs.AI2025-03ICLR被引 9

用时间跳跃连接缓解实时强化学习中的延迟问题

Handling Delay in Real-Time Reinforcement Learning

  • 引入时间跳跃连接与历史观测融合,降低计算延迟
  • 在多种环境和算法下性能稳定,推理速度提升6-350%
  • 适合需要低延迟实时决策的强化学习应用

实时强化学习面临两大挑战:硬件限制导致每秒动作数固定;神经网络计算期间环境可能变化,造成观测延迟。对于一个N层前馈网络,若每个神经元执行时间为τ,总延迟为τN。减少层数可降低延迟,但会削弱模型表达能力。本文提出一种理论驱动的解决方案,结合时间跳跃连接与历史增强观测,在保持模型表达力的同时有效压缩延迟。在四个Mujoco任务和全部MinAtar游戏中验证,采用该结构的模型在不同神经元执行时间、强化学习算法下均表现优异。此外,通过并行神经元计算,推理速度相较标准硬件提升6%-350%。本研究为实时强化学习中的高效智能体设计提供了新思路。

原文摘要 · Abstract (English)

Real-time reinforcement learning (RL) introduces several challenges. First, policies are constrained to a fixed number of actions per second due to hardware limitations. Second, the environment may change while the network is still computing an action, leading to observational delay. The first issue can partly be addressed with pipelining, leading to higher throughput and potentially better policies. However, the second issue remains: if each neuron operates in parallel with an execution time of $τ$, an $N$-layer feed-forward network experiences observation delay of $τN$. Reducing the number of layers can decrease this delay, but at the cost of the network's expressivity. In this work, we explore the trade-off between minimizing delay and network's expressivity. We present a theoretically motivated solution that leverages temporal skip connections combined with history-augmented observations. We evaluate several architectures and show that those incorporating temporal skip connections achieve strong performance across various neuron execution times, reinforcement learning algorithms, and environments, including four Mujoco tasks and all MinAtar games. Moreover, we demonstrate parallel neuron computation can accelerate inference by 6-350% on standard hardware. Our investigation into temporal skip connections and parallel computations paves the way for more efficient RL agents in real-time setting.

强化学习实时系统延迟优化神经网络

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