A Variational Autoencoder for Neural Temporal Point Processes with Dynamic Latent Graphs
Sikun Yang, Hongyuan Zha
TL;DR
This work addresses non-stationary dynamics in asynchronous multivariate event sequences by introducing VAETPP, a variational autoencoder for neural temporal point processes with dynamic latent graphs. The time axis is divided into $K$ sub-intervals, with each interval governed by a latent graph ${z}_{(v,u)}^{k}$ capturing dependencies among event types; a GNN-based encoder builds posterior latents, while a forward RNN defines the prior and a GRNN-based decoder propagates influences to predict inter-event times via a log-normal mixture $p( au|m{ heta})$. Training optimizes the ELBO, $ ext{ELBO}(oldsymbol{ heta},oldsymbol{ ho}) = ext{E}_{q_oldsymbol{ ho}(z|S)}[ extstyle extstyle ext{log }p(S|z, heta)] - ext{KL}ig(q_oldsymbol{ ho}(z|S)ig\ ext{ }|| ext{ }p_oldsymbol{ heta}(z|z^{1:k-1},S^{1:k})ig)$, enabling joint learning of the dynamic graph and event-time distributions. Experiments on NYMVC and Stack Exchange datasets show VAETPP achieves superior negative log-likelihood and RMSE compared to Exponential, RMTPP, FullyNN, LogNormMix, and THP, with interpretable time-varying graphs illustrating changing inter-location influences. The method advances dynamic-graph neural TPPs and offers practical gains for forecasting event times and types in non-stationary settings, while future work may explore automatic interval boundary discovery.
Abstract
Continuously-observed event occurrences, often exhibit self- and mutually-exciting effects, which can be well modeled using temporal point processes. Beyond that, these event dynamics may also change over time, with certain periodic trends. We propose a novel variational auto-encoder to capture such a mixture of temporal dynamics. More specifically, the whole time interval of the input sequence is partitioned into a set of sub-intervals. The event dynamics are assumed to be stationary within each sub-interval, but could be changing across those sub-intervals. In particular, we use a sequential latent variable model to learn a dependency graph between the observed dimensions, for each sub-interval. The model predicts the future event times, by using the learned dependency graph to remove the noncontributing influences of past events. By doing so, the proposed model demonstrates its higher accuracy in predicting inter-event times and event types for several real-world event sequences, compared with existing state of the art neural point processes.
