← Back to Machine Learning (Statistics)
stat.ML

Teaching neural networks to predict ship movements with built-in uncertainty

Jaeyeong Lee, Wonmo Koo, Heeyoung Kim

June 4, 2026

Predicting ship trajectories from sparse, irregular tracking data is crucial for maritime safety, but standard neural approaches don't quantify uncertainty reliably. This work adds structure to Bayesian Neural ODEs by imposing Gaussian process priors on the vector field rather than just weights, encoding smoothness and locality. Combined with probabilistic multiple shooting for long trajectories, the method delivers better-calibrated uncertainty estimates on real AIS datasets.
Published as Function-Space Priors for Bayesian Neural ODEs with Application to Vessel Trajectory Prediction arXiv:2606.06351
Read the original paper →