论文针对多元概率时间序列预测(multivariate probabilistic time series forecasting)任务,提出了TimeGrad模型。
有开源的代码:PytorchTS
概率预测如下图所示,对未来的预测带有概率:

TimeGrad模型基于Diffusion Probabilistic Model,Diffusion Probabilistic Model这里不再介绍,可以简单认为是一个可以拟合复杂分布的概率模型。需要学习的请参见博客《Denoising Diffusion Probabilistic Models简介》。
在了解Diffusion Probabilistic Model的基础上,这篇文章的方法非常简单。如果把TimeGrad基于的Diffusion Probabilistic Model换成高斯分布,TimeGrad就类似DeepAR了。
方法
将多变量时间序列表示为 x i , t 0 ∈ R , i = { i , . . . , D } x_{i,t}^0\in \mathbb{R}, i=\{i,...,D\} xi,t0∈R,i={ i,...,D},其中 t t t是时间index, D D D是多变量的变量数。给一个连续时间序列 t ∈ [ 1 , T ] t\in [1, T] t∈[1,T],将其划分为context window [ 1 , t 0 ) [1,t_0) [1,t0)和prediction window [ t 0 , T ] [t_0,T] [t0,T]。任务目的是用context window的时间序列预测prediction window的时间序列。
TimeGrad的是一个自回归(autoregressive)的模型:
q X ( x t 0 : T 0 ∣ x 1 : t 0 − 1 0 , c 1 : T ) = Π t = t 0 T q X ( x t 0 ∣ x 1 : t − 1 0 , c 1 : T ) (1) q_{\mathcal X}(\mathbf x_{t_0:T}^0 | \mathbf x_{1:t_0-1}^0, \mathbf c_{1:T})=\Pi_{t=t_0}^Tq_{\mathcal X}(\mathbf x_{t}^0 | \mathbf x_{1:t-1}^0, \mathbf c_{1:T}) \tag{1} qX(xt0:T0∣x1:t0−10

本文介绍了TimeGrad模型,一种基于DiffusionProbabilisticModel的多变量时间序列预测方法,通过RNN捕捉时间依赖性并结合概率建模,提供未来预测的不确定性。方法利用连续分级概率评分(CRPS)评估,展示了模型结构与训练过程。

4万+

被折叠的 条评论
为什么被折叠?



