Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HYPRO: A Hybridly Normalized Probabilistic Model for Long-Horizon Prediction of Event Sequences

About

In this paper, we tackle the important yet under-investigated problem of making long-horizon prediction of event sequences. Existing state-of-the-art models do not perform well at this task due to their autoregressive structure. We propose HYPRO, a hybridly normalized probabilistic model that naturally fits this task: its first part is an autoregressive base model that learns to propose predictions; its second part is an energy function that learns to reweight the proposals such that more realistic predictions end up with higher probabilities. We also propose efficient training and inference algorithms for this model. Experiments on multiple real-world datasets demonstrate that our proposed HYPRO model can significantly outperform previous models at making long-horizon predictions of future events. We also conduct a range of ablation studies to investigate the effectiveness of each component of our proposed methods.

Siqiao Xue, Xiaoming Shi, James Y Zhang, Hongyuan Mei• 2022

Related benchmarks

TaskDatasetResultRank
Event PredictionStackOverflow--
58
Event Predictiontaxi
RMSEΔt0.322
40
Long-horizon predictionAMAZON
RMSE (Δt)0.433
26
Event PredictionTaobao (test)
OTD21.547
22
Event PredictionStackOverflow (test)
OTD21.062
22
Event PredictionTaxi (test)
OTD11.875
22
Event PredictionAmazon (test)
OTD24.956
22
Event PredictionRETWEET (test)
OTD31.743
22
Event PredictionTaobao
RMSEΔt0.573
21
Showing 9 of 9 rows

Other info

Code

Follow for update