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

SwiftTS: A Swift Selection Framework for Time Series Pre-trained Models via Multi-task Meta-Learning

About

Pre-trained models exhibit strong generalization to various downstream tasks. However, given the numerous models available in the model hub, identifying the most suitable one by individually fine-tuning is time-consuming. In this paper, we propose \textbf{SwiftTS}, a swift selection framework for time series pre-trained models. To avoid expensive forward propagation through all candidates, SwiftTS adopts a learning-guided approach that leverages historical dataset-model performance pairs across diverse horizons to predict model performance on unseen datasets. It employs a lightweight dual-encoder architecture that embeds time series and candidate models with rich characteristics, computing patchwise compatibility scores between data and model embeddings for efficient selection. To further enhance the generalization across datasets and horizons, we introduce a horizon-adaptive expert composition module that dynamically adjusts expert weights, and the transferable cross-task learning with cross-dataset and cross-horizon task sampling to enhance out-of-distribution (OOD) robustness. Extensive experiments on 14 downstream datasets and 8 pre-trained models demonstrate that SwiftTS achieves state-of-the-art performance in time series pre-trained model selection.

Tengxue Zhang, Biao Ouyang, Yang Shu, Xinyang Chen, Chenjuan Guo, Bin Yang• 2025

Related benchmarks

TaskDatasetResultRank
Model SelectionETTh1
Weighted Kendall's Tau0.543
32
Model SelectionETTm2
Weighted Kendall's Tau1
32
Model SelectionTraffic
Weighted Kendall's Tau0.702
32
Model SelectionZafNoo
Weighted Kendall's Tau0.786
32
Model SelectionCzeLan
Weighted Kendall's Tau0.847
32
Model SelectionAQShunyi
Weighted Kendall's Tau (w)0.939
32
Model SelectionPeMS08
Weighted Kendall's Tau0.505
32
Model SelectionAverage
Weighted Kendall's Tau (w)0.47
32
Model SelectionETTh2
Weighted Kendall's Tau0.436
32
Model SelectionETTm1
Weighted Kendall's Tau0.652
32
Showing 10 of 15 rows

Other info

Follow for update