LLM2D
学习结构增强的时间点过程及其用戈rom瓦茨正则化
Learning Structure-enhanced Temporal Point Processes with Gromov-Wasserstein Regularization
作者: Qingmei Wang, Fanmeng Wang, Bing Su, Hongteng Xu
发布日期: 4/1/2025
arXiv ID: oai:arXiv.org:2503.23002v1

摘要

arXiv:2503.23002v1 类型:交叉 摘要:现实世界中的事件序列通常是由不同的时间点过程(TPPs)生成的,因此具有聚类结构。然而,在事件序列的建模和预测中,大多数现有的TPPs忽略了事件序列内在的聚类结构,导致模型在解释性方面不尽如人意。在本研究中,我们通过Gromov-Wasserstein(GW)正则化学习结构增强的TPPs,在极大似然估计框架中对TPPs在序列级嵌入上的聚类结构施加约束。在训练阶段,所提出的方法利用一个非参数TPP核来正则化基于序列嵌入的相似矩阵。在大规模应用中,我们抽样核矩阵并在Gromov-Wasserstein(GW)差异项中实现正则化,从而在稳定性和计算效率之间找到平衡。通过这种方法学习的TPPs生成了聚类的序列嵌入,并且在预测和聚类性能方面表现出竞争力,显著提高了模型的解释性,而不会牺牲预测准确性。