AUTONOMOUS VEHICLE PREDICTION LAYER TRAINING

The present disclosure generally relates to autonomous vehicle (AV) training and, more specifically, to AV prediction layer training. In some aspects, the present disclosure provides a process for receiving road data representing a real-world environment encountered by an AV and generating, using a prediction layer of the AV, a predicted trajectory of a target vehicle, wherein the predicted trajectory comprises one or more waypoints and wherein the predicted trajectory is based on the road data. In some aspects, the process can further include steps for calculating a distance metric for the predicted trajectory, wherein the distance metric is based on a distance between the one or more waypoints and one or more corresponding drivable areas and updating the prediction layer of the AV based on the distance metric. Systems and machine-readable media are also provided.