欢迎光临散文网 会员登陆 & 注册

MIMO检测-基于因子图的加权高斯近似算法

2022-09-27 23:45 作者:乐吧的数学  | 我要投稿

本文讲解基于因子图的置信传播算法,来做 MIMO detection, 即根据接收到的数据,假定信道系数矩阵已知的前提下,来估计发送的数据。这个文章需要的背景知识很少,只需要基本的概率知识以及高斯分布就可以了。

(录制的视频:https://www.bilibili.com/video/BV1214y1h7v7/

系统图如下:

则:

 y%3D%20Hx%20%2B%20n

信道系数矩阵 H 已知,且已经接收到了数据,那么如果我们要估算发送方的数据,当然最优的做法,是求解下面的概率:

p(x%7Cy%2CH)%20%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F(1)



其中 x 和 y 都是列向量,分别包含 Nt 和 Nr 个元素。 Nt 和 Nr 分别表示发送天线数和接收天线数.


在所有 x 的可能取值中,找上面公式 (1) 的概率的最大值。


但是,这种最大化后验概率的方法,计算量随着发送天线数的增加而急剧增大,因此,我们可以退而求其次,我们不要求全局最优,我们把 Nt 个发送数据分别处理,对于 x_i,我们计算如下的概率:

p%5E%7Bk%2B%7D%20%3D%20p(x_k%20%3D%20%2B1%20%7C%20y%2CH)%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F(2)



如果大于 0.5,则认为 x_k%20%3D%20%2B1,否则,认为 x_k%20%3D%20-1.


我们把公式 (2) 用条件概率公式做一下推导,目的是推导出 用“收到 y”  概率 来表示这个 x_k 的概率。

%5Cbegin%7Baligned%7D%0A%0Ap%5E%7Bk%2B%7D%20%26%3D%20p(x_k%20%3D%20%2B1%20%7C%20y%2CH)%20%3D%5Cfrac%7Bp(x_k%3D%2B1%2Cy%7CH)%7D%7Bp(y%7CH)%7D%20%3D%20%5Cfrac%7Bp(y%7Cx_k%3D%2B1%2CH)%20p(x_k%3D%2B1%7CH)%7D%7Bp(y%7CH)%7D%20%20%5C%5C%0A%0A%26%3D%20%5Cfrac%7B1%7D%7Bp(y%7CH)%7D%20%20%20p(x_k%3D%2B1%7CH)%20%20%20p(y%7Cx_k%3D%2B1%2CH)%0A%0A%5Cquad%20----%20%E5%85%AC%E5%BC%8F(3)%0A%0A%0A%0A%5Cend%7Baligned%7D



其中,因为 x_k 与信道 H 是相互独立的,因此 p(x_k%3D%2B1%7CH)%20%20%3D%20p(x_k%3D%2B1),可以认为是常数。

其中 p(y|H) 用全概率公式展开为

p(y%7CH)%20%3D%20p(y%7Cx_k%3D%2B1%2CH)p(x_k%3D%2B1%7CH)%20%20%2B%20p(y%7Cx_k%3D-1%2CH)p(x_k%3D-1%7CH)



因为 x_k 与信道 H 是相互独立的,所以,上式继续推导为:

%5Cbegin%7Baligned%7D%0A%0Ap(y%7CH)%20%26%3D%20p(y%7Cx_k%3D%2B1%2CH)p(x_k%3D%2B1%7CH)%20%20%2B%20p(y%7Cx_k%3D-1%2CH)p(x_k%3D-1%7CH)%20%20%20%5C%5C%0A%0A%20%26%3D%20p(y%7Cx_k%3D%2B1%2CH)p(x_k%3D%2B1)%20%20%2B%20p(y%7Cx_k%3D-1%2CH)p(x_k%3D-1)%0A%0A%5Cend%7Baligned%7D



在假定  p(x_k%3D%2B1)%20%20%3D%20p(x_k%3D-1)%20%3D0.5,即符号是等概率取值的,则公式 (3) 可以整理为:


%5Cbegin%7Baligned%7D%0A%0Ap%5E%7Bk%2B%7D%20%26%3D%20p(x_k%3D%2B1)%20%5Cfrac%7Bp(y%7Cx_k%3D%2B1%2CH)%7D%7B%20p(y%7Cx_k%3D%2B1%2CH)p(x_k%3D%2B1)%20%20%2B%20p(y%7Cx_k%3D-1%2CH)p(x_k%3D-1)%7D%20%20%5C%5C%0A%0A%26%3D%20%5Cfrac%7Bp(y%7Cx_k%3D%2B1%2CH)%7D%7B%20p(y%7Cx_k%3D%2B1%2CH)%20%20%2B%20p(y%7Cx_k%3D-1%2CH)%7D%20%20%0A%0A%5Cend%7Baligned%7D%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F(4)


至此,我们做一个不太准确的假设,即假设 y_1%2Cy_2%2C....%2Cy_%7BN_r%7D 在 已知 H 和 x_k 的条件下,相互独立。但是,在实际上,这里肯定不是相互独立的,因为每个接收天线都能接收到所有发射天线来的信号,那么这些接收到的数据肯定都包括相互重叠的信息,即来自同一个发射天线的信息。所以,下面的公式,只能是约等于:

p(y%7Cx_k%3D%2B1%2CH)%20%5Capprox%20%20%5Cprod_%7Bi%3D1%7D%5E%7BN_r%7D%20%20p(y_i%7Cx_k%3D%2B1%2CH)%20%20%5C%5C%0A%0Ap(y%7Cx_k%3D-1%2CH)%20%5Capprox%20%20%5Cprod_%7Bi%3D1%7D%5E%7BN_r%7D%20%20p(y_i%7Cx_k%3D-1%2CH)



代入公式 (4) 有:


p%5E%7Bk%2B%7D%20%3D%20%5Cfrac%7B%20%5Cprod_%7Bi%3D1%7D%5E%7BN_r%7D%5Cfrac%7Bp(y_i%7Cx_k%3D%2B1%2CH)%7D%7Bp(y_i%7Cx_k%3D-1%2CH)%7D%20%20%7D%20%20%20%0A%0A%20%7B%20%20%5Cprod_%7Bi%3D1%7D%5E%7BN_r%7D%5Cfrac%7Bp(y_i%7Cx_k%3D%2B1%2CH)%7D%7Bp(y_i%7Cx_k%3D-1%2CH)%7D%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%2B%0A%0A%20%20%20%201%7D%20%20%5Cquad%20-----%20%E5%85%AC%E5%BC%8F(5)


令:

%5CLambda_i%5Ek%20%3D%20log%20%5Cfrac%7Bp(y_i%7Cx_k%3D%2B1%2CH)%7D%7Bp(y_i%7Cx_k%3D-1%2CH)%7D%20%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(6)


那么:

%5Cprod_%7Bi%3D1%7D%5E%7BN_r%7D%5Cfrac%7Bp(y_i%7Cx_k%3D%2B1%2CH)%7D%7Bp(y_i%7Cx_k%3D-1%2CH)%7D%20%3D%20exp(%7B%5Csum_%7Bi%3D1%7D%5E%7BN_r%7D%20%5CLambda_i%5Ek%7D)


代入公式 (5) 有:


p%5E%7Bk%2B%7D%20%3D%5Cfrac%7Bexp(%7B%5Csum_%7Bi%3D1%7D%5E%7BN_r%7D%20%5CLambda_i%5Ek%7D)%7D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%7Bexp(%7B%5Csum_%7Bi%3D1%7D%5E%7BN_r%7D%20%5CLambda_i%5Ek%7D)%20%2B%201%7D%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(7)



至此,我们已经用 y_i 的概率,表示出来了 x_k 的概率,即用接收方的概率信息,来估计发送方发送的是什么数据的概率。


接下来,我们需要更新了的对发送方的估计,来进一步提高对 y_i 的概率的估计,即提高 %5CLambda_i%5Ek 的准确度。看公式 (6) 中的 y_i,我们把 y_i 的公式写出来:

y_i%20%3D%20%5Csum_%7Bj%3D1%7D%5E%7BN_t%7D%20h_%7Bij%7Dx_j%20%2B%20n_i%20%3D%20h_%7Bik%7Dx_k%20%2B%20%20%5Cunderbrace%7B%5Csum_%7Bj%3D1%2Cj%5Cneq%20k%7D%5E%7BN_t%7D%20h_%7Bij%7Dx_j%7D_%7B%E5%B9%B2%E6%89%B0%7D%20%2B%20n_i%20%20%5Cquad%20%20----%20%E5%85%AC%E5%BC%8F(8)

这里,我们把来自不是 x_i 的发送信号,都视作干扰,这个干扰以及加性高斯白噪声项一起,构成了一个符合复高斯分布的随机变量 z_%7Bik%7D

y_i%20%3D%20%5Csum_%7Bj%3D1%7D%5E%7BN_t%7D%20h_%7Bij%7Dx_j%20%2B%20n_i%20%3D%20h_%7Bik%7Dx_k%20%2B%20%20%5Cunderbrace%7B%5Csum_%7Bj%3D1%2Cj%5Cneq%20k%7D%5E%7BN_t%7D%20h_%7Bij%7Dx_j%20%2B%20n_i%20%7D_%7Bz_%7Bik%7D%7D%20%5Cquad%20%20----%20%E5%85%AC%E5%BC%8F(9)

符合如下的复高斯分布:

%5Cbegin%7Baligned%7D%0A%0A%26%20CN(%5Cmu_%7Bz_%7Bik%7D%7D%2C%20%5Csigma%5E2_%7Bz_%7Bik%7D%7D)%20%20%5C%5C%20%5Cquad%5C%5C%0A%0A%5Cmu_%7Bz_%7Bik%7D%7D%20%26%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20h_%7Bij%7D%20E(x_j)%20%20%20%5C%5C%20%5Cquad%5C%5C%0A%0A%5Csigma%5E2_%7Bz_%7Bik%7D%7D%20%26%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20%7Ch_%7Bij%7D%7C%5E2%20%5Ctext%7BVar%7D(x_j)%20%2B%20%5Csigma%5E2%0A%0A%5Cend%7Baligned%7D



那么根据公式 (9) 和上面的假设,则  y_i 是符合 CN(%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7Dx_i%2C%20%5Csigma%5E2_%7Bz_%7Bik%7D%7D)%20 的复高斯分布。

那么:


p(y_i%7Cx_k%3D%2B1%2CH)%20%3D%20%5Cfrac%7B1%7D%7B%5Csqrt%7B%5Cpi%7D%20%5Csigma_%7Bik%7D%7D%20exp(%20-%20%5Cfrac%7B%7Cy_i%20-(%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D(%2B1))%7C%5E2%20%7D%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20)


类似的:

p(y_i%7Cx_k%3D-1%2CH)%20%3D%20%5Cfrac%7B1%7D%7B%5Csqrt%7B%5Cpi%7D%20%5Csigma_%7Bik%7D%7D%20exp(%20-%20%5Cfrac%7B%7Cy_i%20-(%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D(-1)%7C%5E2%20%7D%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20)


代入公式 (6) 有:


%5Cbegin%7Baligned%7D%0A%0A%5CLambda_i%5Ek%20%20%26%3D%20%5Cfrac%7B%7Cy_i%20-(%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D(-1)%7C%5E2%7D%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20-%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%5Cfrac%7B%7Cy_i%20-(%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D(%2B1))%7C%5E2%7D%7B%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20%7D%20%20%5C%5C%20%5Cquad%20%20%5C%5C%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%26%3D%20%5Cfrac%7B%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D%7C%5E2%20-%20%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D-h_%7Bik%7D%7C%5E2%7D%20%20%20%20%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20%20%20%20%5C%5C%20%5Cquad%20%20%5C%5C%0A%0A%5Cend%7Baligned%7D%20%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(10)


其中:


%5Cbegin%7Baligned%7D%0A%0A%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D%7C%5E2%20%26%3D%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D)%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%2Bh_%7Bik%7D)%5E*%20%20%5C%5C%0A%0A%26%3D%20((y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%2Bh_%7Bik%7D)((y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%5E*%2Bh_%7Bik%7D%5E*)%20%20%5C%5C%0A%0A%26%3D%20%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%7C%5E2%20%2B%20%7Ch_%7Bik%7D%7C%5E2%20%2B%20%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%20h_%7Bik%7D%5E*%20%2B%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%5E*%20h_%7Bik%7D%20%20%5C%5C%0A%0A%26%3D%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%7C%5E2%20%2B%20%7Ch_%7Bik%7D%7C%5E2%20%2B%20%202%20%5CRe%20(%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%20h_%7Bik%7D%5E*)%0A%0A%5Cend%7Baligned%7D


同理:


%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D-h_%7Bik%7D%7C%5E2%20%3D%20%7Cy_i%20-%5Cmu_%7Bz_%7Bik%7D%7D%7C%5E2%20%2B%20%7Ch_%7Bik%7D%7C%5E2%20-%20%202%20%5CRe%20(%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%20h_%7Bik%7D%5E*)


代入公式 (10) 有:

%5CLambda_i%5Ek%20%3D%20%5Cfrac%7B4%7D%7B%20%20%7B%5Csigma%5E2_%7Bz_%7Bik%7D%7D%7D%20%20%7D%20%20%20%20%5CRe%20(%20(y_i%20-%5Cmu_%7Bz_%7Bik%7D%7D)%20h_%7Bik%7D%5E*)%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(11)



现在,我们来推导公式 (11) 中用到的两个参数 %5Cmu_%7Bz_%7Bik%7D%7D 和  %5Cmu_%7Bz_%7Bik%7D%7D%24 :

%5Cmu_%7Bz_%7Bik%7D%7D%20%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20h_%7Bij%7D%20E(x_j)


其中

E(x_j)%20%20%3D%20(x_j%3D%2B1)%20p(x_j%3D%2B1)%20%2B%20(x_j%3D-1)%20p(x_j%3D-1)%20%3D%20p%5E%7Bj%2B%7D%20(-1)%20(%201-p%5E%7Bj%2B%7D)%20%3D%202%20p%5E%7Bj%2B%7D%20-%201


则:

%5Cmu_%7Bz_%7Bik%7D%7D%20%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20h_%7Bij%7D%20(%202%20p%5E%7Bj%2B%7D%20-%201%20)


另外,方差的部分:

%5Csigma%5E2_%7Bz_%7Bik%7D%7D%20%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20%7Ch_%7Bij%7D%7C%5E2%20%5Ctext%7BVar%7D(x_j)%20%2B%20%5Csigma%5E2


其中:

%5Cbegin%7Baligned%7D%0A%0A%5Ctext%7BVar%7D(x_j)%20%20%26%3D%20E(x_j%5E2)%20-%20(E(x_j))%5E2%20%3D%20(%201*1*p%5E%7Bj%2B%7D%20%2B%20(-1)(-1)(1-p%5E%7Bj%2B%7D))%20-%20(2%20p%5E%7Bj%2B%7D%20-%201)%5E2%20%20%5C%5C%20%5Cquad%20%20%5C%5C%0A%0A%20%20%20%20%26%3D%204%20p%5E%7Bj%2B%7D%20(%201-%20p%5E%7Bj%2B%7D)%0A%0A%5Cend%7Baligned%7D


则:


%5Csigma%5E2_%7Bz_%7Bik%7D%7D%20%3D%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20%7Ch_%7Bij%7D%7C%5E2%20%20%204p%5E%7Bj%2B%7D(1-p%5E%7Bj%2B%7D)%20%2B%20%5Csigma%5E2


最终,公式 (11) 变为:


%5CLambda_i%5Ek%20%3D%20%0A%0A%5Cfrac%7B4%7D%7B%20%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20%7Ch_%7Bij%7D%7C%5E2%20%20%204p%5E%7Bj%2B%7D(1-p%5E%7Bj%2B%7D)%20%2B%20%5Csigma%5E2%20%20%7D%20%20%20%0A%0A%20%5CRe%20(%20(y_i%20-%20(%20%20%5Csum_%7Bj%3D1%2Cj%20%5Cneq%20k%7D%5E%7BN_t%7D%20%20h_%7Bij%7D%20(%202%20p%5E%7Bj%2B%7D%20-%201%20)%20)%20h_%7Bik%7D%5E*)%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(12)


至此,我们已经有一个迭代的过程了:

1)用 y_i 的概率信息 %5CLambda_i%5Ek 来估算每个发送方数据的概率信息 p%5E%7Bj%2B%7D

2) 根据发送方概率信息 %20p%5E%7Bj%2B%7D,可以计算出相关的均值和方差 %20%5Cmu_%7Bz_%7Bik%7D%7D 和 %5Csigma%5E2_%7Bz_%7Bik%7D%7D%20, 进而可以又来估计 y_i 的概率。


因为我们这中间有一些假设导致的一种近似,所以,我们需要对上面两个步骤做多次迭代,才能收敛到一个稳定值。因为是迭代,所以,在后面的迭代过程中,公式(7) 中,计算左边的值时,需要把我们用来估计的 y_i 对应的概率踢出去,下面的公式中 l 表示要估计的 y 向量中元素的下标(而不是 i ), 公式 (7) 变为:


p%5E%7Bk%2B%7D_l%20%3D%5Cfrac%7Bexp(%7B%5Csum_%7Bi%3D1%2C%20i%5Cneq%20l%7D%5E%7BN_r%7D%20%5CLambda_i%5Ek%7D)%7D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%7Bexp(%7B%5Csum_%7Bi%3D1%2Ci%5Cneq%20l%7D%5E%7BN_r%7D%20%5CLambda_i%5Ek%7D)%20%2B%201%7D%20%20%5Cquad%20----%20%E5%85%AC%E5%BC%8F%20(13)



则公式(12) 和公式 (13) 一起,构成这个算法的迭代过程。


至此,我们引入因子图来表示这种迭代关系以及迭代过程中传递的概率信息(称之为消息)。



MIMO检测-基于因子图的加权高斯近似算法的评论 (共 条)

分享到微博请遵守国家法律