LERE: Learning-Based Low-Rank Matrix Recovery with Rank Estimation

被引:0
作者
Xu, Zhengqin [1 ]
Zhang, Yulun [2 ]
Ma, Chao [1 ]
Yan, Yichao [1 ]
Peng, Zelin [1 ]
Xie, Shoulie [3 ]
Wu, Shiqian [4 ]
Yang, Xiaokang [1 ]
机构
[1] Shanghai Jiao Tong Univ, AI Inst, AMoE Key Lab Artificial Intelligence, Shanghai, Peoples R China
[2] Swiss Fed Inst Technol, Zurich, Switzerland
[3] Inst Infocomm Res A STAR, Signal Proc RF & Opt Dept, Singapore, Singapore
[4] Wuhan Univ Sci & Technol, Sch Informat Sci & Engn, Wuhan, Peoples R China
来源
THIRTY-EIGHTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 14 | 2024年
关键词
ROBUST PCA; COMPLETION; FACTORIZATION; RELAXATION;
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
A fundamental task in the realms of computer vision, Low-Rank Matrix Recovery (LRMR) focuses on the inherent low-rank structure precise recovery from incomplete data and/or corrupted measurements given that the rank is a known prior or accurately estimated. However, it remains challenging for existing rank estimation methods to accurately estimate the rank of an ill-conditioned matrix. Also, existing LRMR optimization methods are heavily dependent on the chosen parameters, and are therefore difficult to adapt to different situations. Addressing these issues, A novel LEarning-based low-rank matrix recovery with Rank Estimation (LERE) is proposed. More specifically, considering the characteristics of the Gerschgorin disk's center and radius, a new heuristic decision rule in the Gerschgorin Disk Theorem is significantly enhanced and the low-rank boundary can be exactly located, which leads to a marked improvement in the accuracy of rank estimation. According to the estimated rank, we select row and column sub-matrices from the observation matrix by uniformly random sampling. A 17-iteration feedforward-recurrent-mixed neural network is then adapted to learn the parameters in the sub-matrix recovery processing. Finally, by the correlation of the row sub-matrix and column sub-matrix, LERE successfully recovers the underlying low-rank matrix. Overall, LERE is more efficient and robust than existing LRMR methods. Experimental results demonstrate that LERE surpasses state-of-the-art (SOTA) methods. The code for this work is accessible at https://github.com/zhengqinxu/LERE.
引用
收藏
页码:16228 / 16236
页数:9
相关论文
共 37 条
  • [1] Unifying Nuclear Norm and Bilinear Factorization Approaches for Low-rank Matrix Decomposition
    Cabral, Ricardo
    De la Torre, Fernando
    Costeira, Joao P.
    Bernardino, Alexandre
    [J]. 2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2013, : 2488 - 2495
  • [2] Robust CUR Decomposition: Theory and Imaging Applications*
    Cai, HanQin
    Hamm, Keaton
    Huang, Longxiu
    Needell, Deanna
    [J]. SIAM JOURNAL ON IMAGING SCIENCES, 2021, 14 (04) : 1472 - 1503
  • [3] Rapid Robust Principal Component Analysis: CUR Accelerated Inexact Low Rank Estimation
    Cai, HanQin
    Hamm, Keaton
    Huang, Longxiu
    Li, Jiaqi
    Wang, Tao
    [J]. IEEE SIGNAL PROCESSING LETTERS, 2021, 28 : 116 - 120
  • [4] Cai HQ, 2019, J MACH LEARN RES, V20
  • [5] Robust Principal Component Analysis?
    Candes, Emmanuel J.
    Li, Xiaodong
    Ma, Yi
    Wright, John
    [J]. JOURNAL OF THE ACM, 2011, 58 (03)
  • [6] Chen YX, 2021, ANN STAT, V49, P2948, DOI [10.1214/21-AOS2066, 10.1214/21-aos2066]
  • [7] Chen YX, 2020, SIAM J OPTIMIZ, V30, P3098, DOI [10.1137/19M1290000, 10.1137/19m1290000]
  • [8] Cohen R, 2019, INT CONF ACOUST SPEE, P3212, DOI 10.1109/ICASSP.2019.8683030
  • [9] Gregor K., 2010, P 27 INT C MACH LEAR, P399, DOI DOI 10.5555/3104322.3104374
  • [10] Fast and Accurate Matrix Completion via Truncated Nuclear Norm Regularization
    Hu, Yao
    Zhang, Debing
    Ye, Jieping
    Li, Xuelong
    He, Xiaofei
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (09) : 2117 - 2130