Tensor Completion with Side Information: A Riemannian Manifold Approach

Tengfei Zhou, Hui Qian, Zebang Shen, Chao Zhang, Congfu Xu

Proceedings of the Twenty-Sixth International Joint Conference on Artificial Intelligence

Main track. Pages 3539-3545.

https://doi.org/10.24963/ijcai.2017/495

PDF BibTeX

Abstract

By restricting the iterate on a nonlinear manifold, the recently proposed Riemannian optimization methods prove to be both efficient and effective in low rank tensor completion problems. However, existing methods fail to exploit the easily accessible side information, due to their format mismatch. Consequently, there is still room for improvement. To fill the gap, in this paper, a novel Riemannian model is proposed to tightly integrate the original model and the side information by overcoming their inconsistency. For this model, an efficient Riemannian conjugate gradient descent solver is devised based on a new metric that captures the curvature of the objective. Numerical experiments suggest that our method is more accurate than the state-of-the-art without compromising the efficiency.

Keywords

  • Machine Learning: Learning Preferences or Rankings
  • Machine Learning: Machine Learning
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards