用随机学习框架让Transformer直接生成多变量时间序列的可信概率预测。
Deep Generative Transformers for Probabilistic Time Series and Spatiotemporal Forecasting
- 通过注入噪声并优化严格合理评分,直接学习条件分布。
- 在多个时间序列和流行病数据集上均实现校准的概率预测且优于现有方法。
- 仅增加常数级计算开销,即可获得高精度概率建模能力,适合需要可靠置信度的场景。
可靠不确定性量化对于多变量时间序列和时空数据预测至关重要。尽管Transformer架构在序列建模方面表现出色,但当前的概率方法通常依赖于受限的参数化似然或分位数目标,限制了其对相关时间序列复杂联合分布的捕捉能力。为克服这些局限,我们提出 extit{Enformer} 及其时空扩展 extit{GEnformer}。这些模型将Transformer的强大表达力与反向回归(engression)这一随机学习范式结合,通过注入随机噪声并优化严格合理的评分目标,直接学习条件预测分布,无需施加参数假设。该设计确保生成一致的多变量轨迹,同时保持Transformer在长程依赖和跨序列交互建模上的高效性。Enformer 的概率能力仅带来与同配置确定性Transformer相当的渐近额外开销。我们在多个主流的时间动态多变量基准以及真实世界流行病数据集上进行了广泛评估,结果表明两种框架均能生成校准的概率预测,并持续优于现有最先进基线。
原文摘要 · Abstract (English)
Reliable uncertainty quantification is paramount for forecasting multivariate time series and spatiotemporal data. While Transformer architectures excel at sequence modeling, current probabilistic approaches typically rely on restrictive parametric likelihoods or quantile-based objectives, thereby limiting their ability to capture complex joint distributions in correlated time series. To overcome these limitations, we propose \textit{Enformer} and its spatiotemporal extension, \textit{GEnformer}. These models synthesize the expressive power of Transformers with engression, a stochastic learning paradigm for modeling conditional distributions. By injecting stochastic noise and optimizing a strictly proper scoring objective, our frameworks directly learn conditional predictive distributions without imposing parametric assumptions. This design ensures the generation of coherent multivariate trajectories while maintaining the Transformer's efficacy in modeling long-range dependencies and cross-series interactions. The probabilistic capability of Enformer is achieved with an asymptotic overhead of only a constant factor over a deterministic Transformer with an identical configuration. We extensively evaluate our frameworks on prominent multivariate benchmarks for temporal dynamics and real-world epidemic datasets for spatiotemporal dynamics. Empirical results demonstrate that both frameworks yield calibrated probabilistic forecasts and consistently outperform state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。