Latent Diffusion Transformer for Probabilistic Time Series Forecasting

Authors

  • Shibo Feng School of Computer Science and Engineering, Nanyang Technological University (NTU), Singapore; Joint NTU-UBC Research Centre of Excellence in Active Living for the Elderly (LILY), NTU, Singapore; Webank-NTU Joint Research Institute on Fintech, NTU, Singapore
  • Chunyan Miao School of Computer Science and Engineering, Nanyang Technological University (NTU), Singapore; Joint NTU-UBC Research Centre of Excellence in Active Living for the Elderly (LILY), NTU, Singapore; Webank-NTU Joint Research Institute on Fintech, NTU, Singapore
  • Zhong Zhang Tencent AI Lab, Shenzhen, China
  • Peilin Zhao Tencent AI Lab, Shenzhen, China

DOI:

https://doi.org/10.1609/aaai.v38i11.29085

Keywords:

ML: Time-Series/Data Streams, KRR: Applications

Abstract

The probability prediction of multivariate time series is a notoriously challenging but practical task. This research proposes to condense high-dimensional multivariate time series forecasting into a problem of latent space time series generation, to improve the expressiveness of each timestamp and make forecasting more manageable. To solve the problem that the existing work is hard to extend to high-dimensional multivariate time series, we present a latent multivariate time series diffusion framework called Latent Diffusion Transformer (LDT), which consists of a symmetric statistics-aware autoencoder and a diffusion-based conditional generator, to implement this idea. Through careful design, the time series autoencoder can compress multivariate timestamp patterns into a concise latent representation by considering dynamic statistics. Then, the diffusion-based conditional generator is able to efficiently generate realistic multivariate timestamp values on a continuous latent space under a novel self-conditioning guidance which is modeled in a non-autoregressive way. Extensive experiments demonstrate that our model achieves state-of-the-art performance on many popular high-dimensional multivariate time series datasets.

Downloads

Published

2024-03-24

How to Cite

Feng, S., Miao, C., Zhang, Z., & Zhao, P. (2024). Latent Diffusion Transformer for Probabilistic Time Series Forecasting. Proceedings of the AAAI Conference on Artificial Intelligence, 38(11), 11979-11987. https://doi.org/10.1609/aaai.v38i11.29085

Issue

Section

AAAI Technical Track on Machine Learning II