CNAI4S
智算平台热点追踪算子框架领域模型科学任务智算套件数据资源智算评测协作社区运行情况

友情链接

科技部工信部北京航空航天大学浙江大学
中山大学北京大学华中科技大学之江实验室
百度飞桨华为昇思中科可控
版权所有:科学智算共性平台 京ICP备18020905号-2
京公网安备11010802046602号
联系邮箱: tianyuc@buaa.edu.cn
AD-WM: Action-Discriminative World Models for Counterfactual Model Predictive Control
返回热点追踪
新闻AI

AD-WM: Action-Discriminative World Models for Counterfactual Model Predictive Control

a
arXiv AI4S Papers
2026年9月25日技术突破
摘要

Latent world models are typically trained to predict factual transitions, whereas model predictive control (MPC) must compare alternative actions from the same state. A model can t

Latent world models are typically trained to predict factual transitions, whereas model predictive control (MPC) must compare alternative actions from the same state. A model can therefore achieve low factual prediction error yet poorly distinguish candidate actions. We introduce AD-WM, an action-discriminative joint-embedding world model for counterfactual MPC. AD-WM combines residual latent dynamics with predictor-level action-recovery regularization, using inverse dynamics and a normalized recovery objective motivated by conditional mutual information. Both objectives encourage planning transitions to preserve action information; their auxiliary heads are discarded at test time, leaving MPC unchanged. On OGBench-Cube, AD-WM improves hard-start success from 3.7% to 52.0% over a matched LeWM baseline and improves mean success over the reproduced baseline in four of five simulation environments. Planning diagnostics show that factual prediction error and whole-bank action ranking do not follow the closed-loop success ordering, whereas CEM-aligned elite regret tracks success more closely. With a frozen V-JEPA 2 encoder and matched DROID post-training, AD-WM also improves zero-shot transfer to our Franka setup, increasing basic pick-and-place success from 42.2% to 71.1% without lab-specific adaptation. These results suggest that world models for planning should preserve action-dependent differences needed for counterfactual selection, rather than optimize factual prediction accuracy alone. More videos and code are available at https://ad-wm.github.io/.

查看原文 →

相关推荐

Estimating suicide risk from text
新闻AI

Estimating suicide risk from text

MIT

LLM Agents Can Easily Tamper With Their Own Traces
新闻AI

LLM Agents Can Easily Tamper With Their Own Traces

Jeremy Qin, David Schmotz, Derck Prinzhorn

Agentic Detection of Online Conspiracies
新闻AI

Agentic Detection of Online Conspiracies

Lior Biton, Oren Tsur

To Trust or Not to Trust: Retrieval-Augmented Fact Checking in Speech
新闻AI

To Trust or Not to Trust: Retrieval-Augmented Fact Checking in Speech

Debajyoti Mazumder, Mamta, Abhirama Subramanyam Penamakuri