PDF Publication   Supplementary   Experimental Code

The goal of recommendation system is to create an algorithm that is capable of learning to give accurate recommendation to the users, such as music recommendation on iTunes, or movie recommendation on Netflix. Most traditional approaches exploit user on item ratings (e.g., 1 to 5 stars) to learn a good recommendation. But in practice, people don't often leave a rating after watching a movie, making the traditional approach less useful. In this work, we assume a more natural case where only boolean ratings are present (e.g., user A watched item B or not), and develop effective algorithm to give accurate recommendation.
     The scenario can be illustrated by the above figure. To solve this problem, an approach is to learn a latent vector for each people and item, where the inner-product of the latent vectors characterize the inclination of the user on the item. Therefore to accurately predict human preference, we have to find the best latent representation for the users and the items. In this work, we proposed a general framework for incorporating feature information (e.g. the age and nationality of users or the characteristic of the item) and graph information (e.g. the Facebook friends relationship of the users) to learn a better representation. Furthermore, we investigate the use of general convex loss (where we are the first to design efficient optimization method for losses other than the square loss), and found that classification loss can find a more accurate latent representation, leading to substantially improved performance.

Publication:
H.-F. Yu, H.-Y. Huang, I. S. Dhillon, C.-J. Lin. A Unified Algorithm for One-class Structured Matrix Factorization with Side Information. AAAI Conference on Artificial Intelligence (AAAI 2017).