A Validation Approach to Over-parameterized Matrix and Image Recovery

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ding, Lijun, Qin, Zhen, Jiang, Liwei, Zhou, Jinxin, Zhu, Zhihui
Format: Preprint
Veröffentlicht: 2022
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918104440766464
author Ding, Lijun
Qin, Zhen
Jiang, Liwei
Zhou, Jinxin
Zhu, Zhihui
author_facet Ding, Lijun
Qin, Zhen
Jiang, Liwei
Zhou, Jinxin
Zhu, Zhihui
contents This paper studies the problem of recovering a low-rank matrix from several noisy random linear measurements. We consider the setting where the rank of the ground-truth matrix is unknown a priori and use an objective function built from a rank-overspecified factored representation of the matrix variable, where the global optimal solutions overfit and do not correspond to the underlying ground truth. We then solve the associated nonconvex problem using gradient descent with small random initialization. We show that as long as the measurement operators satisfy the restricted isometry property (RIP) with its rank parameter scaling with the rank of the ground-truth matrix rather than scaling with the overspecified matrix rank, gradient descent iterations are on a particular trajectory towards the ground-truth matrix and achieve nearly information-theoretically optimal recovery when it is stopped appropriately. We then propose an efficient stopping strategy based on the common hold-out method and show that it detects a nearly optimal estimator provably. Moreover, experiments show that the proposed validation approach can also be efficiently used for image restoration with deep image prior, which over-parameterizes an image with a deep network.
format Preprint
id arxiv_https___arxiv_org_abs_2209_10675
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle A Validation Approach to Over-parameterized Matrix and Image Recovery
Ding, Lijun
Qin, Zhen
Jiang, Liwei
Zhou, Jinxin
Zhu, Zhihui
Optimization and Control
Machine Learning
Image and Video Processing
This paper studies the problem of recovering a low-rank matrix from several noisy random linear measurements. We consider the setting where the rank of the ground-truth matrix is unknown a priori and use an objective function built from a rank-overspecified factored representation of the matrix variable, where the global optimal solutions overfit and do not correspond to the underlying ground truth. We then solve the associated nonconvex problem using gradient descent with small random initialization. We show that as long as the measurement operators satisfy the restricted isometry property (RIP) with its rank parameter scaling with the rank of the ground-truth matrix rather than scaling with the overspecified matrix rank, gradient descent iterations are on a particular trajectory towards the ground-truth matrix and achieve nearly information-theoretically optimal recovery when it is stopped appropriately. We then propose an efficient stopping strategy based on the common hold-out method and show that it detects a nearly optimal estimator provably. Moreover, experiments show that the proposed validation approach can also be efficiently used for image restoration with deep image prior, which over-parameterizes an image with a deep network.
title A Validation Approach to Over-parameterized Matrix and Image Recovery
topic Optimization and Control
Machine Learning
Image and Video Processing
url https://arxiv.org/abs/2209.10675