Joint Embeddings Go Temporal
About
Self-supervised learning has seen great success recently in unsupervised representation learning, enabling breakthroughs in natural language and image processing. However, these methods often rely on autoregressive and masked modeling, which aim to reproduce masked information in the input, which can be vulnerable to the presence of noise or confounding variables. To address this problem, Joint-Embedding Predictive Architectures (JEPA) has been introduced with the aim to perform self-supervised learning in the latent space. To leverage these advancements in the domain of time series, we introduce Time Series JEPA (TS-JEPA), an architecture specifically adapted for time series representation learning. We validate TS-JEPA on both classification and forecasting, showing that it can match or surpass current state-of-the-art baselines on different standard datasets. Notably, our approach demonstrates a strong performance balance across diverse tasks, indicating its potential as a robust foundation for learning general representations. Thus, this work lays the groundwork for developing future time series foundation models based on Joint Embedding.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multivariate Forecasting | ETTh1 | MSE1.16 | 909 | |
| Multivariate Time-series Forecasting | ETTm1 | MSE0.86 | 742 | |
| Multivariate Time-series Forecasting | ETTm2 | MSE1.41 | 593 | |
| Multivariate Time-series Forecasting | Weather | MSE0.44 | 466 | |
| Multivariate Time-series Forecasting | ETTh2 | MSE2.4 | 219 | |
| Anomaly Detection | KPI delay tolerance 7 | F1 Score33.1 | 20 | |
| Anomaly Detection | Yahoo Webscope S5 delay tolerance 3 | F1 Score4.4 | 20 | |
| Time-series classification | UCR 126 (test) | Accuracy75.5 | 11 | |
| Time-series classification | UEA 26 (test) | Accuracy67.8 | 11 | |
| Multivariate Time-series Forecasting | AirQ | MSE1.93 | 10 |