BERT and CNN integrated Neural Collaborative Filtering for Recommender Systems

TL;DR

This paper introduces a novel neural collaborative filtering model integrating BERT and CNN to enhance recommendation systems by effectively utilizing user and item profile data, including images, leading to improved performance.

Contribution

The paper presents a new BERT and CNN integrated NCF model that handles diverse data types and outperforms baseline models on the MovieLens dataset.

Findings

Proposed model achieves 0.72 recall and 0.486 Hit Ratio @ 10.

Incorporating categorical and image data improves recommendation accuracy.

Model outperforms simple NCF and BERT-based NCF baselines.

Abstract

Every day, a significant number of users visit the internet for different needs. The owners of a website generate profits from the user interaction with the contents or items of the website. A robust recommendation system can increase user interaction with a website by recommending items according to the user's unique preferences. BERT and CNN-integrated neural collaborative filtering (NCF) have been proposed for the recommendation system in this experiment. The proposed model takes inputs from the user and item profile and finds the user's interest. This model can handle numeric, categorical, and image data to extract the latent features from the inputs. The model is trained and validated on a small sample of the MovieLens dataset for 25 epochs. The same dataset has been used to train and validate a simple NCF and a BERT-based NCF model and compared with the proposed model. The proposed model outperformed those two baseline models. The obtained result for the proposed model is 0.72 recall and 0.486 Hit Ratio @ 10 for 799 users on the MovieLens dataset. This experiment concludes that considering both categorical and image data can improve the performance of a recommendation system.