Log In Sign Up

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.


page 1

page 2

page 3

page 4


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...

HTM-MAT: An online prediction software toolbox based on cortical machine learning algorithm

HTM-MAT is a MATLAB based toolbox for implementing cortical learning alg...

WARC-DL: Scalable Web Archive Processing for Deep Learning

Web archives have grown to petabytes. In addition to providing invaluabl...

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...