Learning sequential features for cascade outbreak prediction

摘要

Information cascades are ubiquitous in various online social networks. Outbreak of cascades could cause huge and unexpected effects. Therefore, predicting the outbreak of cascades at early stage is of vital importance to avoid potential bad effects and take relevant actions. Existing methods either adopt regression or classification technique with exhaustive feature engineering or predict cascade dynamics via modeling the stochastic process of cascades using a hard-coded diffusion–reaction function. One salient issue of these methods is that these methods heavily depend on human-defined knowledge, features or functions. In this paper, we propose to use recurrent neural network with long short-term memory to directly learn sequential patterns from information cascades, working in a fully data-driven manner. With the learned sequential patterns, the outbreak of cascade could be accurately predicted. Extensive experiments on both Twitter and Sina Weibo datasets demonstrate that our method significantly outperforms state-of-the-art methods at the prediction of cascade outbreaks.