Seahorse: A Unified Benchmarking Framework for Spatiotemporal Event Modeling

원문 초록

Spatiotemporal point processes (STPPs) model event data in continuous time and space, with applications in mobility, epidemiology, and public safety. Recent neural STPPs span expressive intensities, density models, continuous-time latent dynamics, normalizing-flow spatial decoders, and score-based generative mechanisms. Yet comparison remains fragile because implementations differ in preprocessing, coordinate normalization, splits, likelihood conventions, and evaluation protocols. We present Seahorse, a unified framework for reproducible training and evaluation of STPPs. Seahorse provides an encode-evolve-decode interface and an executable benchmarking contract that standardizes dataset handling, tuning, raw-coordinate likelihood reporting, and run artifacts across classical baselines, neural likelihood models, continuous-time Neural STPP variants, and sample-based generative models. On 13 real datasets and 19 presets, no model dominates: likelihood, calibration, and next-event prediction favor different models, and the strongest models take from minutes to hours to train. Using HawkesNest as a controlled synthetic stress-test suite, we show how diagnostic benchmarking can probe neural STPP behavior under increasing spatiotemporal entanglement. Seahorse provides reusable infrastructure for comparing, diagnosing, and extending STPP models beyond implementation-specific leaderboards. The implementation, including preset configurations and reproduction scripts, is available at https://github.com/YahyaAalaila/seahorse.