We work on artificial intelligence and machine learning methods to improve autonomy and traffic management in intelligent transportation systems (ITS).

Evaluating STGNNs for air transportation networks

Visualization of the US air transportation network studied in our work.

This work builds upon our earlier work using STGNNs for railway networks by exploring potential benefits of STGNNs with various graph convolution operations towards delay prediction in air transportation networks.

Predicting delays in railway networks

Notional view of an STGNN architecture and the portion of the British railway network studied in our work.

This work leverages spatio-temporal graph neural networks (STGNNs) to predict delays in railway networks. We find that STGNNs improve predication accuracy relative to linear and vanilla multi-layer perceptron models. See our ITSC paper presenting our results for more details.