Tensor Completion via Gaussian Process Based Initialization

Yermek KapushevIvan OseledetsEvgeny Burnaev

Skolkovo Institute of Science and Technology

arXiv 2019


In this paper, we consider the tensor completion problem representing the solution in the tensor train (TT) format. It is assumed that tensor is of high order, and tensor values are generated by an unknown smooth function. The assumption allows us to develop an efficient initialization scheme based on Gaussian Process Regression and TT-cross approximation technique. The proposed approach can be used in conjunction with any optimization algorithm that is usually utilized in tensor completion problems. We empirically justify that in this case the reconstruction error improves compared to the tensor completion with random initialization. As an additional benefit, our technique automatically selects rank thanks to using the TT-cross approximation technique.



Copy bibtex