Hybrid recommendation model based on multi-head attention mechanism and cross network fusion

Shaoguo Cui; Gang Zhang; Aodi Wang

doi:10.1117/12.2671053

28 April 2023 Hybrid recommendation model based on multi-head attention mechanism and cross network fusion

Shaoguo Cui, Gang Zhang, Aodi Wang

Proceedings Volume 12610, Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022); 126100X (2023) https://doi.org/10.1117/12.2671053
Event: Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022), 2022, Wuhan, China

Abstract

The combination of low and high order features in the recommendation system is crucial to the predicted click through rate. This paper designs an Attention Deep Cross Attention Recognition Machine (ADCAFM). Traditional recommendation models only use attention factor decomposers and deep cross networks to extract low and high order features, but the diversity of deep cross networks mining user interests is weak. Therefore, this paper extracts the feature depth of different subspaces by integrating the multi head attention mechanism to solve the problem of user interest diversity in deep cross network mining; Finally, the low and high order combined features are effectively fused and recommended together. Through experimental comparison on Criteo and Movielens-100K data sets, the AUC index is used for evaluation. Compared with the benchmark model, the AUC index is 1.85% and 1.55% higher.

Citation Download Citation

Shaoguo Cui, Gang Zhang, and Aodi Wang "Hybrid recommendation model based on multi-head attention mechanism and cross network fusion", Proc. SPIE 12610, Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022), 126100X (28 April 2023); https://doi.org/10.1117/12.2671053

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available