Search papers, labs, and topics across Lattice.
This paper introduces a Deconstruct-Recompose Paradigm (DRP) for pre-training reinforcement learning agents using local motion representations derived from videos. By focusing on local motion patterns rather than treating the agent as a single entity, the method enhances transferability across different domains and improves sample efficiency. Experimental results show that DRP significantly outperforms existing global modeling approaches in various robotic control and manipulation tasks.
Local motion representations can drastically improve reinforcement learning efficiency and transferability across diverse tasks, challenging the conventional global modeling approach.
Pre-training on large-scale videos to improve reinforcement learning efficiency is promising yet remains challenging. Existing methods typically treat the agent as an indivisible entity, modeling motion patterns globally. Such global modeling is tightly coupled with the morphology, hindering transfer across domains. In contrast, despite the vast disparity in global motions, the local components exhibit similar motion patterns across different agents. Building on this insight, we propose a novel Deconstruct-Recompose Paradigm (DRP) for learning transferable local motion representations. Specifically, in the Deconstruct phase, we identify multiple local points and track their frame-wise motions, defining each as an Atomic Action. We introduce a Dual-Attention Encoder (DAE) to learn local motion representations from these Atomic Actions, capturing their spatiotemporal relationships. In the Recompose phase, we compose local motion representations with a learnable Motion Aggregation Token [MAT] via latent dynamics model learning. Additionally, an adapter bridges local motion and downstream action-specific dynamics to accelerate policy learning. Extensive experiments demonstrate that our method effectively transfers to diverse robotic control and manipulation tasks, significantly improving sample efficiency and performance.