Skip to main content
To KTH's start page

Soon Hoe Lim: Elucidating the Design Choice of Probability Paths in Flow Matching for Forecasting

Time: Mon 2024-10-28 15.15 - 16.15

Location: 3721 (Lindstedtsvägen 25)

Participating: Soon Hoe Lim (KTH)

Export to calendar

Abstract

Flow matching has recently emerged as a powerful paradigm for generative modeling and has been extended to probabilistic time series forecasting in latent spaces. However, the impact of the specific choice of probability path model on forecasting performance remains under-explored. In this talk, we demonstrate that forecasting spatio-temporal data with flow matching is highly sensitive to the selection of the probability path model. Motivated by this insight, we propose a novel probability path model designed to improve forecasting performance. Our empirical results across various dynamical system benchmarks show that our model achieves faster convergence during training and improved predictive performance compared to existing probability path models. Importantly, our approach is efficient during inference, requiring only a few sampling steps. This makes our proposed model practical for real-world applications and opens new avenues for probabilistic forecasting. This talk is based on joint work with N. Benjamin Erichson, Michael Mahoney, and collaborators from Cornell, Emory, LBNL, ICSI, and UC Berkeley.