Enhancing Event Sequence Modeling with Contrastive Relational Inference

Yan Wang,Zhixuan Chu,Tao Zhou,Caigao Jiang,Hongyan Hao,Minjie Zhu,Xindong Cai,Qing Cui,Longfei Li,James Y Zhang,Siqiao Xue,Jun Zhou
DOI: https://doi.org/10.1109/icassp48485.2024.10446896
2023-01-01
Abstract: Neural temporal point processes(TPPs) have shown promise for modeling continuous-time event sequences. However, capturing the interactions between events is challenging yet critical for performing inference tasks like forecasting on event sequence data. Existing TPP models have focused on parameterizing the conditional distribution of future events but struggle to model event interactions. In this paper, we propose a novel approach that leverages Neural Relational Inference (NRI) to learn a relation graph that infers interactions while simultaneously learning the dynamics patterns from observational data. Our approach, the Contrastive Relational Inference-based Hawkes Process (CRIHP), reasons about event interactions under a variational inference framework. It utilizes intensity-based learning to search for prototype paths to contrast relationship constraints. Extensive experiments on three real-world datasets demonstrate the effectiveness of our model in capturing event interactions for event sequence modeling tasks.
What problem does this paper attempt to address?