用小波域+旋转路由注意力,高效预测多变量时间序列
WaveRoRA: Wavelet Rotary Route Attention for Multivariate Time Series Forecasting
- 在小波域融合时频信息,捕捉多尺度趋势与周期性
- 引入路由注意力机制,计算复杂度从二次降至线性
- 在8个真实数据集上超越主流模型,适合长序列预测场景
近年来,基于Transformer的模型在多变量时间序列预测(MTSF)中取得显著进展。然而,以往工作仅关注时域或频域特征提取,难以充分捕捉趋势与周期特性。为此,我们提出一种小波学习框架,小波域可同时整合时频信息,实现信号在不同尺度下的局部特征分析。此外,Transformer中的Softmax自注意力机制具有二次复杂度,导致长序列建模时计算成本过高。因此,我们提出新型注意力机制——旋转路由注意力(RoRA):利用旋转位置编码注入相对位置信息,并引入少量路由令牌 $r$,从 $KV$ 矩阵聚合信息后重分配至 $Q$ 矩阵,实现线性复杂度。我们进一步提出WaveRoRA,通过RoRA在小波域捕捉序列间依赖关系。在8个真实世界数据集上的大量实验表明,WaveRoRA在性能上优于现有最先进模型,且计算开销更低。代码已开源:https://github.com/Leopold2333/WaveRoRA。
原文摘要 · Abstract (English)
In recent years, Transformer-based models (Transformers) have achieved significant success in multivariate time series forecasting (MTSF). However, previous works focus on extracting features either from the time domain or the frequency domain, which inadequately captures the trends and periodic characteristics. To address this issue, we propose a wavelet learning framework to model complex temporal dependencies of the time series data. The wavelet domain integrates both time and frequency information, allowing for the analysis of local characteristics of signals at different scales. Additionally, the Softmax self-attention mechanism used by Transformers has quadratic complexity, which leads to excessive computational costs when capturing long-term dependencies. Therefore, we propose a novel attention mechanism: Rotary Route Attention (RoRA). Unlike Softmax attention, RoRA utilizes rotary position embeddings to inject relative positional information to sequence tokens and introduces a small number of routing tokens $r$ to aggregate information from the $KV$ matrices and redistribute it to the $Q$ matrix, offering linear complexity. We further propose WaveRoRA, which leverages RoRA to capture inter-series dependencies in the wavelet domain. We conduct extensive experiments on eight real-world datasets. The results indicate that WaveRoRA outperforms existing state-of-the-art models while maintaining lower computational costs. Our code is available at https://github.com/Leopold2333/WaveRoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。