Simple, fast, and flexible framework for matrix completion with infinite width neural networks.
Authors | |
Keywords | |
Abstract | SignificanceMatrix completion is a fundamental problem in machine learning that arises in various applications. We envision that our infinite width neural network framework for matrix completion will be easily deployable and produce strong baselines for a wide range of applications at limited computational costs. We demonstrate the flexibility of our framework through competitive results on virtual drug screening and image inpainting/reconstruction. Simplicity and speed are showcased by the fact that most results in this work require only a central processing unit and commodity hardware. Through its connection to semisupervised learning, our framework provides a principled approach for matrix completion that can be easily applied to problems well beyond those of image completion and virtual drug screening considered in this paper. |
Year of Publication | 2022
|
Journal | Proc Natl Acad Sci U S A
|
Volume | 119
|
Issue | 16
|
Pages | e2115064119
|
Date Published | 2022 Apr 19
|
ISSN | 1091-6490
|
DOI | 10.1073/pnas.2115064119
|
PubMed ID | 35412891
|
Links | |
Grant list | DMS-1651995 / National Science Foundation (NSF)
N00014-17-1-2147 / DOD | United States Navy | Office of Naval Research (ONR)
N00014-18-1-2765 / DOD | United States Navy | Office of Naval Research (ONR)
MIT-IBM Watson AI Lab / MIT-IBM Watson AI Lab
Simons Investigator Award / Simons Foundation
IIS-1815697 / National Science Foundation (NSF)
DMS-2031883 / National Science Foundation (NSF)
814639 / Simons Foundation
|