Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

EventFlow: Forecasting Temporal Point Processes with Flow Matching

About

Continuous-time event sequences, in which events occur at irregular intervals, are ubiquitous across a wide range of industrial and scientific domains. The contemporary modeling paradigm is to treat such data as realizations of a temporal point process, and in machine learning it is common to model temporal point processes in an autoregressive fashion using a neural network. While autoregressive models are successful in predicting the time of a single subsequent event, their performance can degrade when forecasting longer horizons due to cascading errors and myopic predictions. We propose EventFlow, a non-autoregressive generative model for temporal point processes. The model builds on the flow matching framework in order to directly learn joint distributions over event times, side-stepping the autoregressive process. EventFlow is simple to implement and achieves a 20%-53% lower forecast error than the nearest baseline on standard TPP benchmarks while simultaneously using fewer model calls at sampling time.

Gavin Kerrigan, Kai Nelson, Padhraic Smyth• 2024

Related benchmarks

TaskDatasetResultRank
Event count predictionPUBG (test)
MARE0.4
11
Event count predictionReddit-C (test)
MARE0.7
11
Event count predictionReddit-S (test)
MARE0.16
11
Event count predictionTaxi (test)
MARE28
11
Event count predictionTwitter (test)
MARE0.46
11
Probabilistic ForecastingTaxi (test)
MSE0.16
11
Unconditional generation of event sequencesPUBG (test)
MMD1.5
11
Unconditional generation of event sequencesReddit-C (Red.C) (test)
MMD0.7
11
Unconditional generation of event sequencesReddit-S (test)
MMD0.7
11
Unconditional generation of event sequencesTaxi (test)
MMD (1e-2)3.5
11
Showing 10 of 47 rows

Other info

Follow for update