动作经验字典助力世界动作模型
Learning Skills from Historical Action Trajectories: Action Experience Dictionary for World Action Models
清华团队提出AED,让AI模型能复用历史动作经验,提升跨任务动作预测能力。
研究人员开发了动作经验字典(AED),将历史物理动作轨迹编码为共享动作嵌入。该方法在模拟基准和跨 embodiment 真实世界场景中验证有效。AED 通过预训练动作检索器获取嵌入,并通过交叉注意力进行视觉条件化。研究引入了感知运动转换损失,监督随机时间间隔内视觉特征变化的预测。
Learning Skills from Historical Action Trajectories: Action Experience Dictionary for World Action Models
World Action Models (WAMs) couple visual dynamics prediction with action generation, yet they do not explicitly support the reuse of action experience across manipulation tasks. Furthermore, existing WAMs struggle to capture underlying cross-task semantic relationships that could guide target action prediction, as redundant background elements interfere with the extraction of key visual information. To address these challenges, we develop a novel Action Experience Dictionary (AED) that encodes historical physical action trajectories into shared action embeddings to support skill reuse and model cross-task relationships. Specifically, we first aggregate historical actions to align with visual observations and retrieve action embeddings from the AED using a pretrained action tokenizer. Subsequently, we visually condition the pooled embeddings through cross-attention and prepend them to noisy action tokens, providing interaction context and action intent for prediction. To model action-related motion and reduce reliance on irrelevant background cues, we introduce a motion-aware transition loss that supervises visual feature change prediction over random temporal intervals. Experiments on simulation benchmarks and in real-world cross-embodiment settings verify the effectiveness of our AED. The anonymous project website is available at \href{https://github.com/JiahuaDong/AED}{AED}.