A Unified Algorithm for One-Cass Structured Matrix Factorization with Side Information

Authors

  • Hsiang-Fu Yu University of Texas at Austin
  • Hsin-Yuan Huang National Taiwan University
  • Inderjit Dhillon University of Texas at Austin
  • Chih-Jen Lin National Taiwan University

DOI:

https://doi.org/10.1609/aaai.v31i1.10863

Keywords:

Matrix Factorization, PU Learning, Multi-label Learning

Abstract

In many applications such as recommender systems and multi-label learning the task is to complete a partially observed binary matrix. Such PU learning (positive-unlabeled) problems can be solved by one-class matrix factorization (MF). In practice side information such as user or item features in recommender systems are often available besides the observed positive user-item connections. In this work we consider a generalization of one-class MF so that two types of side information are incorporated and a general convex loss function can be used. The resulting optimization problem is very challenging, but we derive an efficient and effective alternating minimization procedure. Experiments on large-scale multi-label learning and one-class recommender systems demonstrate the effectiveness of our proposed approach.

Downloads

Published

2017-02-13

How to Cite

Yu, H.-F., Huang, H.-Y., Dhillon, I., & Lin, C.-J. (2017). A Unified Algorithm for One-Cass Structured Matrix Factorization with Side Information. Proceedings of the AAAI Conference on Artificial Intelligence, 31(1). https://doi.org/10.1609/aaai.v31i1.10863