arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.04975cs.LGcs.AI

UniTST:有效建模多元时间序列预测中的序列间与序列内依赖关系

UniTST: Effectively Modeling Inter-Series and Intra-Series Dependencies for Multivariate Time Series Forecasting

Salesforce(赛富时);University of California Los Angeles(加利福尼亚大学洛杉矶分校);National University of Singapore(新加坡国立大学)

Juncheng Liu, Chenghao Liu, Gerald Woo, Yiwei Wang, Bryan Hooi, Caiming Xiong, Doyen Sahoo

AI总结:

本文提出UniTST模型,通过在展平的patch token上使用统一注意力机制直接建模多元时间序列中的序列间与序列内依赖,并引入dispatcher模块降低复杂度,在多个数据集上取得出色预测性能。

摘要:

基于Transformer的模型已成为多元时间序列预测(MTSF)的强大工具。然而,现有Transformer模型往往难以同时捕捉MTS数据中跨变量维度和时间维度的复杂依赖关系。最近一些模型被提出通过两个顺序或并行的注意力机制分别捕捉变量依赖和时间依赖。然而,这些方法无法直接且明确地学习复杂的序列间和序列内依赖关系。在本工作中,我们首先证明这些依赖关系非常重要,因为它们通常存在于现实世界数据中。为了直接建模这些依赖关系,我们提出了一种基于Transformer的模型UniTST,它在展平的patch token上包含统一的注意力机制。此外,我们添加了一个dispatcher模块,该模块降低了复杂度,并使模型在变量数量可能很大的情况下仍然可行。尽管我们提出的模型采用了简单的架构,但我们在多个时间序列预测数据集上进行的大量实验表明,它具有出色的性能。

英文摘要:

Transformer-based models have emerged as powerful tools for multivariate time series forecasting (MTSF). However, existing Transformer models often fall short of capturing both intricate dependencies across variate and temporal dimensions in MTS data. Some recent models are proposed to separately capture variate and temporal dependencies through either two sequential or parallel attention mechanisms. However, these methods cannot directly and explicitly learn the intricate inter-series and intra-series dependencies. In this work, we first demonstrate that these dependencies are very important as they usually exist in real-world data. To directly model these dependencies, we propose a transformer-based model UniTST containing a unified attention mechanism on the flattened patch tokens. Additionally, we add a dispatcher module which reduces the complexity and makes the model feasible for a potentially large number of variates. Although our proposed model employs a simple architecture, it offers compelling performance as shown in our extensive experiments on several datasets for time series forecasting.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