用可解析的自动机任务测试强化学习是否真学到了隐藏状态。
When Does Reward Teach State? A Hidden-Automaton Instrument and a Group-Language Warning Signal

- 设计可计算最优回报的隐藏自动机任务,直接测量状态学习
- 89/103个群语言自动机任务中,早期警告信号准确预判状态学习失败
- 区分感知缺失与规划缺陷,揭示奖励不能证明状态已学会
强化学习代理获得高奖励,是否真正学习了任务的隐藏状态,还是仅找到了与奖励相关的捷径?我们构建了一个可直接测量该问题的工具:任务为一个部分可控的隐藏有限自动机。由于自动机结构已知,最优回报可计算,原始奖励转化为归一化得分;每一步的真实状态也已知,可通过线性探测器检验代理是否追踪其从未观测的状态。分别测量后发现,弱策略梯度强化学习虽获高奖励,但状态探测始终处于随机水平。能否摆脱捷径取决于优化器、训练预算和任务结构。若自动机为置换(群语言)结构,仅从转移表即可提前预警,153个保留测试自动机中,该信号在103次触发时正确预测了89次失败;但无信号不保证安全。该工具还能区分两种奖励无法分辨的失败:感知缺口(网络可表示状态却未捕捉),和规划缺口(状态可解码但奖励仍低)。高奖励本身并非状态学习的证据。
原文摘要 · Abstract (English)
Does a reinforcement-learning agent that earns high reward actually learn its task's hidden state, or only a shortcut that correlates with reward? We build an instrument that makes this question directly measurable: the task is a hidden finite automaton that the agent partially controls. Because the automaton is known, the best achievable return is computable, and raw reward becomes a normalized score. The true state is also known at every step, so a linear probe can test whether the agent tracks a state it never observes. Measured separately, reward and state learning come apart: weak on-policy RL earns reward while the state probe stays at chance. Whether the agent escapes this shortcut depends on the optimizer, the training budget, and the task's structure. Structure gives an early warning: when the automaton is a permutation (group-language) automaton, a property readable from its transition table before any training, the agent will likely fail to learn the state. On a stratified set of 153 held-out automata the warning is right in 89 of the 103 cases where it fires; its absence, though, is no guarantee of safety. The instrument also tells apart two failures that reward alone cannot: a perception gap, where the linear probe finds no trace of a state the network could have represented, and a planning gap, where the state is decodable yet reward stays low. High reward alone is not evidence that the task's state was learned.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。