Learning Structure-enhanced Temporal Point Processes with Gromov-Wasserstein Regularization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Qingmei, Wang, Fanmeng, Su, Bing, Xu, Hongteng
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915218619105280
author Wang, Qingmei
Wang, Fanmeng
Su, Bing
Xu, Hongteng
author_facet Wang, Qingmei
Wang, Fanmeng
Su, Bing
Xu, Hongteng
contents Real-world event sequences are often generated by different temporal point processes (TPPs) and thus have clustering structures. Nonetheless, in the modeling and prediction of event sequences, most existing TPPs ignore the inherent clustering structures of the event sequences, leading to the models with unsatisfactory interpretability. In this study, we learn structure-enhanced TPPs with the help of Gromov-Wasserstein (GW) regularization, which imposes clustering structures on the sequence-level embeddings of the TPPs in the maximum likelihood estimation framework.In the training phase, the proposed method leverages a nonparametric TPP kernel to regularize the similarity matrix derived based on the sequence embeddings. In large-scale applications, we sample the kernel matrix and implement the regularization as a Gromov-Wasserstein (GW) discrepancy term, which achieves a trade-off between regularity and computational efficiency.The TPPs learned through this method result in clustered sequence embeddings and demonstrate competitive predictive and clustering performance, significantly improving the model interpretability without compromising prediction accuracy.
format Preprint
id arxiv_https___arxiv_org_abs_2503_23002
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Structure-enhanced Temporal Point Processes with Gromov-Wasserstein Regularization
Wang, Qingmei
Wang, Fanmeng
Su, Bing
Xu, Hongteng
Machine Learning
Artificial Intelligence
60G55, 62M10
Real-world event sequences are often generated by different temporal point processes (TPPs) and thus have clustering structures. Nonetheless, in the modeling and prediction of event sequences, most existing TPPs ignore the inherent clustering structures of the event sequences, leading to the models with unsatisfactory interpretability. In this study, we learn structure-enhanced TPPs with the help of Gromov-Wasserstein (GW) regularization, which imposes clustering structures on the sequence-level embeddings of the TPPs in the maximum likelihood estimation framework.In the training phase, the proposed method leverages a nonparametric TPP kernel to regularize the similarity matrix derived based on the sequence embeddings. In large-scale applications, we sample the kernel matrix and implement the regularization as a Gromov-Wasserstein (GW) discrepancy term, which achieves a trade-off between regularity and computational efficiency.The TPPs learned through this method result in clustered sequence embeddings and demonstrate competitive predictive and clustering performance, significantly improving the model interpretability without compromising prediction accuracy.
title Learning Structure-enhanced Temporal Point Processes with Gromov-Wasserstein Regularization
topic Machine Learning
Artificial Intelligence
60G55, 62M10
url https://arxiv.org/abs/2503.23002