Massively Parallel Exact Inference for Hawkes Processes

View PDF

Abstract:Multivariate Hawkes processes are a widely used class of self-exciting point processes, but maximum likelihood estimation naively scales as $O(N^2)$ in the number of events. The canonical linear exponential Hawkes process admits a faster $O(N)$ recurrence, but prior work evaluates this recurrence sequentially, without exploiting parallelization on modern GPUs. We show that the Hawkes process intensity can be expressed as a product of sparse transition matrices admitting a linear-time associative multiply, enabling computation via a parallel prefix scan. This yields a simple yet massively parallelizable algorithm for maximum likelihood estimation of linear exponential Hawkes processes. Our method reduces the computational complexity to approximately $O(N/P)$ with $P$ parallel processors, and naturally yields a batching scheme to maintain constant memory usage, avoiding GPU memory constraints. Importantly, it computes the exact likelihood without any additional assumptions or approximations, preserving the simplicity and interpretability of the model. We demonstrate orders-of-magnitude speedups on simulated and real datasets, scaling to thousands of nodes and tens of millions of events, substantially beyond scales reported in prior work. We provide an open-source PyTorch library implementing our optimizations.

Subjects:

Machine Learning (cs.LG)

Cite as: arXiv:2604.01342 [cs.LG]

(or arXiv:2604.01342v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2604.01342

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Ahmer Raza [view email] [v1] Wed, 1 Apr 2026 19:52:25 UTC (583 KB)