精读笔记

Problem Setting

论文处理的是大规模 PDDL 任务规划中的 object pruning：给定固定 symbolic domain、初始状态和目标，从大量 symbolic objects 中选出足够小但仍可支持可行计划的对象子集。这里的“对象”不仅是物体，也包括位置、package、vehicle、hub 等 PDDL entity。

真正困难不是 classical planner 找不到解，而是 full grounding / heuristic search 在长视距、多对象、多逻辑约束下成本过高；但如果用 learned scorer 预先裁剪对象，裁剪错误又会直接改变问题本身。漏掉一个关键 object，子问题不可解；保留太多 irrelevant objects，搜索空间又膨胀。以前 PLOI / Flax 的卡点在于 scorer 学的是 full-space planner 产生的静态标签，而部署时面对的是自己预测出来的 pruned space。这使得 object pruning 不是一个普通 supervised relevance prediction，而是一个闭环决策问题。

关键矛盾可以概括为：规划效率要求 aggressive pruning，规划可解性要求 conservative coverage；prior 的训练目标没有对这个部署时 trade-off 负责。

Motivation

已有路线不够的根本原因是训练分布和测试分布不一致。离线 full-space supervision 告诉 scorer“某个完整问题里的计划用了哪些对象”，但部署时 scorer 的排序会决定哪些对象先进入 active set，planner 看到的是一系列由 scorer 错误诱导出的子问题。长视距规划中这种 exposure bias 会被 expansion 过程放大：早期错排对象会导致 timeout、错误扩张或无意义 repair，最终不是单点分类错误，而是整个 search trajectory 偏移。

作者的核心观察是：object importance 的价值只通过下游 planner 的有限时间搜索体现，因此训练信号也应该来自 planner 在 scorer 当前诱导的 pruned space 中实际找到的计划，而不是来自一个与部署过程脱节的 full-space teacher。缺的不是更强的 GNN，而是把 scorer 和 planner 放进同一个反馈回路，并且让下层 planner 在早期 scorer 很差时仍能返回可用 supervision。

Core Idea

论文真正核心是把 object-importance learning 改写成 imperative bilevel loop：上层神经 scorer 预测每个对象的重要性，下层 symbolic planner 按这些分数构造 pruned planning problem 并搜索可行计划；若找到计划，就把计划中出现的对象作为动态 pseudo-label 回传训练 scorer。这样 scorer 学到的不是某个离线 full-space 解的对象集合，而是在自身 pruning policy 诱导的 search distribution 下，哪些对象会帮助 planner 找到解。

这带来的本质变化是监督信息流的重组。prior 是 planner 离线生成标签，模型上线后单向影响 planner；iFlax 是模型影响 planner，planner 的返回结果再影响模型。这个闭环引入的 inductive bias 是“对象重要性应相对于当前 pruned search process 定义”，而不是相对于原始完整问题定义。直觉上它更 scalable，因为训练目标直接惩罚那些会让有限预算 planner 失败或膨胀的排序；也更可能泛化到更大 object set，因为 learned policy 学的是 expansion/recovery 过程中的相对对象优先级，而不是固定规模下的标签模式。

但需要明确：这不是严格意义上 differentiable bilevel optimization。planner 是黑盒离散求解器，所谓上层优化主要是用下层成功计划生成 BCE pseudo-label。理论说法更像 predict-then-optimize / DAgger-style self-training 的规划版本。

Method

方法的关键机制可以压缩为三个层次。

第一，object scorer 解决“对象进入搜索空间的顺序”问题。PDDL task 被编码成 object-centric relational graph，GNN 输出每个 object 的 importance score。这个模块本身不是创新重点；它的必要性在于提供一个可排序、可跨 object count 泛化的 pruning prior。核心变化不是 graph network，而是 scorer 的训练标签来自当前 planner feedback。

第二，score-guided iterative expansion 解决“如何把分数转成规划收益”问题。planner 不直接在 full object set 上搜索，而是从 goal objects 开始，按 score threshold 逐步加入对象。这个机制把 object relevance prediction 变成一种 search-space curriculum：先尝试小问题，失败再扩大。它是节省 planning time 的主要接口，也让 scorer 的排序质量直接影响 search cost。

