Learned Turbo-type Affine Rank Minimization

Xuehai He,Xiaojun Yuan,Zhipeng Xue
DOI: https://doi.org/10.1109/wcsp.2019.8928053
2019-01-01
Abstract:This paper is focused on the efficient algorithm design for affine rank minimization (ARM). Given the proliferation of the literature on the ARM problem, the existing algorithms mostly take a model-based approach in the algorithm design, and so are sensitive to the generation model of the linear measurement operator or to the generation model of the low-rank matrix to be recovered. It is known that, all these algorithms do not work well, e.g., when the low-rank matrix is ill-conditioned. As inspired by the success of learned iterative soft-thresholding (LIST) and learned approximate message passing (LAMP), we develop a learning-based message passing algorithm named learned turbo-type affine rank minimization (LTARM) to cope with the ARM problem. The LTARM algorithm learns its parameters from data, and hence is robust to the generation models of the linear operator and the low-rank matrix. We derive analytical expressions for the partial derivatives involved in training the LTARM network. Given the large size of the low-rank matrix in a typical ARM problem, these analytical expressions are of essential importance for the development of computationally feasible training in LTARM networks. Numerical results demonstrate that LTARM significantly outperforms the state-of-the-art counterparts for various generation models of the linear operator and the low-rank matrix, especially when the low-rank matrix is ill-conditioned.
What problem does this paper attempt to address?