论文

ProtoFlow:原型引导的流匹配多变量时间序列预测

ProtoFlow: Prototype-Guided Flow Matching for Multivariate Time Series Forecasting

精选理由

ProtoFlow用学习到的原型先验替代通用噪声初始化,加速训练收敛,在时间序列预测任务上表现优异。

ProtoFlow结合向量量化自编码与原型先验流匹配,将多变量序列映射到离散潜在空间。该方法在基准数据集上实现高效推理和优越预测性能,避免了自回归令牌预测的展开不匹配问题。ProtoFlow使用DiT-based rectified flow从历史观测条件下的先验传输到未来潜在表示。

原文 · arXiv cs.AI

ProtoFlow: Prototype-Guided Flow Matching for Multivariate Time Series Forecasting

Generative modeling has shown strong promise for multivariate time mseries (MTS) forecasting, especially scale to high-dimensional settings. Diffusion-based methods achieve competitive performance but typically require many sampling steps at inference. VAE-based non-iterative forecasting frameworks have therefore emerged as an efficient alternative. Within this line of work, vector quantization (VQ) enables controllable latent space modeling by mapping multivariate series into compact discrete representations. Existing VQ-based forecasting methods, however, typically rely on autoregressive (AR) token generation, which suffers from exposure bias and training-inference mismatch. Flow matching provides an efficient non-autoregressive alternative for latent forecasting, but existing formulations usually initialize transport from a generic Gaussian prior. We instead observe that the trained VQ codebook already captures representative latent prototypes and can thus serve as a more informative prior for flow matching. Based on this insight, we propose ProtoFlow, a forecasting framework that combines vector-quantized autoencoding with Prototype-prior Flow matching. Our method first maps multivariate sequences into a discrete latent space, then constructs a structured prior from the learned codebook, and finally learns a DiT-based rectified flow to transport samples from this prior to future latent representations conditioned on historical observations. By replacing generic noise initialization with a learned prototype prior, ProtoFlow avoids the rollout mismatch of AR token prediction and promotes faster training convergence. Extensive experiments on benchmark datasets show that it consistently achieves superior forecasting performance with efficient inference.