第三，3R recovery 解决“闭环训练下 planner 经常不给信号”问题。Repair 用 relaxed plan 补漏掉的关键对象；Restart 丢弃被污染的 active set，从 relaxed sketch 重新构造更干净的 base set；Rollback 回退到上一个 active set 并细粒度加入对象，避免一次 expansion 跨过 tractability boundary。三者不是单纯工程补丁，而是对应三类不同 failure mode：incompleteness、bloat、over-expansion。它带来的核心变化是让下层 planner 在 scorer 早期不可靠时仍能较高概率返回可行 plan，从而避免训练只依赖少数成功样本产生偏置。

训练时，成功计划被转成 object mask，用 BCE 更新 scorer；部署时 scorer 固定，planner 使用相同 expansion + 3R，只是不再更新参数。

Key Insight / Why It Works

这篇最重要的 insight 是：object pruning 的学习目标必须和 planner 实际运行的 pruned distribution 对齐。full-space plan-used objects 并不等于 finite-budget pruned planner 所需要的对象集合；后者取决于 scorer 的排序、expansion 策略、planner heuristic 以及 recovery 行为。iFlax 有效的核心原因正是把 supervision 移到这个闭环分布上。

最可能的核心贡献是“adaptive pseudo-supervision + recovery-stabilized lower-level planning”的组合，而不是单独的 GNN scorer。消融也支持这一点：仅加 3R 可以明显改善 Flax，但闭环训练配合 3R 才出现最大收益；没有 3R 的 iFlax 反而仍然脆弱，说明在线学习本身不是免费的，必须有稳定的下层 oracle 近似。

从机制归因看，它本质上混合了几种东西： 1. representation alignment：训练标签来自部署时同类 pruned problems，缓解 exposure bias； 2. curriculum / active set scheduling：iterative expansion 让 planner 先在小空间里找解； 3. test-time compute / portfolio search：3R 并行 recovery 实际增加了策略多样性和预算利用效率； 4. retrieval-like object selection：scorer 更像在 relational graph 中检索可能进入 causal chain 的 objects，而不是执行显式长程推理； 5. domain-structured weak supervision：Relax/Comp rules 提供了隐藏的 domain prior。

需要直说的是，论文里“bilevel optimization”的表述比实际机制更强。没有看到对离散 planner 的真正优化梯度，也没有证明 pseudo-label 的 fixed point 对应某种最优 pruning policy。它更像把 planner 作为在线标签生成器，用成功计划不断重标注 scorer。有效性来自 distribution correction 和 recovery coverage，而不是一个严格可解的 bilevel objective。

3R 中最有价值的可能是 Restart 和 Rollback，而 Repair 更接近 Flax 已有思想。Table VI 也显示 Restart 贡献最大，这说明很多失败不是因为缺一个对象，而是 active set 被错误对象污染。这个观察很重要：object pruning 领域常把 failure 理解成 false negative，但实际 false positive 同样会杀死 planning efficiency。

Relation To Prior Work

最接近的是 PLOI 和 Flax：都学习 object importance，用对象裁剪降低 symbolic planner 的搜索空间。与 PLOI 的差异是 iFlax 不只是预测 included objects，而是把预测放进 iterative planning feedback loop；与 Flax 的差异更直接：Flax 仍主要依赖 fixed offline supervision 和单一路径 relaxation repair，而 iFlax 用当前 scorer 诱导的 pruned plan 生成动态 supervision，并把 recovery 扩展成 3R portfolio。

从技术谱系看，它属于 learned pruning / learned relevance for symbolic planning，而不是学习 symbolic model、学习 skills、或端到端 TAMP。它也借用了 predict-then-optimize / imperative learning 的语言，但实际更接近 self-training with an optimization oracle：planner 不是 differentiable layer，而是一个会产生 pseudo-label 的 black-box teacher。

哪些看似新其实是重组：GNN object scorer、relaxed plan 修复、iterative expansion、portfolio-style recovery 都不是全新概念。实质创新在于把这些组合成一个能闭环训练的系统，并明确针对 object-pruned planning 的 exposure bias。尤其是把 recovery 不只当 test-time fallback，而当 training signal stabilizer，这是比单纯工程增强更有意义的部分。

Dataset / Evaluation

