PREDICTING AGENT TRAJECTORIES IN THE PRESENCE OF ACTIVE EMERGENCY VEHICLES

Methods, systems, and apparatus, including computer programs encoded on a computer storage medium, that obtain scene features in an environment that includes an autonomous vehicle, a first target agent, and a second target agent, and determines whether the first target agent is an emergency vehicle that is active at a current time point. In response to determining that the first target agent is an emergency vehicle that is active at the current time point, an input is generated from the scene features. The input can characterize the scene and indicate that the first target agent is an emergency vehicle that is active at the current time point. Also in response, the input can be processed using a machine learning model that is configured to generate a trajectory prediction output for the second target agent that characterizes predicted future behavior of the second target agent after the current time point.