用谱表示法重构强化学习,提升稳定性与可解释性。
Spectral Representation-based Reinforcement Learning
- 基于转移算子的谱分解,构建系统动态的抽象表示
- 在20+任务上表现媲美或超越当前最优基线
- 适用于部分可观测MDP,理论清晰且计算高效
在状态空间和动作空间庞大的现实应用中,强化学习通常采用函数逼近来表示策略、价值函数和动态模型。尽管神经网络等强大近似方法具有高表达能力,但常伴随理论模糊、优化不稳、探索困难及高计算成本等问题。本文提出以谱表示为视角,解决上述难题。基于转移算子的谱分解,该框架为后续策略优化提供有效系统动态抽象,并具备明确的理论刻画。我们揭示了如何为具有潜在变量结构或能量基础结构的转移算子构建谱表示,对应不同的数据驱动学习方法。每种方法均在该框架下实现有效的强化学习算法。我们还严格将该谱视角扩展至部分可观测马尔可夫决策过程。最终,在DeepMind Control Suite超过20个挑战性任务上验证,所提算法性能达到或优于当前主流无模型与有模型基线。
原文摘要 · Abstract (English)
In real-world applications with large state and action spaces, reinforcement learning (RL) typically employs function approximations to represent core components like the policies, value functions, and dynamics models. Although powerful approximations such as neural networks offer great expressiveness, they often present theoretical ambiguities, suffer from optimization instability and exploration difficulty, and incur substantial computational costs in practice. In this paper, we introduce the perspective of spectral representations as a solution to address these difficulties in RL. Stemming from the spectral decomposition of the transition operator, this framework yields an effective abstraction of the system dynamics for subsequent policy optimization while also providing a clear theoretical characterization. We reveal how to construct spectral representations for transition operators that possess latent variable structures or energy-based structures, which implies different learning methods to extract spectral representations from data. Notably, each of these learning methods realizes an effective RL algorithm under this framework. We also provably extend this spectral view to partially observable MDPs. Finally, we validate these algorithms on over 20 challenging tasks from the DeepMind Control Suite, where they achieve performances comparable or superior to current state-of-the-art model-free and model-based baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。