CoST: Contrastive Learning of Disentangled Seasonal-Trend Representations for Time Series Forecasting

By Gerald Woo, Chenghao Liu, Doyen Sahoo, Akshat Kumar, Steven C. H. Hoi. International Conference on Learning Representations, 2022

483 citations recorded by the source corpus.

Deep learning has been actively studied for time series forecasting, and the mainstream paradigm is based on the end-to-end training of neural network architectures, ranging from classical LSTM/RNNs to more recent TCNs and Transformers. Motivated by the recent success of representation learning in computer vision and natural language processing, we argue that a more promising paradigm for time series forecasting, is to first learn disentangled feature representations, followed by a simple regression fine-tuning step -- we justify such a paradigm from a causal perspective. Following this principle, we propose a new time series representation learning framework for time series forecasting named CoST, which applies contrastive learning methods to learn disentangled seasonal-trend representations. CoST comprises both time domain and frequency domain contrastive losses to learn discriminative trend and seasonal representations, respectively. Extensive experiments on real-world datasets show that CoST consistently outperforms the state-of-the-art methods by a considerable margin, achieving a 21.3% improvement in MSE on multivariate benchmarks. It is also robust to various choices of backbone encoders, as well as downstream regressors. Code is available at https://github.com/salesforce/CoST.

Subfields: Management Science and Operations Research, Signal Processing

View Paper: https://www.semanticscholar.org/paper/c090ff3dec01e06f46735b7b9ab133a5db8c73c3

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Management Science and Operations Research

  1. Gerald Woo
  2. Silvio Savarese
  3. Steven C. H. Hoi
  4. Chenghao Liu
  5. Marinka Zitnik
  6. Caiming Xiong
  7. Akshat Kumar
  8. Doyen Sahoo

In Signal Processing

  1. Doyen Sahoo
  2. Steven C. H. Hoi
  3. Gerald Woo
  4. Silvio Savarese
  5. Defu Lian
  6. Chenghao Liu
  7. Marinka Zitnik
  8. Caiming Xiong

Citations in the corpus