4 个月前

DeTPP:利用目标检测实现鲁棒的长时域事件预测

DeTPP:利用目标检测实现鲁棒的长时域事件预测

摘要

长时域事件预测在多个领域中至关重要,包括零售、金融、医疗保健和社会网络。传统方法,如标记时间点过程(Marked Temporal Point Processes, MTPP),通常依赖自回归模型来预测未来的多个事件。然而,这些模型经常遇到收敛到常数或重复输出的问题,这限制了它们的有效性和普遍适用性。为了解决这些问题,我们提出了一种基于检测的时间点过程(Detection-based Temporal Point Processes, DeTPP)的新方法,该方法受到计算机视觉中物体检测技术的启发。DeTPP采用了一种独特的匹配损失函数,有选择地优先考虑可可靠预测的事件,从而在推理过程中提高了预测的准确性和多样性。我们的方法在长时域事件预测方面建立了新的最先进水平,在现有的MTPP和下K个事件预测方法上实现了高达77%的相对改进。值得注意的是,DeTPP在大规模交易数据集上的下一个事件预测准确性也提高了多达2.7%。此外,DeTPP也是最快的推理方法之一。DeTPP的实现代码已公开发布在GitHub上。

代码仓库

ivan-chai/hotpp-benchmark
官方
pytorch
GitHub 中提及
ivan-chai/torch-linear-assignment
pytorch
GitHub 中提及

基准测试

基准方法指标
point-processes-on-agegroup-transactions-mtppDeTPP
OTD: 6.66
T-mAP: 9.17
point-processes-on-amazon-mtppDeTPP
OTD: 5.98
T-mAP: 37.18
point-processes-on-retweet-mtppDeTPP
OTD: 134.4
T-mAP: 57.37
point-processes-on-stackoverflow-mtppDeTPP
OTD: 12.06
T-mAP: 23.11

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DeTPP:利用目标检测实现鲁棒的长时域事件预测 | 论文 | HyperAI超神经