10
24

Fast and Sample Efficient Inductive Matrix Completion via Multi-Phase Procrustes Flow

Xiao Zhang
S. Du
Quanquan Gu
Abstract

We revisit the inductive matrix completion problem that aims to recover a rank-rr matrix with ambient dimension dd given nn features as the side prior information. The goal is to make use of the known nn features to reduce sample and computational complexities. We present and analyze a new gradient-based non-convex optimization algorithm that converges to the true underlying matrix at a linear rate with sample complexity only linearly depending on nn and logarithmically depending on dd. To the best of our knowledge, all previous algorithms either have a quadratic dependency on the number of features in sample complexity or a sub-linear computational convergence rate. In addition, we provide experiments on both synthetic and real world data to demonstrate the effectiveness of our proposed algorithm.

View on arXiv
Comments on this paper