Graph neural network-based light-weight recommendation system

Lin Sun; Shaohua Kuang; Shujian Chen; Xinlin Li; Baobin Duan

doi:10.1117/12.2682297

26 May 2023 Graph neural network-based light-weight recommendation system

Lin Sun, Shaohua Kuang, Shujian Chen, Xinlin Li, Baobin Duan

Proceedings Volume 12700, International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023); 127003I (2023) https://doi.org/10.1117/12.2682297
Event: International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023), 2023, Nanchang, China

Abstract

The recommendation system aims to obtain valuable information for users and alleviate information overload. Graph Neural Network (GNN) is one the mainstream method of recommendation systems for its powerful capabilities of graph data representation and deep feature extraction. But there are still problems with the efficiency and accuracy of GNNs. Therefore, a High-Efficiency Graph Neural Network (HEGNN) is proposed in this paper to build a lightweight graph recommendation system. HEGNN strengthens the local and global preferences of users with attention blocks. It abandons the feature transformation and nonlinear activation layer of vanilla GNNs. Only the basic components are reserved to improve efficiency. Comprehensive comparative experiments with nine baseline algorithms are carried out on three benchmark datasets which include Amazon-Book Dataset, Yelp2018 Dataset, and Gowalla Dataset. Compared with existing recommendation methods such as NGCF and LightGCN, HEGCN not only achieves the highest score on two evaluation metrics of Normalized Discounted Cumulative Gain and Recall but also requires the least training time.

Citation Download Citation

Lin Sun, Shaohua Kuang, Shujian Chen, Xinlin Li, and Baobin Duan "Graph neural network-based light-weight recommendation system", Proc. SPIE 12700, International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023), 127003I (26 May 2023); https://doi.org/10.1117/12.2682297

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