Abstract
The prediction of online information diffusion trends on social networks is crucial for understanding people’s interests and concerns, and has many real-world applications in fields such as business, politics and social security. Existing research on information diffusion prediction has predominantly focused on either macro level prediction of the future popularity of online information or micro level user activation prediction. However, information diffusion prediction on micro or macro level only may lead to one-sided prediction results, and there is a lack of research on implementing micro and macro level diffusion prediction tasks at the same time. Since micro and macro level diffusion prediction tasks are related to each other, we propose a unified information diffusion model which can jointly predicting the micro level user activation probability and macro level information popularity based on multi-task learning framework. We utilize graph neural network to learn user representation from both the information cascades and social network structure. Comparing with micro and macro level baseline methods separately, the prediction results of our proposed model outperform all baseline methods and proves the effectiveness of the proposed method.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Cao, Q., Shen, H., Cen, K., Ouyang, W., Cheng, X.: DeepHawkes: bridging the gap between prediction and understanding of information cascades. In: Proceedings of the 2017 ACM on Conference on Information and Knowledge Management, pp. 1149–1158. Association for Computing Machinery, New York (2017)
Chen, X., Zhang, K., Zhou, F., Trajcevski, G., Zhong, T., Zhang, F.: Information cascades modeling via deep multi-task learning. In: Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 885–888. Association for Computing Machinery, New York (2019)
Chen, X., Zhou, F., Zhang, K., Trajcevski, G., Zhong, T., Zhang, F.: Information diffusion prediction via recurrent cascades convolution. In: IEEE 35th International Conference on Data Engineering, pp. 770–781 (2019)
Li, C., Ma, J., Guo, X., Mei, Q.: DeepCas: an end-to-end predictor of information cascades. In: Proceedings of the 26th International Conference on World Wide Web, pp. 577–586. International World Wide Web Conferences Steering Committee, Republic and Canton of Geneva, CHE (2017)
Sankar, A., Zhang, X., Krishnan, A., Han, J.: InF-VAE: a variational autoencoder framework to integrate homophily and influence in diffusion prediction. In: Proceedings of the 13th International Conference on Web Search and Data Mining, pp. 510–518. Association for Computing Machinery, New York (2020)
Sun, L., Rao, Y., Zhang, X., Lan, Y., Yu, S.: MS-HGAT: memory-enhanced sequential hypergraph attention network for information diffusion prediction. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 36. pp. 4156–4164. AAAI Press (2022)
Wang, Z., Chen, C., LI, W.: A sequential neural information diffusion model with structure attention. In: Proceedings of the 27th ACM International Conference on Information and Knowledge Management, pp. 1795–1798. Association for Computing Machinery, New York (2018)
Wang, Z., Wei, W., Cong, G., Li, X.L., Mao, X.L., Qiu, M.: Global context enhanced graph neural networks for session-based recommendation. In: Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 169–178. Association for Computing Machinery, New York (2020)
Yang, C., Sun, M., Liu, H., Han, S., Liu, Z., Luan, H.: Neural diffusion model for microscopic cascade study. IEEE Trans. Knowl. Data Eng. 33(3), 1128–1139 (2021)
Yang, C., Tang, J., Sun, M., Cui, G., Liu, Z.: Multi-scale information diffusion prediction with reinforced recurrent networks. In: Proceedings of the 28th International Joint Conference on Artificial Intelligence, pp. 4033–4039. AAAI Press (2019)
Yu, L., Cui, P., Wang, F., Song, C., Yang, S.: From micro to macro: uncovering and predicting information cascading process with behavioral dynamics. In: Proceedings of the 2015 IEEE International Conference on Data Mining, pp. 559–568. IEEE Computer Society, USA (2015)
Yuan, C., Li, J., Zhou, W., Lu, Y., Zhang, X., Hu, S.: DyHGCN: a dynamic heterogeneous graph convolutional network to learn users’ dynamic preferences for information diffusion prediction. In: Hutter, F., Kersting, K., Lijffijt, J., Valera, I. (eds.) ECML PKDD 2020. LNCS (LNAI), vol. 12459, pp. 347–363. Springer, Cham (2021). https://doi.org/10.1007/978-3-030-67664-3_21
Acknowledgement
This work is supported by the National Natural Science Foundation of China No. 62172428, 61732022, 61732004.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Shang, Y., Zhou, B., Zeng, X., Chen, K. (2023). A Unified Information Diffusion Prediction Model Based on Multi-task Learning. In: Yang, X., et al. Advanced Data Mining and Applications. ADMA 2023. Lecture Notes in Computer Science(), vol 14179. Springer, Cham. https://doi.org/10.1007/978-3-031-46674-8_18
Download citation
DOI: https://doi.org/10.1007/978-3-031-46674-8_18
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-46673-1
Online ISBN: 978-3-031-46674-8
eBook Packages: Computer ScienceComputer Science (R0)