评估覆盖了三类结构差异较大的 PDDL benchmark：MazeNamo 偏空间重排和异质 affordance，SokoMindPlus 偏不可逆推箱依赖，LogisticsPlus 偏交通/资源/锁钥式 symbolic coupling。这比只在 MazeNamo 上做结果更有说服力，基本支持“不是单一 domain trick”的 claim。训练小规模、测试更大规模，也在一定程度上检验 object-centric scorer 的 size generalization。

消融设计相对关键：Flax、Flax3R、iFlax w/o 3R、iFlax 能分离出“在线训练”和“3R recovery”的作用。验证曲线中的 FR、OSR、SSP 也比单纯报 planning time 更接近论文 claim：iFlax 不只是成功更多，还倾向于用更少对象和更少 search effort 找解。

但 evaluation 仍有明显限制。第一，Relax/Comp rules 是 domain-specific 的，benchmark 间迁移并不等于无工程泛化。第二，real-world / Isaac Sim 主要证明系统可跑，不足以证明 planner 对连续可行性、感知误差和 motion cost 有内生建模。第三，3R 是并行 recovery，和 baseline 的 time-budget / compute fairness 需要非常仔细；文中虽报告同机同预算，但增益中有多少来自更好的 scorer、有多少来自更强 test-time portfolio，仍不完全清楚。第四，pseudo-label 来自同一个 planner，可能强化 planner heuristic bias；benchmark 若与 relaxation prior 高度匹配，泛化会显得更好。

Limitation

核心前提很强：必须有预定义 PDDL domain、离散 predicates、有限动作集合、可枚举 objects，以及能把场景可靠转成 symbolic problem 的 perception pipeline。方法不处理连续动力学、几何可达性和 motion-level cost；真实系统中 task planning 与 motion planning 仍是分离的，因此 high-level plan 可行不代表执行最优或鲁棒。

更深的限制是，iFlax 把 planning scalability 问题部分转移成“设计好的 object abstraction + relaxation/complement rules + recovery portfolio”问题。Relax/Comp 的质量直接影响 Repair/Restart；Table V 已经显示 relaxation rule 不同会显著改变结果。这说明方法并非完全 domain-general，至少需要领域专家提供能保留 causal skeleton 的 relaxation。文中未充分说明在更复杂 domain 中如何系统设计这些规则。

泛化也要谨慎理解。GNN 支持不同 object count，但不等于学会通用逻辑推理。很多能力可能来自训练分布覆盖了相似的 relational motifs，scorer 学到的是对象检索/排序 heuristic，而非显式长期状态建模。尤其在多个可行计划、强资源竞争、动态环境、或 motion infeasibility 会反向影响 symbolic relevance 的情况下，plan-used-object pseudo-label 可能非常不稳定。

另一个问题是目标函数偏向 feasibility 和 speed，而非 optimality。pseudo-label 只标记当前返回计划中的对象，会惩罚未出现在该计划但可能支持更短、更稳健或 motion-feasible 计划的对象。这可能让模型收敛到 planner heuristic 偏好的局部 plan family。论文没有给 formal guarantee，也没有充分讨论 pseudo-label 噪声如何影响长期训练。

最后，增益归因仍不完全干净。iFlax 的优势可能同时来自 distribution alignment、更多训练回合中的 planner interaction、更强 recovery、并行 test-time compute、domain-specific relaxed hints。论文消融已经做得不错，但仍不能完全分离这些因素。

Takeaway

• 1. 对 learned pruning 类规划方法，关键不是让 relevance classifier 更准，而是让训练分布匹配 scorer 自己诱导的 pruned search distribution。
• 这个 insight 可迁移到 grounding reduction、TAMP object selection、constraint activation、MILP variable selection 等问题。
• 2. 在线 planner feedback 必须配套 recovery/stabilization。
• 否则早期模型错误会让下层 solver 不返回解，self-training 信号会变稀疏且有偏。

一句话总结

iFlax 是 object-pruned symbolic planning 从离线 relevance imitation 走向闭环 solver-feedback training 的一次系统化推进，真正贡献在于用 recovery-stabilized pseudo-supervision 缓解 pruning planner 的 exposure bias，而不是提出了一个新的规划器或新的 GNN。