从单条轨迹中高效学习环境可控动态,突破传统方法需重置的限制。
Learning a Fast Mixing Exogenous Block MDP using a Single Trajectory
- 基于单次连续交互轨迹,设计可证明高效的表示学习算法。
- 样本复杂度仅依赖可控状态空间大小与编码器函数类,与噪声因子无关。
- 适用于需要快速适应新任务的强化学习场景,尤其适合无重置环境。
为训练能快速适应新目标或奖励函数的智能体,序列决策环境中的高效无监督表示学习至关重要。已有框架如外生块马尔可夫决策过程(Ex-BMDP)形式化了这一问题:环境高维观测包含可控制因子(随动作确定性演化于小状态空间)和外生因子(代表时序相关噪声,可能高度复杂)。目标是学习将观测映射到可控制潜在空间的编码器及其动态。先前工作(Efroni et al., 2022b)表明,在回合制设定下,该问题可实现仅依赖可控制空间大小的样本复杂度,而与噪声因子大小无关。然而,此前研究未解决仅能获取单一连续轨迹情形下的样本复杂度边界。本文提出STEEL——首个在函数逼近设置下,从单条轨迹中学习Ex-BMDP可控制动态的可证明样本高效算法。STEEL的样本复杂度仅依赖可控制状态空间大小与编码器函数类规模,且最差情况下线性依赖外生噪声的混合时间。我们证明了STEEL的正确性与样本效率,并在两个玩具问题上进行了验证。代码已开源:https://github.com/midi-lab/steel。
原文摘要 · Abstract (English)
In order to train agents that can quickly adapt to new objectives or reward functions, efficient unsupervised representation learning in sequential decision-making environments can be important. Frameworks such as the Exogenous Block Markov Decision Process (Ex-BMDP) have been proposed to formalize this representation-learning problem (Efroni et al., 2022b). In the Ex-BMDP framework, the agent's high-dimensional observations of the environment have two latent factors: a controllable factor, which evolves deterministically within a small state space according to the agent's actions, and an exogenous factor, which represents time-correlated noise, and can be highly complex. The goal of the representation learning problem is to learn an encoder that maps from observations into the controllable latent space, as well as the dynamics of this space. Efroni et al. (2022b) has shown that this is possible with a sample complexity that depends only on the size of the controllable latent space, and not on the size of the noise factor. However, this prior work has focused on the episodic setting, where the controllable latent state resets to a specific start state after a finite horizon. By contrast, if the agent can only interact with the environment in a single continuous trajectory, prior works have not established sample-complexity bounds. We propose STEEL, the first provably sample-efficient algorithm for learning the controllable dynamics of an Ex-BMDP from a single trajectory, in the function approximation setting. STEEL has a sample complexity that depends only on the sizes of the controllable latent space and the encoder function class, and (at worst linearly) on the mixing time of the exogenous noise factor. We prove that STEEL is correct and sample-efficient, and demonstrate STEEL on two toy problems. Code is available at: https://github.com/midi-lab/steel.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。