View in EDS

Predictive Low Rank Matrix Learning Under Partial Observations: Mixed-Projection ADMM.

Saved in:
Bibliographic Details
Title: Predictive Low Rank Matrix Learning Under Partial Observations: Mixed-Projection ADMM.
Authors: Bertsimas, Dimitris1 (AUTHOR) dbertsim@mit.edu, Johnson, Nicholas A. G.2 (AUTHOR) nagj@mit.edu
Source: Machine Learning. Jun2026, Vol. 115 Issue 6, p1-53. 53p.
Abstract: We study the problem of learning a partially observed matrix under the low rank assumption in the presence of fully observed side information that depends linearly on the true underlying matrix. This problem consists of an important generalization of the Matrix Completion problem, a central problem in Statistics, Operations Research and Machine Learning, that arises in applications such as recommendation systems, signal processing, system identification and image denoising. We formalize this problem as an optimization problem with an objective that balances the strength of the fit of the reconstruction to the observed entries with the ability of the reconstruction to be predictive of the side information. We derive a mixed-projection reformulation of the resulting optimization problem and present a strong semidefinite cone relaxation. We design an efficient, scalable alternating direction method of multipliers algorithm that produces high quality feasible solutions to the problem of interest. Our numerical results demonstrate that in the small rank regime (), our algorithm outputs solutions that achieve on average lower objective value and lower reconstruction error than the solutions returned by the best performing benchmark method on synthetic data. The runtime of our algorithm is competitive with and often superior to that of the benchmark methods. Our algorithm is able to solve problems with rows and columns in less than a minute. On large scale real world data, our algorithm produces solutions that achieve lower out of sample error than benchmark methods in less execution time. [ABSTRACT FROM AUTHOR]
Copyright of Machine Learning is the property of Springer Nature and its content may not be copied or emailed to multiple sites without the copyright holder's express written permission. Additionally, content may not be used with any artificial intelligence tools or machine learning technologies. However, users may print, download, or email articles for individual use. This abstract may be abridged. No warranty is given about the accuracy of the copy. Users should refer to the original published version of the material for the full abstract. (Copyright applies to all Abstracts.)
Database: Engineering Source
Description
Abstract:We study the problem of learning a partially observed matrix under the low rank assumption in the presence of fully observed side information that depends linearly on the true underlying matrix. This problem consists of an important generalization of the Matrix Completion problem, a central problem in Statistics, Operations Research and Machine Learning, that arises in applications such as recommendation systems, signal processing, system identification and image denoising. We formalize this problem as an optimization problem with an objective that balances the strength of the fit of the reconstruction to the observed entries with the ability of the reconstruction to be predictive of the side information. We derive a mixed-projection reformulation of the resulting optimization problem and present a strong semidefinite cone relaxation. We design an efficient, scalable alternating direction method of multipliers algorithm that produces high quality feasible solutions to the problem of interest. Our numerical results demonstrate that in the small rank regime (), our algorithm outputs solutions that achieve on average lower objective value and lower reconstruction error than the solutions returned by the best performing benchmark method on synthetic data. The runtime of our algorithm is competitive with and often superior to that of the benchmark methods. Our algorithm is able to solve problems with rows and columns in less than a minute. On large scale real world data, our algorithm produces solutions that achieve lower out of sample error than benchmark methods in less execution time. [ABSTRACT FROM AUTHOR]
ISSN:08856125
DOI:10.1007/s10994-026-07005-1