Abstract
Click-through rate (CTR) prediction aims to estimate the probability of a user clicking on a particular item, making it one of the core tasks in various recommendation platforms. In such systems, user behavior data are crucial for capturing user interests, which has garnered significant attention from both academia and industry, leading to the development of various user behavior modeling methods. However, existing models still face unresolved issues, as they fail to capture the complex diversity of user interests at the semantic level, refine user interests effectively, and uncover users’ potential interests. To address these challenges, we propose a novel model called knowledge-enhanced Interest segment division attention network (KISDAN), which can effectively and comprehensively model user interests. Specifically, to leverage the semantic information within user behavior sequences, we employ the structure of a knowledge graph to divide user behavior sequence into multiple interest segments. To provide a comprehensive representation of user interests, we further categorize user interests into strong and weak interests. By leveraging both the knowledge graph and the item co-occurrence graph, we explore users’ potential interests from two perspectives. This methodology allows KISDAN to better understand the diversity of user interests. Finally, we extensively evaluate KISDAN on three benchmark datasets, and the experimental results consistently demonstrate that the KISDAN model outperforms state-of-the-art models across various evaluation metrics, which validates the effectiveness and superiority of KISDAN.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Data availability
The data used in this article are available in the online supplementary material. Supplementary materials are available at http://jmcauley.ucsd.edu/data/amazon/, https://grouplens.org/datasets/movielens/1m/ and https://grouplens.org/datasets/hetrec-2011/.
Code availability
The code for this paper has been uploaded to Github: https://github.com/java-jay/KISDAN.
References
Zhou G, Zhu X, Song C, Fan Y, Zhu H, Ma X, Yan Y, Jin J, Li H, Gai K (2018) Deep interest network for click-through rate prediction. In: Proceedings of the 24th ACM SIGKDD international conference on knowledge discovery & data mining. pp 1059–1068
Li F, Chen Z, Wang P, Ren Y, Zhang D, Zhu X (2019) Graph intention network for click-through rate prediction in sponsored search. In: Proceedings of the 42nd international ACM SIGIR conference on research and development in information retrieval. pp 961–964
Zhou G, Mou N, Fan Y, Pi Q, Bian W, Zhou C, Zhu X, Gai K (2019) Deep interest evolution network for click-through rate prediction. Proceedings of the AAAI conference on artificial intelligence 33:5941–5948
Feng Y, Lv F, Shen W, Wang M, Sun F, Zhu Y, Yang K (2019) Deep session interest network for click-through rate prediction. In: Proceedings of the 28th international joint conference on artificial intelligence. pp2301–2307
Pi Q, Bian W, Zhou G, Zhu X, Gai K (2019) Practice on long sequential user behavior modeling for click-through rate prediction. In: Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining. pp2671–2679
Xiao Z, Yang L, Jiang W, Wei Y, Hu Y, Wang H (2020) Deep multi-interest network for click-through rate prediction. In: Proceedings of the 29th ACM international conference on information & knowledge management. pp2265–2268
Jiang W, Jiao Y, Wang Q, Liang C, Guo L, Zhang Y, Sun Z, Xiong Y, Zhu Y (2022) Triangle graph interest network for click-through rate prediction. In: Proceedings of the fifteenth ACM international conference on web search and data mining. pp401–409
Fan Z, Ou D, Gu Y, Fu B, Li X, Bao W, Dai X-Y, Zeng X, Zhuang T, Liu Q (2022) Modeling users’ contextualized page-wise feedback for click-through rate prediction in e-commerce search. In: Proceedings of the fifteenth ACM international conference on web search and data mining. pp 262–270
Lin Q, Zhou W-J, Wang Y, Da Q, Chen Q-G, Wang B (2022) Sparse attentive memory network for click-through rate prediction with long sequences. In: Proceedings of the 31st ACM international conference on information & knowledge management. pp3312–3321
Dong J, Yu Y, Zhang Y, Lv Y, Wang S, Jin B, Wang Y, Wang X, Wang D (2023) A deep behavior path matching network for click-through rate prediction. Companion proceedings of the ACM web conference 2023:538–542
Feng Y, Hu B, Lv F, Liu Q, Zhang Z, Ou W (2020) Atbrg: adaptive target-behavior relational graph network for effective recommendation. In: Proceedings of the 43rd international ACM SIGIR conference on research and development in information retrieval. pp 2231–2240
Feng Y, Lv F, Hu B, Sun F, Kuang K, Liu Y, Liu Q, Ou W (2020) Mtbrn: multiplex target-behavior relation enhanced network for click-through rate prediction. In: Proceedings of the 29th ACM international conference on information & knowledge management. pp 2421–2428
Zhang F, Yuan NJ, Lian D, Xie X, Ma W-Y (2016) Collaborative knowledge base embedding for recommender systems. In: Proceedings of the 22nd ACM SIGKDD international conference on knowledge discovery and data mining. pp 353–362
Wang H, Zhang F, Xie X, Guo M (2018) Dkn: deep knowledge-aware network for news recommendation. In: Proceedings of the 2018 World Wide Web conference. pp 1835–1844
Yu X, Ren X, Sun Y, Gu Q, Sturt B, Khandelwal U, Norick B, Han J (2014) Personalized entity recommendation: A heterogeneous information network appoach. In: Proceedings of the 7th ACM international conference on web search and data mining. pp 283–292
Hu B, Shi C, Zhao WX, Yu PS (2018) Leveraging meta-path based context for top-n recommendation with a neural co-attention model. In: Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. pp 1531–1540
Wang X, Wang D, Xu C, He X, Cao Y, Chua T-S (2019) Explainable reasoning over knowledge graphs for recommendation. Proceedings of the AAAI Conference on artificial intelligence 33:5329–5336
Wang H, Zhang F, Wang J, Zhao M, Li W, Xie X, Guo M (2018) Rippenet: Propagating user preferences on the knowledge graph for recommender systems. In: Proceedings of the 27th ACM international conference on information and knowledge management. pp 417–426
Wang H, Zhao M, Xie X, Li W, Guo M (2019) Knowledge graph convolutional networks for recommender systems. In: The World Wide Web conference. pp 3307–3313
Wang X, He X, Cao Y, Liu M, Chua T-S (2019) Kgat: knowledge graph attention network for recommendation. In: Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining. pp 950–958
Wang Z, Lin G, Tan H, Chen Q, Liu X (2020) Ckan: Collaborative knowledge-aware attentive network for recommender systems. In: Proceedings of the 43rd international ACM SIGIR conference on research and development in information retrieval. pp 219–228
Wang X, Huang T, Wang D, Yuan Y, Liu Z, He X, Chua T-S (2021) Learning intents behind interactions with knowledge graph for recommendation. Proceedings of the web conference 2021:878–887
Chen Y, Yang Y, Wang Y, Bai J, Song X, King I (2022) Attentive knowledge-aware graph convolutional networks with collaborative guidance for personalized recommendation. In: 2022 IEEE 38th international conference on data engineering (ICDE). IEEE, pp 299–311
Zou D, Wei W, Wang Z, Mao X-L, Zhu F, Fang R, Chen D (2022) Improving knowledge-aware recommendation with multi-level interactive contrastive learning. In: Proceedings of the 31st ACM international conference on information & knowledge management. pp 2817–2826
Zou D, Wei W, Mao X-L, Wang Z, Qiu M, Zhu F, Cao X (2022) Multi-level cross-view contrastive learning for knowledge-aware recommender system. In: Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval. pp 1358–1368
Du Y, Zhu X, Chen L, Zheng B, Gao Y (2022) Hakg: hierarchy-aware knowledge gated network for recommendation. In: Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval. pp 1390–1400
Huang S, Hu C, Kong W, Liu Y (2023) Disentangled contrastive learning for knowledge-aware recommender system. In: International semantic web conference. Springer, pp 140–158
Rendle S (2010) Factorization machines. In: 2010 IEEE international conference on data mining. IEEE, pp 995–1000
Guo H, Tang R, Ye Y, Li Z, He X (2017) Deepfm: a factorization-machine based neural network for ctr prediction. In: Proceedings of the 26th international joint conference on artificial intelligence. pp 1725–1731
Lian J, Zhou X, Zhang F, Chen Z, Xie X, Sun G (2018) xdeepfm: combining explicit and implicit feature interactions for recommender systems. In: Proceedings of the 24th ACM SIGKDD international conference on knowledge discovery & data mining. pp 1754–1763
Hadsell R, Chopra S, LeCun Y (2006) Dimensionality reduction by learning an invariant mappng. In: 2006 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR’06), vol. 2, pp 1735–1742. IEEE
Chen T, Kornblith S, Norouzi M, Hinton G (2020) A simple framework for contrastive learning of visual representations. In: International conference on machine learning. PMLR, pp 1597–1607
Yang Z, Cheng Y, Liu Y, Sun M (2019) Reducing word omission errors in neural machine translation: a contrastive learning appoach. In: Proceedings of the 57th annual meeting of the Association for Computational Linguistics. pp 6191–6196
Xie X, Sun F, Liu Z, Wu S, Gao J, Zhang J, Ding B, Cui B (2022) Contrastive learning for sequential recommendation. In: 2022 IEEE 38th international conference on data engineering (ICDE). IEEE, pp 1259–1273
Wu J, Wang X, Feng F, He X, Chen L, Lian J, Xie X (2021) Self-supervised graph learning for recommendation. In: Proceedings of the 44th international ACM SIGIR conference on research and development in information retrieval. pp 726–735
Wei Y, Wang X, Li Q, Nie L, Li Y, Li X, Chua T-S (2021) Contrastive learning for cold-start recommendation. In: Proceedings of the 29th ACM international conference on multimedia. pp 5382–5390
Zhang L, Liu Y, Zhou X, Miao C, Wang G, Tang H (2022) Diffusion-based graph contrastive learning for recommendation with implicit feedback. In: International Conference on database systems for advanced appications. Springer, pp 232–247
Zheng Y, Gao C, Chang J, Niu Y, Song Y, Jin D, Li Y (2022) Disentangling long and short-term interests for recommendation. Proceedings of the ACM web conference 2022:2256–2267
Zhang Y, Liu Y, Xu Y, Xiong H, Lei C, He W, Cui L, Miao C (2022) Enhancing sequential recommendation with graph contrastive learning. arXiv preprint arXiv:2205.14837
Guo W, Zhang C, He Z, Qin J, Guo H, Chen B, Tang R, He X, Zhang R (2022) Miss: multi-interest self-supervised learning framework for click-through rate prediction. In: 2022 IEEE 38th international conference on data engineering (ICDE). IEEE, pp 727–740
Zhang S, Li B, Yao D, Feng F, Zhu J, Fan W, Zhao Z, He X, Chua T-s, Wu F (2022) Ccl4rec: contrast over contrastive learning for micro-video recommendation. arXiv preprint arXiv:2208.08024
Wang F, Wang Y, Li D, Gu H, Lu T, Zhang P, Gu N (2023) Cl4ctr: A contrastive learning framework for CTR prediction. In: Proceedings of the sixteenth ACM international conference on web search and data mining. pp 805–813
Kenton JDM-WC, Toutanova LK (2019) Bert: Pre-training of deep bidirectional transformers for language understanding, 4171–4186
Zhu C, Chen B, Zhang W, Lai J, Tang R, He X, Li Z, Yu Y (2021) Aim: automatic interaction machine for click-through rate prediction. IEEE Trans Knowl Data Eng 35(4):3389–3403
Zheng Z, Zhang C, Gao X, Chen G (2022) Hien: hierarchical intention embedding network for click-through rate prediction. In: Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval. pp 322–331
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China under Grant 61672158, 61972097 and U21A20472, in part by the Major Science and Technology project of Fujian Province (China) under Granted No. 2021HZ022007, in part by the Industry-Academy Cooperation Project under Grant 2021H6022, in part by the Natural Science Foundation of Fujian Province under Grant 2020J01494, in part by the Collaborative Innovation Platform Project of Fuzhou City under Grant 2023-P-002.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Liu, Z., Chen, S., Chen, Y. et al. A knowledge-enhanced interest segment division attention network for click-through rate prediction. Neural Comput & Applic 36, 21817–21837 (2024). https://doi.org/10.1007/s00521-024-10330-y
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00521-024-10330-y