Proceedings

Scalable Maximum Margin Matrix Factorization by Active Riemannian Subspace Search

Yan Yan, Mingkui Tan, Ivor W. Tsang, Yi Yang, Chengqi Zhang, Qinfeng Shi

Download BibTeX Cite

Proceedings of the Twenty-Fourth International Joint Conference on Artificial Intelligence. Special Track on Machine Learning. Pages 3988-3993.

Abstract

The user ratings in recommendation systems are usually in the form of ordinal discrete values. To give more accurate prediction of such rating data, maximum margin matrix factorization (M3F) was proposed. Existing M3F algorithms, however, either have massive computational cost or require expensive model selection procedures to determine the number of latent factors (i.e. the rank of the matrix to be recovered), making them less practical for large scale data sets. To address these two challenges, in this paper, we formulate M3F with a known number of latent factors as the Riemannian optimization problem on a fixed-rank matrix manifold and present a block-wise nonlinear Riemannian conjugate gradient method to solve it efficiently. We then apply a simple and efficient active subspace search scheme to automatically detect the number of latent factors. Empirical studies on both synthetic data sets and large real-world data sets demonstrate the superior efficiency and effectiveness of the proposed method.

Yan Yan, Mingkui Tan, Ivor W. Tsang, Yi Yang, Chengqi Zhang, and Qinfeng Shi. 2015. Scalable Maximum Margin Matrix Factorization by Active Riemannian Subspace Search. In Proceedings of the Twenty-Fourth International Joint Conference on Artificial Intelligence, IJCAI-15. IJCAI, 3988–3993. https://ijcai.org/proceedings/2015/560
Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards