用隐变量学习长期动作,高效解决随机环境下的高维决策问题。
Scalable Decision-Making in Stochastic Environments through Learned Temporal Abstraction
- 通过条件VQ-VAE学习时序扩展的宏观动作,降低动作维度。
- 在连续控制与机器人抓取任务中,性能优于现有模型基方法。
- 适合高维动作空间、随机性强的复杂环境决策场景。
高维连续动作空间中的序列决策,尤其在随机环境中面临显著计算挑战。本文在传统离线强化学习设定下研究此问题,即智能体需基于随机行为策略收集的数据进行决策。提出潜变量宏动作规划器(L-MAP),通过状态条件化的向量量化变分自编码器(VQ-VAE)学习一组时序扩展的宏观动作,有效降低动作维度。L-MAP采用独立学习的先验模型作为潜在转移模型,支持高效采样合理动作。规划阶段利用蒙特卡洛树搜索(MCTS)处理环境和行为策略中的随机性。在包括具有内在随机动力学的连续控制任务及高维机器人手操作在内的离线强化学习设置中,L-MAP通过在离散潜在动作空间中高效搜索,获得高期望回报。实验表明,尽管动作维度增加,L-MAP仍保持低决策延迟。在多种任务中显著优于现有模型基方法,并达到强模型无关演员-评论家基线的水平,验证了该方法在复杂随机环境中高维动作空间决策的有效性。
原文摘要 · Abstract (English)
Sequential decision-making in high-dimensional continuous action spaces, particularly in stochastic environments, faces significant computational challenges. We explore this challenge in the traditional offline RL setting, where an agent must learn how to make decisions based on data collected through a stochastic behavior policy. We present Latent Macro Action Planner (L-MAP), which addresses this challenge by learning a set of temporally extended macro-actions through a state-conditional Vector Quantized Variational Autoencoder (VQ-VAE), effectively reducing action dimensionality. L-MAP employs a (separate) learned prior model that acts as a latent transition model and allows efficient sampling of plausible actions. During planning, our approach accounts for stochasticity in both the environment and the behavior policy by using Monte Carlo tree search (MCTS). In offline RL settings, including stochastic continuous control tasks, L-MAP efficiently searches over discrete latent actions to yield high expected returns. Empirical results demonstrate that L-MAP maintains low decision latency despite increased action dimensionality. Notably, across tasks ranging from continuous control with inherently stochastic dynamics to high-dimensional robotic hand manipulation, L-MAP significantly outperforms existing model-based methods and performs on-par with strong model-free actor-critic baselines, highlighting the effectiveness of the proposed approach in planning in complex and stochastic environments with high-dimensional action spaces.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。