Matrix Completion with Covariate Information

Xiaojun Mao,Song Xi Chen,Raymond K. W. Wong
DOI: https://doi.org/10.1080/01621459.2017.1389740
IF: 4.369
2019-01-01
Journal of the American Statistical Association
Abstract:This article investigates the problem of matrix completion from the corrupted data, when the additional covariates are available. Despite being seldomly considered in the matrix completion literature, these covariates often provide valuable information for completing the unobserved entries of the high-dimensional target matrix A(0). Given a covariate matrix X with its rows representing the row covariates of A(0), we consider a column-space-decomposition model A(0) = X-0 + B-0, where (0) is a coefficient matrix and B-0 is a low-rank matrix orthogonal to X in terms of column space. This model facilitates a clear separation between the interpretable covariate effects (X-0) and the flexible hidden factor effects (B-0). Besides, our work allows the probabilities of observation to depend on the covariate matrix, and hence a missing-at-random mechanism is permitted. We propose a novel penalized estimator for A(0) by utilizing both Frobenius-norm and nuclear-norm regularizations with an efficient and scalable algorithm. Asymptotic convergence rates of the proposed estimators are studied. The empirical performance of the proposed methodology is illustrated via both numerical experiments and a real data application.
What problem does this paper attempt to address?