Deep Continuous-Time State-Space Models for Marked Event Sequences

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chang, Yuxin, Boyd, Alex, Xiao, Cao, Kass-Hout, Taha, Bhatia, Parminder, Smyth, Padhraic, Warrington, Andrew
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915570436276224
author Chang, Yuxin
Boyd, Alex
Xiao, Cao
Kass-Hout, Taha
Bhatia, Parminder
Smyth, Padhraic
Warrington, Andrew
author_facet Chang, Yuxin
Boyd, Alex
Xiao, Cao
Kass-Hout, Taha
Bhatia, Parminder
Smyth, Padhraic
Warrington, Andrew
contents Marked temporal point processes (MTPPs) model sequences of events occurring at irregular time intervals, with wide-ranging applications in fields such as healthcare, finance and social networks. We propose the state-space point process (S2P2) model, a novel and performant model that leverages techniques derived for modern deep state-space models (SSMs) to overcome limitations of existing MTPP models, while simultaneously imbuing strong inductive biases for continuous-time event sequences that other discrete sequence models (i.e., RNNs, transformers) do not capture. Inspired by the classical linear Hawkes processes, we propose an architecture that interleaves stochastic jump differential equations with nonlinearities to create a highly expressive intensity-based MTPP model, without the need for restrictive parametric assumptions for the intensity. Our approach enables efficient training and inference with a parallel scan, bringing linear complexity and sublinear scaling while retaining expressivity to MTPPs. Empirically, S2P2 achieves state-of-the-art predictive likelihoods across eight real-world datasets, delivering an average improvement of 33% over the best existing approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2412_19634
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deep Continuous-Time State-Space Models for Marked Event Sequences
Chang, Yuxin
Boyd, Alex
Xiao, Cao
Kass-Hout, Taha
Bhatia, Parminder
Smyth, Padhraic
Warrington, Andrew
Machine Learning
Marked temporal point processes (MTPPs) model sequences of events occurring at irregular time intervals, with wide-ranging applications in fields such as healthcare, finance and social networks. We propose the state-space point process (S2P2) model, a novel and performant model that leverages techniques derived for modern deep state-space models (SSMs) to overcome limitations of existing MTPP models, while simultaneously imbuing strong inductive biases for continuous-time event sequences that other discrete sequence models (i.e., RNNs, transformers) do not capture. Inspired by the classical linear Hawkes processes, we propose an architecture that interleaves stochastic jump differential equations with nonlinearities to create a highly expressive intensity-based MTPP model, without the need for restrictive parametric assumptions for the intensity. Our approach enables efficient training and inference with a parallel scan, bringing linear complexity and sublinear scaling while retaining expressivity to MTPPs. Empirically, S2P2 achieves state-of-the-art predictive likelihoods across eight real-world datasets, delivering an average improvement of 33% over the best existing approaches.
title Deep Continuous-Time State-Space Models for Marked Event Sequences
topic Machine Learning
url https://arxiv.org/abs/2412.19634