Timezone: »
We study the problem of estimating a rank-1 signal in the presence of rotationally invariant noise--a class of perturbations more general than Gaussian noise. Principal Component Analysis (PCA) provides a natural estimator, and sharp results on its performance have been obtained in the high-dimensional regime. Recently, an Approximate Message Passing (AMP) algorithm has been proposed as an alternative estimator with the potential to improve the accuracy of PCA. However, the existing analysis of AMP requires an initialization that is both correlated with the signal and independent of the noise, which is often unrealistic in practice. In this work, we combine the two methods, and propose to initialize AMP with PCA. Our main result is a rigorous asymptotic characterization of the performance of this estimator. Both the AMP algorithm and its analysis differ from those previously derived in the Gaussian setting: at every iteration, our AMP algorithm requires a specific term to account for PCA initialization, while in the Gaussian case, PCA initialization affects only the first iteration of AMP. The proof is based on a two-phase artificial AMP that first approximates the PCA estimator and then mimics the true AMP. Our numerical simulations show an excellent agreement between AMP results and theoretical predictions, and suggest an interesting open direction on achieving Bayes-optimal performance.
Author Information
Marco Mondelli (IST Austria)
Ramji Venkataramanan (University of Cambridge)
More from the Same Authors
-
2022 : Mean-field analysis for heavy ball methods: Dropout-stability, connectivity, and global convergence »
Diyuan Wu · Vyacheslav Kungurtsev · Marco Mondelli -
2023 Poster: Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Model »
Peter Súkeník · Marco Mondelli · Christoph Lampert -
2022 : Poster Session 1 »
Andrew Lowy · Thomas Bonnier · Yiling Xie · Guy Kornowski · Simon Schug · Seungyub Han · Nicolas Loizou · xinwei zhang · Laurent Condat · Tabea E. Röber · Si Yi Meng · Marco Mondelli · Runlong Zhou · Eshaan Nichani · Adrian Goldwaser · Rudrajit Das · Kayhan Behdin · Atish Agarwala · Mukul Gagrani · Gary Cheng · Tian Li · Haoran Sun · Hossein Taheri · Allen Liu · Siqi Zhang · Dmitrii Avdiukhin · Bradley Brown · Miaolan Xie · Junhyung Lyle Kim · Sharan Vaswani · Xinmeng Huang · Ganesh Ramachandra Kini · Angela Yuan · Weiqiang Zheng · Jiajin Li -
2022 Poster: The price of ignorance: how much does it cost to forget noise structure in low-rank matrix estimation? »
Jean Barbier · TianQi Hou · Marco Mondelli · Manuel Saenz -
2022 Poster: Memorization and Optimization in Deep Neural Networks with Minimum Over-parameterization »
Simone Bombari · Mohammad Hossein Amani · Marco Mondelli -
2021 Poster: When Are Solutions Connected in Deep Networks? »
Quynh Nguyen · Pierre Bréchet · Marco Mondelli -
2020 Poster: Global Convergence of Deep Networks with One Wide Layer Followed by Pyramidal Topology »
Quynh Nguyen · Marco Mondelli