Learning Long-Context Diffusion Policies via Past-Token Prediction

Marcel Torne, Andy Tang, Yuejiang Liu +1

Reasoning over long sequences of observations and actions is essential for many robotic tasks. Yet, learning effective long-context policies from demonstrations remains challenging…