arxiv Towards Generalizable and Interpretable Motion Prediction: A Deep Variational Bayes Approach