PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models

by   Benedek Rozemberczki, et al.

We present PyTorch Geometric Temporal a deep learning framework combining state-of-the-art machine learning algorithms for neural spatiotemporal signal processing. The main goal of the library is to make temporal geometric deep learning available for researchers and machine learning practitioners in a unified easy-to-use framework. PyTorch Geometric Temporal was created with foundations on existing libraries in the PyTorch eco-system, streamlined neural network layer definitions, temporal snapshot generators for batching, and integrated benchmark datasets. These features are illustrated with a tutorial-like case study. Experiments demonstrate the predictive performance of the models implemented in the library on real world problems such as epidemiological forecasting, ridehail demand prediction and web-traffic management. Our sensitivity analysis of runtime shows that the framework can potentially operate on web-scale datasets with rich temporal features and spatial structure.



There are no comments yet.


page 1

page 2

page 3

page 4


Chickenpox Cases in Hungary: a Benchmark Dataset for Spatiotemporal Signal Processing with Graph Neural Networks

Recurrent graph convolutional neural networks are highly effective machi...

PyTorch Tabular: A Framework for Deep Learning with Tabular Data

In spite of showing unreasonable effectiveness in modalities like Text a...

Mlr3spatiotempcv: Spatiotemporal resampling methods for machine learning in R

Spatial and spatiotemporal machine-learning models require a suitable fr...

Signal-Envelope: A C++ library with Python bindings for temporal envelope estimation

Signals can be interpreted as composed of a rapidly varying component mo...

Deep Learning for Spatiotemporal Modeling of Urbanization

Urbanization has a strong impact on the health and wellbeing of populati...

Data-driven Algorithm Selection and Parameter Tuning: Two Case studies in Optimization and Signal Processing

Machine learning algorithms typically rely on optimization subroutines a...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.