A Flexible Two-Tower Model for Item Cold-Start Recommendation

Abstract

One of the main challenges in recommendation system is the item cold-start problem, where absence of historical interactions or ratings in new items makes recommendation difficult. In order to solve the cold-start problem, hybrid neural network models using meta data of the item as a feature is widely used. However, existing cold-start models tend to focus too much on utilizing the side information of items, which may not be flexible enough to capture the interaction information of users. In this study, we propose a flexible framework for better capturing the interaction information of users. Specifically, we incorporate the multiple choice learning scheme into the two-tower neural network which is a popular recommendation model that consists of two towers -one for users and one for items. In our proposed framework, we construct two encoders. One of the two encoders, the tightly-coupled encoder, focuses on the side information of items with which the user has interacted, the other one, loosely-coupled encoder, focuses the user's interaction information. We utilize Gumbel-Softmax to stochastically select the encoder, enhancing the flexibility that considers not only item feature but also user interaction information. We evaluate our proposed framework on two datasets -the MLIMDb dataset which is a combination of widely used the MovieLens and IMDb datasets based on common movies, and the CiteULike dataset. The experimental results show that our proposed framework achieves state-of-the-art results on cold-start recommendation. In the Recall@150 experiments on the CiteULike dataset, we achieved improvement of approximately 2.7% compared to the base model. In the Recall@150 experiments on the MLIMDb dataset, we achieved improvement of approximately 5.2% compared to the base model. We further show our proposed model improves the performance in the warmstart settings. In the Recall@100 experiments on the Citeulike dataset, we observed an improvement of approximately 1.3% compared to the base model. In the Recall@100 experiments on the MLIMDb dataset, we observed an improvement of approximately 3.9% compared to the base model. Our proposed framework provides a flexible approach for capturing the diverse aspects of users in recommendation systems, even for cold-start items. As demonstrated through extensive experiments, our proposed model outperforms several state-of-the-art models on both datasets.