HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

LightCTS: A Lightweight Framework for Correlated Time Series Forecasting

Zhichen Lai; Dalin Zhang; Huan Li; Christian S. Jensen; Hua Lu; Yan Zhao

LightCTS: A Lightweight Framework for Correlated Time Series Forecasting

Abstract

Correlated time series (CTS) forecasting plays an essential role in many practical applications, such as traffic management and server load control. Many deep learning models have been proposed to improve the accuracy of CTS forecasting. However, while models have become increasingly complex and computationally intensive, they struggle to improve accuracy. Pursuing a different direction, this study aims instead to enable much more efficient, lightweight models that preserve accuracy while being able to be deployed on resource-constrained devices. To achieve this goal, we characterize popular CTS forecasting models and yield two observations that indicate directions for lightweight CTS forecasting. On this basis, we propose the LightCTS framework that adopts plain stacking of temporal and spatial operators instead of alternate stacking that is much more computationally expensive. Moreover, LightCTS features light temporal and spatial operator modules, called L-TCN and GL-Former, that offer improved computational efficiency without compromising their feature extraction capabilities. LightCTS also encompasses a last-shot compression scheme to reduce redundant temporal features and speed up subsequent computations. Experiments with single-step and multi-step forecasting benchmark datasets show that LightCTS is capable of nearly state-of-the-art accuracy at much reduced computational and storage overheads.

Code Repositories

ai4cts/lightcts
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
correlated-time-series-forecasting-onLightCTS
FLOPs(M): 239
Parameters(K): 27
correlated-time-series-forecasting-on-metr-laLightCTS
FLOPs(M): 71
MAE @ 12 step: 3.42
MAPE @ 12 step: 9.46%
Parameters(K): 133
RMSE @ 12 step: 0.0721
correlated-time-series-forecasting-on-pemsLightCTS
FLOPs(M): 208
MAE @ 12 step: 1.89
MAPE @ 12 step: 4.39
Parameters(K): 236
RMSE @ 12 step: 4.32
correlated-time-series-forecasting-on-solarLightCTS
FLOPs(M): 169
Parameters(K): 38
traffic-prediction-on-pems04LightCTS
FLOPs(M): 147
MAE: 18.79
MAPE: 12.8%
Parameters(K): 185
RMSE: 0.3014
traffic-prediction-on-pems08LightCTS
FLOPs(M): 70
MAE: 14.63
MAPE: 9.43%
Parameters(K): 177
RMSE: 0.2349

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
LightCTS: A Lightweight Framework for Correlated Time Series Forecasting | Papers | HyperAI