Generalized Unitary Approximate Message Passing for Double Linear Transformation Model

The double linear transformation model Y = AXB + W plays an important role in a variety of science and engineering applications, where X is estimated through known transformation matrices A and B from the noisy measurement Y. Decoupling X from Y is a formidable task due to the high complexity brought by the multiplication of the unknown matrix (vector) with the transformation matrix (M-UMTM). Unitary approximate message passing (UAMP) has been verified as a low complexity and strong robustness solution to the M-UMTM problems. However, it has only been used to tackle the problems with a single linear transformation matrix. In this work, we develop a generalized algorithm, namely, generalized double UAMP (GD-UAMP) for the target model, which not only inherits the low complexity of AMP, but also enhances robustness by employing double unitary transformation. As a generalized algorithm, GD-UAMP can be applied to address the generalized Bayesian inference problem, i.e., the arbitrary prior probability of X and likelihood function of Z, where Z = AXB is the noiseless measurement. We verify the feasibility of the proposed algorithm in the channel estimation problem for various wireless communication systems. Numerical results demonstrate that the proposed algorithm can perfectly fit different scenarios and showcase superior performance compared with benchmarks.