InjectTST: A Transformer Method of Injecting Global Information into Independent Channels for Long Time Series Forecasting

Ce Chi, Xing Wang, Kexin Yang, Zhiyan Song, Di Jin +3 more
3/5/2024
cs.LGcs.AI

No confident code implementation found for this paper.

Abstract

Transformer has become one of the most popular architectures for multivariate time series (MTS) forecasting. Recent Transformer-based MTS models generally prefer channel-independent structures with the observation that channel independence can alleviate noise and distribution drift issues, leading to more robustness. Nevertheless, it is essential to note that channel dependency remains an inherent characteristic of MTS, carrying valuable information. Designing a model that incorporates merits of both channel-independent and channel-mixing structures is a key to further improvement of MTS forecasting, which poses a challenging conundrum. To address the problem, an injection method for global information into channel-independent Transformer, InjectTST, is proposed in this paper. Instead of designing a channel-mixing model directly, we retain the channel-independent backbone and gradually inject global information into individual channels in a selective way. A channel identifier, a global mixing module and a self-contextual attention module are devised in InjectTST. The channel identifier can help Transformer distinguish channels for better representation. The global mixing module produces cross-channel global information. Through the self-contextual attention module, the independent channels can selectively concentrate on useful global information without robustness degradation, and channel mixing is achieved implicitly. Experiments indicate that InjectTST can achieve stable improvement compared with state-of-the-art models.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{chi2024injecttst,
  title  = {InjectTST: A Transformer Method of Injecting Global Information into Independent Channels for Long Time Series Forecasting},
  author = {Ce Chi and Xing Wang and Kexin Yang and Zhiyan Song and Di Jin and Lin Zhu and Chao Deng and Junlan Feng},
  year   = {2024},
  eprint = {2403.02814},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.02814v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.02814.svg)](https://codeofpaper.com/paper/2403.02814)

Discussion