Tensor factorization using auxiliary information

  • Authors:
  • Atsuhiro Narita;Kohei Hayashi;Ryota Tomioka;Hisashi Kashima

  • Affiliations:
  • Department of Mathematical Informatics, The University of Tokyo, Tokyo, Japan;Graduate School of Information Science, Nara Institute of Science and Technology, Ikoma, Nara, Japan;Department of Mathematical Informatics, The University of Tokyo, Tokyo, Japan;Department of Mathematical Informatics, The University of Tokyo, Tokyo, Japan and Synthesis of Knowledge for Information Oriented Society

  • Venue:
  • ECML PKDD'11 Proceedings of the 2011 European conference on Machine learning and knowledge discovery in databases - Volume Part II
  • Year:
  • 2011

Quantified Score

Hi-index 0.00

Visualization

Abstract

Most of the existing analysis methods for tensors (or multiway arrays) only assume that tensors to be completed are of low rank. However, for example, when they are applied to tensor completion problems, their prediction accuracy tends to be significantly worse when only limited entries are observed. In this paper, we propose to use relationships among data as auxiliary information in addition to the low-rank assumption to improve the quality of tensor decomposition. We introduce two regularization approaches using graph Laplacians induced from the relationships, and design iterative algorithms for approximate solutions. Numerical experiments on tensor completion using synthetic and benchmark datasets show that the use of auxiliary information improves completion accuracy over the existing methods based only on the low-rank assumption, especially when observations are sparse.