TensoCast — Traffic forecasting
Tensor decompositions capturing spatial-temporal structure in traffic sensor data, improving prediction accuracy by 35% over the baseline.
- Role
- ML Engineer
- Timeline
- 2024
- Status
- Research project
- Area
- Research / forecasting
Overview
A research project exploring whether tensor decompositions can capture spatial-temporal structure before downstream forecasting. It covers preprocessing, decomposition, model comparison, and evaluation in Python on the METR-LA traffic dataset.
Problem
Traffic data carries spatial and temporal structure that many baseline approaches do not model well.
Approach
Tensor decomposition to represent multidimensional traffic dynamics, with predictive models layered on top for forecasting.
01
Tensorize
METR-LA sensor data
02
Decompose
Low-rank tensor methods
03
Forecast
LSTM, XGBoost
04
Evaluate
Against the baseline
Tensor preprocessing and decomposition feed PyTorch and classical ML models, with experiments comparing performance across modeling choices.
Outcomes
- Improved prediction accuracy by 35% over the baseline.
- Combined tensor decomposition and predictive models in a single pipeline.
- Compared decomposition-based models against classical baselines in the same experimental setup.
Challenges
- Choosing decompositions that preserve useful structure.
- Balancing interpretability with predictive performance.
- Handling noisy and incomplete sensor data.
Stack
Python · PyTorch · NumPy · scikit-learn · XGBoost