BERT and CNN integrated Neural Collaborative Filtering for Recommender Systems
By: Abdullah Al Munem, Sumona Yeasmin, Mohammad Rezwanul Huq
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.
Similar Papers
Benchmarking Deep Neural Networks for Modern Recommendation Systems
Information Retrieval
Finds best movies and products for you.
BeLightRec: A lightweight recommender system enhanced with BERT
Information Retrieval
Improves movie suggestions by understanding item details.
BiCoRec: Bias-Mitigated Context-Aware Sequential Recommendation Model
Information Retrieval
Recommends more of what you *really* like.