Closing the Train-Test Gap in World Models for Gradient-Based Planning

Arjun Parthasarathy, Nimit Kalra, Rohun Agrawal +4

World models paired with model predictive control (MPC) can be trained offline on large-scale datasets of expert trajectories and enable generalization to a wide range of planning…