Simple, fast, and flexible framework for matrix completion with infinite width neural networks.

Proc Natl Acad Sci U S A
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