arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.05302cs.LGstat.ML

Mamba Hawkes过程

Mamba Hawkes Process

Shenzhen Research Institute of Big Data(深圳大数据研究院);The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳))

Anningzhe Gao, Shan Dai, Yan Hu

AI总结:

针对不规则异步事件序列中传统Hawkes过程难以建模非线性及神经网络长程依赖、效率不足的问题,提出基于Mamba的MHP及融合Transformer的MHP-E,并通过理论与实验验证其优越性能。

摘要:

不规则且异步的事件序列广泛存在于社交媒体、金融和医疗保健等许多领域。传统时间点过程(TPP),如Hawkes过程,往往难以有效建模相互抑制和非线性。尽管近期包括RNN和Transformer在内的神经网络模型解决了其中一些问题,但它们在长期依赖和计算效率方面仍面临挑战。本文提出Mamba Hawkes Process(MHP),利用Mamba状态空间架构捕捉长程依赖和动态事件交互。结果表明,MHP在多个数据集上优于现有模型。此外,我们提出Mamba Hawkes Process Extension(MHP-E),结合Mamba与Transformer模型以增强预测能力。本文展示了Mamba架构在Hawkes过程中的新颖应用、灵活且可扩展的模型结构,以及对状态空间模型与Hawkes过程协同作用的理论分析。实验结果证明MHP和MHP-E均具有优越性能,推动了时间点过程建模领域的发展。

英文摘要:

Irregular and asynchronous event sequences are prevalent in many domains, such as social media, finance, and healthcare. Traditional temporal point processes (TPPs), like Hawkes processes, often struggle to model mutual inhibition and nonlinearity effectively. While recent neural network models, including RNNs and Transformers, address some of these issues, they still face challenges with long-term dependencies and computational efficiency. In this paper, we introduce the Mamba Hawkes Process (MHP), which leverages the Mamba state space architecture to capture long-range dependencies and dynamic event interactions. Our results show that MHP outperforms existing models across various datasets. Additionally, we propose the Mamba Hawkes Process Extension (MHP-E), which combines Mamba and Transformer models to enhance predictive capabilities. We present the novel application of the Mamba architecture to Hawkes processes, a flexible and extensible model structure, and a theoretical analysis of the synergy between state space models and Hawkes processes. Experimental results demonstrate the superior performance of both MHP and MHP-E, advancing the field of temporal point process modeling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