Search papers, labs, and topics across Lattice.
The paper introduces Keyframe-Chaining VLA (KC-VLA), a novel framework designed to address the limitations of existing Vision-Language-Action (VLA) models in handling long-horizon robot manipulation tasks with Non-Markovian dependencies. KC-VLA employs an automatic keyframe selector to identify distinct state transitions and a progress-aware query mechanism to dynamically retrieve task-relevant historical frames, which are then integrated as visual tokens to ground the policy. Experiments on a new suite of Non-Markovian manipulation tasks in the ManiSkill simulator demonstrate that KC-VLA significantly outperforms existing methods by explicitly modeling long-horizon temporal context.
Forget relying on immediate observations: Keyframe-Chaining VLA lets robots nail long-horizon tasks by remembering and chaining together only the *important* past states.
Existing Vision-Language-Action (VLA) models often struggle to generalize to long-horizon tasks due to their heavy reliance on immediate observations. While recent studies incorporate retrieval mechanisms or extend context windows to handle procedural tasks, they often struggle to capture Non-Markovian dependencies, where optimal actions rely solely on specific past states rather than the current observation. To address this, we introduce Keyframe-Chaining VLA, a framework that extracts and links key historical frames to model long-horizon dependencies. Specifically, we propose an automatic keyframe selector that learns a discriminative embedding space, effectively identifying distinct state transitions. To capture task-critical information, we design a progress-aware query mechanism that dynamically retrieves historical frames based on their temporal relevance to the current execution phase. These selected keyframes are integrated into the VLA as interleaved visual tokens, explicitly grounding the policy in the long-horizon temporal context. Finally, we introduce a suite of four Non-Markovian manipulation tasks built upon the ManiSkill simulator to measure task success rates. Experimental results demonstrate that our method achieves superior performance, effectively tackling robot manipulation tasks characterized by long-horizon temporal dependencies. Code is available at https://github.com/cytoplastm/KC-VLA.
