NEURAL NETWORK WITH LANE AGGREGATION FOR LANE SELECTION PREDICTION OF MOVING OBJECTS DURING AUTONOMOUS DRIVING

In one embodiment, an autonomous driving system of an ADV perceives a driving environment surrounding the ADV based on sensor data obtained from various sensors, including detecting one or more lanes and at least a moving obstacle or moving object. For each of the lanes identified, an NN lane feature encoder is applied to the lane information of the lane to extract a set of lane features. For a given moving obstacle, an NN obstacle feature encoder is applied to the obstacle information of the obstacle to extract a set of obstacle features. Thereafter, a lane selection predictive model is applied to the lane features of each lane and the obstacle features of the moving obstacle to predict which of the lanes the moving obstacle intends to select.