arXiv:2411.01842cs.LGstat.ML2024-11NeurIPS被引 16

一种可应对多种预测时长的稳定时间序列预测模型

ElasTST: Towards Robust Varied-Horizon Forecasting with Elastic Time-Series Transformer

  • 采用非自回归架构与结构化注意力掩码,实现多时长远期预测
  • 在多个数据集上优于现有模型,最长预测时长达96步仍保持高精度
  • 适合工业界需要灵活调整预测长度的实时场景

众多工业领域需要能够在不同预测时长下提供稳健预测的模型。尽管近年来在时间序列预测架构设计和通用预训练模型开发方面取得进展,但对模型在推理阶段适应多样化预测时长能力的系统性评估仍显不足。本文提出弹性时间序列变压器(ElasTST),通过引入非自回归设计、占位符与结构化自注意力掩码,使未来输出对推理时长调整保持不变。模型集成可调旋转位置编码以捕捉时间序列特定周期特征,增强对不同预测时长的适应性。此外,采用多尺度补丁设计,有效融合细粒度与粗粒度信息。训练阶段使用时长重加权策略,在单一固定时长设置下近似实现跨多时长随机采样效果。通过与先进时间序列架构及当代基础模型的广泛实验对比,验证了ElasTST独特设计的有效性。结果表明,ElasTST是解决实际中多样化预测时长需求的稳健方案。

原文摘要 · Abstract (English)

Numerous industrial sectors necessitate models capable of providing robust forecasts across various horizons. Despite the recent strides in crafting specific architectures for time-series forecasting and developing pre-trained universal models, a comprehensive examination of their capability in accommodating varied-horizon forecasting during inference is still lacking. This paper bridges this gap through the design and evaluation of the Elastic Time-Series Transformer (ElasTST). The ElasTST model incorporates a non-autoregressive design with placeholders and structured self-attention masks, warranting future outputs that are invariant to adjustments in inference horizons. A tunable version of rotary position embedding is also integrated into ElasTST to capture time-series-specific periods and enhance adaptability to different horizons. Additionally, ElasTST employs a multi-scale patch design, effectively integrating both fine-grained and coarse-grained information. During the training phase, ElasTST uses a horizon reweighting strategy that approximates the effect of random sampling across multiple horizons with a single fixed horizon setting. Through comprehensive experiments and comparisons with state-of-the-art time-series architectures and contemporary foundation models, we demonstrate the efficacy of ElasTST's unique design elements. Our findings position ElasTST as a robust solution for the practical necessity of varied-horizon forecasting.

时间序列多时长预测变压器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。