Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to main content

Advertisement

A knowledge-enhanced interest segment division attention network for click-through rate prediction

  • Original Article
  • Published:
Neural Computing and Applications Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Algorithm 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5

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.

Notes

  1. http://jmcauley.ucsd.edu/data/amazon/.

  2. https://grouplens.org/datasets/movielens/1m/.

  3. https://grouplens.org/datasets/hetrec-2011/.

References

  1. 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

  2. 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

  3. 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

    Article  Google Scholar 

  4. 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

  5. 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

  6. 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

  7. 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

  8. 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

  9. 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

  10. 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

    Google Scholar 

  11. 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

  12. 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

  13. 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

  14. 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

  15. 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

  16. 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

  17. 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

    Article  Google Scholar 

  18. 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

  19. 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

  20. 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

  21. 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

  22. 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

    Google Scholar 

  23. 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

  24. 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

  25. 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

  26. 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

  27. 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

  28. Rendle S (2010) Factorization machines. In: 2010 IEEE international conference on data mining. IEEE, pp 995–1000

  29. 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

  30. 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

  31. 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

  32. 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

  33. 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

  34. 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

  35. 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

  36. 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

  37. 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

  38. 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

    Google Scholar 

  39. 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

  40. 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

  41. 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

  42. 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

  43. Kenton JDM-WC, Toutanova LK (2019) Bert: Pre-training of deep bidirectional transformers for language understanding, 4171–4186

  44. 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

    Article  Google Scholar 

  45. 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

Download references

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

Authors

Corresponding author

Correspondence to Yuzhong Chen.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s00521-024-10330-y

Keywords