CasFlow: Exploring Hierarchical Structures and Propagation Uncertainty for Cascade Prediction

Xovee Xu,Fan Zhou,Kunpeng Zhang,Siyuan Liu,Goce Trajcevski
DOI: https://doi.org/10.1109/tkde.2021.3126475
IF: 9.235
2022-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Understanding in-network information diffusion is a fundamental problem in many applications and one of the primary challenges is to predict the information cascade size. Most of the existing models rely either on hypothesized point process (e.g., Poisson and Hawkes processes), or simply predict the information propagation via deep neural networks. However, they fail to simultaneously capture the underlying global and local structures of a cascade and the propagation uncertainty in the diffusion, which may result in unsatisfactory prediction performance. To address these, in this work we propose a novel probabilistic cascade prediction framework CasFlow: Hierarchical Cascade Normalizing Flows. CasFlow allows a non-linear information diffusion inference and models the information diffusion process by learning the latent representation of both the structural and temporal information. It is a pattern-agnostic model leveraging normalizing flows to learn the node-level and cascade-level latent factors in an unsupervised manner. In addition, CasFlow is capable of capturing both the cascade representation uncertainty and node infection uncertainty, while enabling hierarchical pattern learning of information diffusion. Extensive experiments conducted on real-world datasets demonstrate that CasFlow reduces the prediction error to 21.0% by only observing half an hour of cascades, compared to state-of-the-art approaches, while also enabling model interpretability.
computer science, information systems, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?