arXiv:2606.09028cs.CVcs.AI2026-06被引 2

用轻量矩阵诊断潜在世界模型的行动一致性,提速百倍评估效率。

ATM: Action-Consistency Transfer Matrix for Diagnosing and Improving Latent World Models

论文配图:ATM: Action-Consistency Transfer Matrix for Diagnosing and Improving Latent World Models
图 1 · 摘自论文原文
  • 提出ATM矩阵,通过对比真实与预测动作信息诊断模型表征质量。
  • 在非平凡成功差距下实现可靠排序,评估速度提升超100倍。
  • 可作为训练信号优化规划性能,无需改动原有规划器。

潜在世界模型广泛用于控制与目标条件规划,但评估其表征是否适合规划通常需依赖慢速、耦合规划器的模拟器评估(如CEM),耗时数分钟至数小时,且过程黑箱、受模型复杂度影响。本文提出ATM(Action-Consistency Transfer Matrix),一种轻量级后验探针方法,通过比较真实编码转移与模型预测转移中的动作信息,生成可解释矩阵,揭示表征质量、转移域不一致及失败模式,无需模拟回放。该矩阵可压缩为简单筛选得分,用于任务内检查点、变体与模型间的排序。当真实成功差距显著时,ATM实现高可靠性成对排名,将原本数分钟至数小时的CEM评估压缩至秒级转移分析,在本研究中提速超过100倍。进一步引入AITS,证明动作可辨识性不仅是诊断工具,还可作为有效训练信号改进下游规划,无需修改规划器。

原文摘要 · Abstract (English)

Latent world models are increasingly used for control and goal-conditioned planning, yet assessing whether their learned representations are useful for planning usually requires slow, planner-coupled simulator evaluation with CEM or similar planners. Such evaluation is black-box and model-complexity-dependent: under the same protocol, different world models may require minutes to hours per checkpoint. In this work, we propose ATM, an Action-Consistency Transfer Matrix for diagnosing whether latent transitions preserve action semantics relevant to planning. ATM compares action information in real encoded transitions and model-predicted transitions through lightweight post-hoc probes, producing an interpretable matrix that reveals representation quality, transition-domain inconsistency, and failure modes without simulator rollout. It can also be collapsed into a simple screening score for within-task ranking across checkpoints, variants, and world models. When the true success gap is non-trivial, ATM achieves highly reliable pairwise ranking, while reducing minutes-to-hours CEM evaluation to seconds-level transition analysis, yielding more than 100x speedup in our setup. We further introduce AITS, showing that action-identifiability is not only diagnostic but also a useful training signal for improving downstream planning without changing the planner.

世界模型规划评估动作一致性高效诊断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。