Time series / Model
UniTS
- Organisation
- Harvard University (MIMS lab)
- Family
- UniTS
- Released
- 2024-02-29 (arXiv v1 submission date (https://arxiv.org/abs/2403.00131v1) and GitHub release 'ckpt' published 2024-02-29 (https://github.com/mims-harvard/UniTS/releases/tag/ckpt))
- Checkpoints
- units_x128_pretrain_checkpoint.pth, units_x32_pretrain_checkpoint.pth, units_x64_supervised_checkpoint.pth
- Weights
- Open · MIT
- Links
- PaperCode
Training data disclosure: Full list
Level determined for 100% of benchmark cells.
- TestTest data seen0
- SeenSeen, split unclear0
- TrainTraining data seen8
- Related data seen4
- NoNot seen37
- ?Unknown0
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Other declared training data
-
SharePriceIncrease · split: train
Not part of the UCR 2018 / UEA 2018 archives.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
Blink · split: train
Not part of the UCR 2018 / UEA 2018 archives.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
MotionSenseHAR · split: train
Not part of the UCR 2018 / UEA 2018 archives.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
EMOPain · split: train
Not part of the UCR 2018 / UEA 2018 archives.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UCR Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UCR 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
UEA Multivariate Time Series Classification Archive (2018) · split: train of ts-classification
Single member dataset of the UEA 2018 archive (not the whole archive); training split.
Evidence
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
arxiv.org · paper · Table 7 (classification rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
NN5 · split: unknown
NN5 (Monash), loaded via gluonts; split not stated.
Evidence
“NN5P 112 [98]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 -
Electricity (ECL, 321 clients) · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“ECLP 96 [102]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
ETTh1 · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“ETTh1P 96 [127]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
Exchange-Rate · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“ExchangeP 192 [53]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
ILI (CDC influenza-like illness, national) · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“ILIP 60 [11]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
Traffic (PeMS San Francisco Bay, 862 sensors) · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“TrafficP 96 [85]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15 -
Weather (Jena, 2020; Autoformer) · split: train of long-horizon
Several forecast horizons of this dataset are separate tasks in Table 7; Time-Series-Library train split.
Evidence
“WeatherP 96 [110]”
arxiv.org · paper · Table 7 (forecasting rows) · retrieved 2026-10-15 “For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
arxiv.org · paper · Appendix (Datasets) · retrieved 2026-10-15 “_, train_loader_list = self._get_data(flag='train')”
raw.githubusercontent.com · code · exp/exp_pretrain.py · retrieved 2026-10-15
Benchmark cells
GIFT-Eval
Jena Weather (GIFT-Eval) · related dataset Weather (Jena, 2020; Autoformer) was used
Evidence (6)
“WeatherP 96 [110]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
“We incorporate Jena Weather5 dataset following Autoformer (Wu et al., 2021).”
“is recorded every 10 minutes for 2020 whole year, which contains 21 meteorological indicators, such as air temperature, humidity, etc.”
“https://www.bgc-jena.mpg.de/wetter/”
Not seen BizITObs - Application · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen BizITObs - Service · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen BizITObs - L2C · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Bitbrains - Fast Storage · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Bitbrains - rnd · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Restaurant (Recruit Restaurant Visitor Forecasting) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
ETT1 (GIFT-Eval) · related dataset ETTh1 was used
Evidence (5)
“ETTh1P 96 [127]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
“From Informer (Zhou et al., 2021) we utilize ETT1 and ETT2 datasets, which denote electricity transformer temperature”
“ETTh1&ETTh2 7 17,420”
Not seen ETT2 (GIFT-Eval) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Loop Seattle (LibCity) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen SZ-Taxi (LibCity) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M_DENSE (LibCity) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Solar (LSTNet, 137 Alabama PV plants) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Hierarchical Sales (Mancuso et al., 2020) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Yearly · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Quarterly · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Monthly · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Weekly · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Daily · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen M4 Hourly · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Hospital · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen COVID Deaths · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen US Births · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Saugeen River Flow · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Temperature Rain · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen KDD Cup 2018 · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Car Parts · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
ElectricityLoadDiagrams20112014 (UCI) · related dataset Electricity (ECL, 321 clients) was used
Evidence (6)
“ECLP 96 [102]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
“The raw dataset is in https://archive.ics.uci.edu/ml/datasets/ElectricityLoadDiagrams20112014.”
“Because the some dimensions are equal to 0. So we eliminate the records in 2011. Final we get data contains electircity consumption of 321 clients from 2012 to 2014.”
“https://archive.ics.uci.edu/ml/datasets/ElectricityLoadDiagrams20112014”
Long-horizon forecasting (ETT, Electricity, Traffic, Weather, Exchange, ILI)
Training data seen ETTh1 · only the training split was used, not the benchmark's test data
Evidence (3)
“ETTh1P 96 [127]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Not seen ETTh2 · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
ETTm1 · related dataset ETTh1 was used
Evidence (4)
“ETTh1P 96 [127]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
“we also provide the hourly-level variants for fast development (marked by *h*), i.e. ETT-small-h1 and ETT-small-h2.”
Not seen ETTm2 · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Training data seen Electricity (ECL, 321 clients) · only the training split was used, not the benchmark's test data
Evidence (3)
“ECLP 96 [102]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Training data seen Traffic (PeMS San Francisco Bay, 862 sensors) · only the training split was used, not the benchmark's test data
Evidence (3)
“TrafficP 96 [85]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Training data seen Weather (Jena, 2020; Autoformer) · only the training split was used, not the benchmark's test data
Evidence (3)
“WeatherP 96 [110]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Training data seen Exchange-Rate · only the training split was used, not the benchmark's test data
Evidence (3)
“ExchangeP 192 [53]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Training data seen ILI (CDC influenza-like illness, national) · only the training split was used, not the benchmark's test data
Evidence (3)
“ILIP 60 [11]”
“For subsets of a dataset such as ETTh1, we start by splitting the data into training and testing sets based on distinct time intervals of a long time series sequence, following splits in [112].”
“_, train_loader_list = self._get_data(flag='train')”
Time-series anomaly detection
Not seen TSB-AD-U (TSB-AD univariate) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen TSB-AD-M (TSB-AD multivariate) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen UCR Time Series Anomaly Archive (Hexagon ML/UCR, KDD Cup 2021) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Server Machine Dataset (SMD) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Mars Science Laboratory (MSL) telemetry anomaly dataset · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Soil Moisture Active Passive (SMAP) telemetry anomaly dataset · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Pooled Server Metrics (PSM) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Secure Water Treatment (SWaT) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Yahoo S5 Labeled Anomaly Detection Dataset (Yahoo Webscope S5) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Not seen Numenta Anomaly Benchmark (NAB) · absent from the complete list of training data
Evidence (3)
“For multi-task learning on forecasting and classification, we compiled 38 datasets from several sources [79, 33, 82].”
“Table 7: Multi-task datasets for classification and forecasting.”
“For transfer learning on new tasks and datasets, we load the model weights pre-trained on 38 datasets”
Time-series classification
Training data seen UCR Time Series Classification Archive (2018) · only the training split was used, not the benchmark's test data
Evidence (3)
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
“Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
“_, train_loader_list = self._get_data(flag='train')”
Training data seen UEA Multivariate Time Series Classification Archive (2018) · only the training split was used, not the benchmark's test data
Evidence (3)
“SharePriceIncrease [79] JapaneseVowels [52] SpokenArabicDigits [5] Heartbeat [61] ECG5000 [35] NonInvasiveFetalECGThorax1 [95] Blink [19] FaceDetection [40] SelfRegulationSCP2 [7] ElectricDevices [60] Trace [93] FordB [23] MotionSenseHAR [75] EMOPain [27] UWaveGestureLibrary [63] Chinatown [23] MelbournePedestrian [23] PEMS-SF [20]”
“Datasets were aggregated from the Monash Forecasting Repository [33], Time Series Classification Website [79], and Time Series Library [112]. The combined training set consists of over 35 million timesteps and over 6,000 variables.”
“_, train_loader_list = self._get_data(flag='train')”
Notes
Self-supervised (x32/x128) and multi-task supervised (x64) checkpoints, all trained on the training splits of the 38 Table 7 tasks (20 forecasting tasks over 7 datasets + 18 classification datasets). Classification members: 7 UCR and 7 UEA datasets. No anomaly data in pre-training; anomaly detection is few-shot fine-tuning (see units-few-shot-anomaly-detection). Other released fine-tuned checkpoints (few-shot new-data classification/forecasting incl. ECG200, SelfRegulationSCP1, RacketSports, Handwriting, Epilepsy, StarLightCurves from Table 8; imputation) are not separate entries here.
Sources
- UniTS: A Unified Multi-Task Time Series Model (arXiv v3, NeurIPS 2024) · paper · retrieved 2026-10-15
- mims-harvard/UniTS README · readme · retrieved 2026-10-15
- UniTS GitHub releases (checkpoint assets) · code · retrieved 2026-10-15
- UniTS pre-training data config · code · retrieved 2026-10-15