Social Information Popularity Prediction Based on Heterogeneous Diffusion Attention Network

Xueqi Jia,Jiaxing Shang,Linjiang Zheng,Dajiang Liu,Weiwei Cao,Hong Sun
DOI: https://doi.org/10.18293/seke2022-075
2022-01-01
Abstract:—Information popularity prediction on social media platforms is a valuable and challenging issue. However, existing studies either neglect the correlation among different cascades, or lack a comprehensive consideration of user behavioral proximity and preference with respect to different messages. In this paper we propose a graph neural network-based framework named HeDAN (heterogeneous diffusion attention network), which com- prehensively considers various factors affecting the information diffusion to predict the information popularity more accurately. Specifically, we first construct a heterogeneous diffusion graph with two types of nodes ( user and message ) and three types of relations ( Friendship , Interaction , and Interest ). Among them, Friendship reflects the strength of social relationship between users, Interaction reflects the behavioral proximity between users, and Interest reflects user preference to messages. Next, a graph neural network model with hierarchical attention mechanism is proposed to learn from these relations. Specifically, at the node- level, we utilize the graph attention network to learn the subgraph structure and generate the representations of nodes under each specific relationship. At the semantic-level, we distinguish the importance of different nodes in different relations via multi- head self-attention mechanism. Extensive experimental results on three datasets show the superior performance of our proposed model over the state-of-the-arts.
What problem does this paper attempt to address?