Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

TS-Memory: Plug-and-Play Memory for Time Series Foundation Models

About

Time Series Foundation Models (TSFMs) achieve strong zero-shot forecasting through large-scale pre-training, but adapting them to downstream domains under distribution shift remains challenging. Existing solutions face a trade-off: Parametric Adaptation can cause catastrophic forgetting and requires costly multi-domain maintenance, while Non-Parametric Retrieval improves forecasts but incurs high inference latency due to datastore search. We propose Parametric Memory Distillation and implement it as TS-Memory, a lightweight memory adapter that augments frozen TSFMs. TS-Memory is trained in two stages. First, we construct an offline, retrieval-leakage-safe kNN teacher that synthesizes confidence-aware quantile targets from retrieved futures. Second, we distill this retrieval-induced distributional correction into a lightweight memory adapter via confidence-gated supervision. During inference, TS-Memory fuses memory and backbone predictions with constant-time overhead, enabling retrieval-free deployment. Experiments across diverse TSFMs and benchmarks demonstrate consistent improvements in both point and probabilistic forecasting over representative adaptation methods, with efficiency comparable to the frozen backbone. Code: https://github.com/sisuolv/TS-Memory.

Sisuo Lyu, Siru Zhong, Tiegang Chen, Weilin Ruan, Qingxiang Liu, Taiqiang Lv, Qingsong Wen, Raymond Chi-Wing Wong, Yuxuan Liang• 2026

Related benchmarks

TaskDatasetResultRank
Long-term forecastingETTh1
MSE0.421
500
Long-term forecastingETTm2
MSE0.267
391
Long-term forecastingETTh2
MSE0.354
376
Long-term time-series forecastingWeather (test)
MSE0.205
240
Long-term time-series forecastingTraffic (test)
MSE0.367
182
Long-term forecastingElectricity
MSE0.155
167
Long-term forecastingExchange (test)
MAE0.406
147
Long-term forecastingElectricity (test)
MSE0.144
85
Long-term forecastingETTh2 v1 (test)
MSE0.34
36
Long-term forecastingETTh1 v1 (test)
MSE0.396
28
Showing 10 of 14 rows

Other info

Follow for update