Hybrid Neural Network Approach with Physical Constraints for Predicting the Potential Occupancy Set of Surrounding Vehicles

The reliable and uncertainty-aware prediction of surrounding vehicles remains a key challenge in autonomous driving. However, existing methods often struggle to quantify and incorporate uncertainty effectively. To address these challenges, we propose a hybrid architecture that combines a data-driven neural trajectory predictor with physically grounded constraints to forecast future vehicle occupancy. Specifically, the physical constraints are derived from vehicle kinematic principles and embedded into the network as additional loss terms during training. This integration ensures that predicted trajectories conform to feasible and physically realistic motion boundaries. Furthermore, a mixture density network (MDN) is employed to estimate predictive uncertainty, transforming deterministic trajectory predictions into spatial probability distributions. This enables a probabilistic occupancy representation, offering a richer and more informative description of the potential future positions of surrounding vehicles. The proposed model is trained and evaluated on the Aerial Dataset for China’s Congested Highways and Expressways (AD4CHE), which contains representative driving scenarios in China. Experimental results demonstrate that the model achieves strong fitting performance while maintaining high physical plausibility in its predictions.