Neural network improves prediction of complex space time patterns

Spatiotemporal Kronecker Covariance Neural Networks

Machine Learning

Summary

Predicting patterns that change over space and time is tough, especially when data is limited and relationships are complex. The authors created a new kind of neural network, called KVNN, that breaks down these patterns into separate space and time parts to understand them better. This approach improves prediction accuracy while using fewer parameters and handles noisy data well. It also fixes problems found in older methods that were either too simple or missed key relationships across time.

What this means in practice

  • For data science teams: Enhance forecasting models for datasets with complex spatial and temporal dependencies using fewer parameters and robust estimation.
  • For environmental monitoring groups: Improve predictions of environmental variables that change over space and time by capturing separate spatial and temporal relationships in sensor data.

Authors

Andrea Cavallo, Athanasios Georgoutsos, Elvin Isufi

Abstract

Multivariate time series contain complex patterns that span across both space and time. While covariance-based statistical tools like spatiotemporal Principal Component Analysis (ST-PCA) help identify these patterns, they are limited to linear operations and prone to estimation errors with limited data. Recent covariance-based spatiotemporal neural networks offer more stable, non-linear alternatives, but they ignore correlations across different time steps. To solve this, we introduce the Kronecker coVariance Neural Network (KVNN), a temporal graph neural network that represents the spatiotemporal covariance matrix via a sum of Kronecker products where spatial and temporal dependencies are decoupled. By implementing filtering operations on spatial and temporal components, KVNNs achieve expressive processing capabilities, admit a rigorous spectral analysis, and are provably stable to finite-sample estimation errors, ultimately addressing all of ST-PCA's limitations. We show on five real-world datasets that KVNNs achieve strong forecasting performance, often requiring significantly fewer trainable parameters than competitive methods, and are consistent under estimation noise.