精读笔记
Problem Setting
这篇论文处理的不是一般意义上的 image-based RL,而是一个更窄但更 planning-centric 的问题:给定高层动作库和视觉转移样本,学习一个能从 start/goal images 直接产出 action sequence 的 symbolic world model。困难点在于,训练数据只告诉你“某图像状态执行某 action 后到了另一图像状态”,但规划需要知道“哪些 action 在哪些抽象条件下可用,以及它们跨状态共享什么 effect”。
以前路线卡在两个极端:视觉 world model / rollout 试图预测未来 latent 或图像,长 horizon 下容易 compound error;latent graph search 避免 rollout 但仍是 flat enumeration,泛化基本等价于图覆盖;LatPlan 类方法试图学 binary latent 和 STRIPS,但在真实视觉 rearrangement 中 predicate/operator 不稳定。关键矛盾是:局部 transition observation 是稀疏、视觉化、带 aliasing 的;而 STRIPS planning 要求全局一致、离散、可复用的 action semantics。
Motivation
作者真正想补的是 visual abstraction 与 classical planning 之间的断层。已有 visual planning 方法通常学到的是“可预测的 latent”,不一定是“可规划的 state”。一个 latent 对 dynamics loss 好,不代表它的 bits 对应稳定 facts;一个 graph 能连通训练样本,也不代表它能在未枚举状态上复用 action model。
核心观察是:高层 manipulation primitives 本身已经暗含了 symbolic structure。若同一个 action id 在多个状态中产生规律性变化,那么应该把这些边解释成同一个 operator,而不是记成多条独立边。缺的是一个从视觉 transition graph 中反推 predicates、preconditions 和 effects 的机制,尤其是需要利用“某状态没有某 action 边”这种负证据来学 precondition。
Core Idea
STRIPS-WM 的核心不是“用神经网络加规划器”,而是把 visual transition data 重新组织为一个可被 STRIPS 解释的离散约束满足问题。先由视觉模型产生一个有限 task graph;再要求这个 graph 的所有 labeled edges 能被少量 predicate bits 和每个 action 一个共享 operator 解释。这样,planning 不再依赖 rollout 的连续预测,也不依赖 flat graph 的边枚举,而依赖 learned precondition/effect 的闭包。
这个建模方式引入的 inductive bias 很强:同名 action 必须有状态无关的 add/delete effects,applicability 必须由 predicate preconditions 决定,未观察到的 trusted action edges 会反向约束 preconditions。它本质上把“视觉状态空间压缩”升级成“动作语义归纳”。相比 prior,真正不同点是负证据驱动的 precondition learning 加上 SAT/CP-SAT 对 STRIPS semantics 的硬约束,而不是 latent architecture 本身。
Method
方法中真正必要的机制有三层。
第一层是 image-grounded task graph。它解决的是 raw images 不能直接进入 symbolic learner 的问题。FSQ/JEPA-style dynamics 的作用是得到有限、近似 deterministic 的 abstract nodes。这里的关键不是重建图像,而是让 code 保留 action-relevant differences,并尽量避免同一 code 在同一 action 下有多个 successor。这个阶段决定下游上限。
第二层是 CP-SAT symbolic induction。它解决的是 flat graph 没有可复用 action semantics 的问题。每个 node 的 predicate vector 和每个 action 的 precondition/add/delete masks 被联合求解;observed edges 强制 action applicable 且 effect 匹配 successor;trusted missing pairs 强制至少一个 precondition 被违反。这个机制把可达边、不可达边和 shared effects 放进同一个离散优化问题,是论文最核心的部分。
第三层是 visual predicate grounding。它解决 test-time image 到 symbolic state 的接口问题。离线 CP-SAT 得到的 predicate labels 被用来训练图像分类器,start/goal images 被映射成 predicate vectors 后交给 classical planner。这个环节必要但不深;它更多是让 learned symbolic model 可部署,而不是贡献新的 reasoning 能力。
Key Insight / Why It Works
最重要的 insight 是:precondition learning 不能只靠 positive transitions。只看 observed edges,空 precondition 几乎总是能解释成功执行,结果就是 action 到处可用,planning 时产生大量 false positives。论文用 trusted missing state-action pairs 作为 negative evidence,本质上把 closed-world assumption 局部化,并用它迫使 solver 找到能排除不可用 action 的 predicates。这是方法真正有效的关键。
第二个有效点是 shared operator constraint。flat graph 中每条边都是独立记忆;STRIPS operator 要求同一 action 在所有 source states 上共享 effects 与 preconditions。这会自动产生压缩:如果视觉 graph 过分裂但这些节点 action-equivalent,CP-SAT 可以给它们相同 predicate vector,从而修复视觉 abstraction 的 over-splitting。DinnerTable Real 的结果基本说明了这一点。
第三个有效点是把 neural uncertainty 隔离在前后端,把核心规划模型交给离散 solver。视觉 encoder 负责生成候选 graph,classifier 负责 test-time grounding;中间的 operator induction 是 symbolic hard constraint。这种 pipeline 的优点是错误类型更可诊断:transition slack、applicability slack、false-positive actions 都能直接暴露问题。
但也要直接说:这不是开放式视觉推理。能力很可能主要来自覆盖良好的 high-level transition data + 强 STRIPS inductive bias + closed-world negative evidence。长程规划成功并不证明模型学到了可外推的 relational structure;它证明在这些小型 rearrangement domains 中,grounded propositional compression 比 latent rollout/graph memory 更合适。视觉模块的增益来源不清,可能主要是 adequate abstraction 而不是架构本身。
Relation To Prior Work
最接近的谱系有三条:visual world models / latent planning、latent graph / roadmap planning、symbolic abstraction / action model learning。STRIPS-WM 属于第三条,但前端接了 raw image abstraction。
相对 visual rollout 和 latent dynamics,它的本质差异是放弃预测未来 observation,改为学习 symbolic transition rule。相对 LSR/WM-BFS 这类 graph search,它不是枚举边,而是用 predicates/operators 对图进行解释和压缩。相对 LatPlan,它更强调从 action-labeled visual transitions 中构造 grounded task graph,再用 CP-SAT 和 negative evidence 学 preconditions;这比单纯 autoencoding binary symbolic states 更有约束。
看似新的部分有不少是已有思想重组:JEPA/FSQ 离散视觉抽象、SAT-based action model learning、STRIPS operator semantics、skills-to-symbols 都不是新概念。实质创新在于把这些组合成一个从 images 到 grounded propositional STRIPS model 的端到端 pipeline,并明确把 trusted missing actions 作为 precondition supervision。它没有推进 lifted symbolic learning,但在 grounded visual planning setting 下给了一个更实用的 recipe。
Dataset / Evaluation
实验覆盖三个 rearrangement domain:小型 BlocksWorld、模拟 DinnerTable、真实图像 DinnerTable Real。真实世界部分是 real images,但不是完整真实机器人闭环部署;评估通过 ground-truth transition graph 执行返回 plan,这比只在 learned model 中判成功更严格,但仍然是离线 task graph evaluation。
benchmark 支持的 claim 是有限但明确的:在固定高层动作库、有限状态空间、相对充分覆盖的 visual transition 数据下,STRIPS-style operator induction 比 rollout、latent graph search 和 LatPlan baseline 更稳。尤其 negative evidence ablation 很有说服力,说明 precondition 不是自动从 positive edges 中涌现的。
但 evaluation 没有真正验证跨 object count、跨 action arguments、跨 scene distribution 的组合泛化。DinnerTable Real 中 visual graph over-splitting 被 symbolic layer collapse 是亮点,但这仍发生在同一任务族内。文中未充分说明在更稀疏采样、更不完整 action coverage、或更强 perceptual aliasing 下 trusted missing pairs 如何选择,因此核心假设的脆弱性没有被充分 stress test。
Limitation
最根本限制是 grounded propositional。每个 action id 对应一个 operator,没有 lifted arguments,没有 object-centric schemas。因此它不能自然泛化到新物体数、新位置集合、新 action parameterization;扩展需要更多 grounded actions 和更多数据,状态/动作组合会很快膨胀。
第二个限制是 closed-world negative evidence。方法需要相信某些 state 中未观测到某 action 就意味着不可用。如果 dataset coverage 不完整,这会直接变成错误 precondition。论文通过 trusted states 和 slack 缓解,但 trusted set 的选择本身是关键且文中未充分说明其真实部署可靠性。
第三个限制是视觉 abstraction 上限。CP-SAT 可以修复 over-splitting,但无法可靠修复 harmful aliasing:如果两个 action-relevantly different states 被合并成同一 node,同一 action 出现多个 successor 或 precondition/effect 冲突,solver 只能用 slack 或失败。此时 symbolic layer 不是万能抽象器,只是在已有 graph 上做解释。
第四个限制是 scalability。CP-SAT 对 predicate dimension、node count、action count、negative pairs 数量敏感。当前 domains 很小,且 action library 是固定 finite identifiers。更大真实 manipulation domains 下,求解成本和数据覆盖需求可能成为主瓶颈。所谓 reasoning 也可能退化为在高覆盖离散 task graph 上做 symbolic compression;泛化强度不应被过度解读。
Takeaway
- 1)对 visual planning 来说,比“更好的 latent rollout”更重要的可能是把 latent transitions 约束成 action model;preconditions 是长程规划稳定性的核心,而不是附属字段。
- 2)negative evidence 是 symbolic model learning 中最容易被低估的信号。
- 只拟合成功转移会学到过度可用的 actions;要让 planner 可用,必须显式学习不可用性。
- 3)一个实用方向是 neural abstraction + symbolic consistency repair:视觉模块可以先产生 noisy/over-split graph,再由 STRIPS/SAT 层压缩和纠错。
一句话总结
STRIPS-WM 是一条从视觉转移数据到 grounded propositional STRIPS model 的神经-符号管线,其真正贡献不是视觉编码,而是用 CP-SAT 和负证据把 flat latent task graph 归纳成可规划的 precondition/effect action model。
