DiTS用双流Transformer提升多变量时间序列预测能力
DiTS: Multimodal Diffusion Transformers Are Time Series Forecasters
- 将内生与外生变量视为不同模态,设计双流Transformer捕捉跨变量依赖
- 在多个基准上达到最优性能,即使无未来外生变量也表现优异
- 适合需要生成式预测的金融、气象等领域研究者
尽管生成式时间序列建模可实现更灵活的概率预测,但现有模型未能充分处理多维时间序列特性。主流扩散Transformer(DiT)依赖简单条件控制和单流Transformer主干,难以利用协变量感知预测中的跨变量依赖。受多模态扩散Transformer启发,我们提出时间序列扩散Transformer(DiTS),将内生与外生变量视为不同模态。为更好捕捉变量间与变量内依赖,设计了专用于时间序列的双流Transformer块,包含用于自回归建模的时间注意力模块和用于跨变量建模的变量注意力模块。不同于图像中将二维标记网格展平为一维序列的做法,本设计利用多变量依赖的低秩特性,降低计算开销。实验表明,无论是否存在未来外生变量观测,DiTS在多个基准上均达领先性能,展现出超越传统确定性深度预测模型的独特生成预测优势。
原文摘要 · Abstract (English)
While generative modeling on time series facilitates more capable and flexible probabilistic forecasting, existing generative time series models do not address the multi-dimensional properties of time series data well. The prevalent architecture of Diffusion Transformers (DiT), which relies on simplistic conditioning controls and a single-stream Transformer backbone, tends to underutilize cross-variate dependencies in covariate-aware forecasting. Inspired by Multimodal Diffusion Transformers that integrate textual guidance into video generation, we propose Diffusion Transformers for Time Series (DiTS), a general-purpose architecture that frames endogenous and exogenous variates as distinct modalities. To better capture both inter-variate and intra-variate dependencies, we design a dual-stream Transformer block tailored for time-series data, comprising a Time Attention module for autoregressive modeling along the temporal dimension and a Variate Attention module for cross-variate modeling. Unlike the common approach for images, which flattens 2D token grids into 1D sequences, our design leverages the low-rank property inherent in multivariate dependencies, thereby reducing computational costs. Experiments show that DiTS achieves state-of-the-art performance across benchmarks, regardless of the presence of future exogenous variate observations, demonstrating unique generative forecasting strengths over traditional deterministic deep forecasting models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。