极化码的matlab仿真(2)——编码

时间:2023-03-09 04:03:02
极化码的matlab仿真(2)——编码

第二篇我们来介绍一下极化码的编码。

首先为了方便进行编码,我们需要进行数组的定义

signal = randi([0,1],1,ST);       %信息位比特,随机二进制数
frozen = zeros(1,FT); %固定位比特,规定全为0
encode = zeros(1,N * block); %编码后的比特
noise = snr(i) ^ 1/2 * randn(1,N * block); %加性高斯白噪声

极化码的编码重点在于生成矩阵的产生,以及信息位、冻结位的选取。

  • 我们先来看生成矩阵的产生。

极化码的matlab仿真(2)——编码

这是Arikan论文中的编码示意图,好像挺复杂,不过看不懂也没关系。我们来看一下编码过程中都做了哪些事。

首先是向量元素的翻转,通过翻转矩阵RN来实现,然后是信道的联合和信道的分裂。

什么?你问我为什么要这样做?我也不知道,极化码本身就建立在信道极化的现象之上,信道极化就是信道以特定方式联合和分裂所产生的现象。要想问为什么这么做,到土耳其找Arikan教授喝茶去吧。

论文中将这个翻转过程简化为矩阵运算,这就为我们进行程序仿真提供了方便:

极化码的matlab仿真(2)——编码

其中:

极化码的matlab仿真(2)——编码

BN 为排序矩阵,以N=8为例,解释它的作用:

极化码的matlab仿真(2)——编码

传化过程为:%将向量下标减一后,转化为二进制数%----%将得到的二进制数反序排列%----%将反序后的二进制数转化为十进制数,加一%

例:

极化码的matlab仿真(2)——编码

我们通过 BN 的递推式可以求得 BN再将其与 F n 次克劳尼克积相乘,就能够得到生成矩阵。

这样说你明白了吗?什么?没有???

不错,说明你是个正常人。实际编码的时候,如果真的在matlab中将上述过程模拟一遍来求生成矩阵的话,是非常耗时耗力的。我们还有另外一条路可以走——找规律。

极化码的matlab仿真(2)——编码

上式为 GN 的原始求解方法,其中 I 为单位矩阵,F 定义为[1 0; 1 1],RN 为翻转矩阵,对于 极化码的matlab仿真(2)——编码,有:

N=4:

极化码的matlab仿真(2)——编码

N=8:

极化码的matlab仿真(2)——编码

可以发现 A 的元素排列十分有规律。前 N/2 列元素总是奇偶成对出现“1”,后 N/2 元素仅出现在偶数位,且与同处在一行的前一个“1”保持固定距离。根据此规律编写程序如下:

for i = 1 : N/2
A(2 * i - 1, i) = 1;
A(2 * i, i) = 1;
A(2 * i, N / 2 + i) = 1;
end

递归部分:

G = A * kron(eye(2),Gpre);    %Gpre即上一层递归所得生成矩阵
Gpre = G;

如果将生成矩阵的产生编写为一个函数,则代码如下:

function GN = G(n)
N = 2 ^ n;
Gpre = 1;
for i=1:n       %每一层递归都相当于计算一个新的生成矩阵
Ni = 2 ^ i;    %这个新的生成矩阵的维度为 Ni/2
G = zeros(Ni);
%Fn = zeros(Ni);
A = zeros(Ni);
for j = 1 : Ni / 2
A(2 * j - 1 , j) = 1;
A(2 * j , j) = 1;
A(2 * j , Ni / 2 + j) = 1;
end
G = A*kron(eye(2),Gpre);
Gpre = G;
end
GN = G;
end
  • 其次,来看信息位与冻结位的选取

信道极化过程中,有一部分信道的信道容量 I(W) 可以到达1,另一部分则趋近于0。信道容量反映了信道无失真传输的最大信息率,我们可以通过计算联合、分裂后各信道的信道容量并对它们进行排序,然后根据码率,选择排序靠前的信道作为信息传输的信道,剩余的信道用来传输冻结位。

另一种方法是计算巴氏参数Z(W),对于一个给定信道,巴氏参数越大说明该信道越不可靠。因此我们只需计算出联合、分裂后信道的巴氏参数,并对它们进行排序,然后根据码率选择巴氏参数较小的信道作为信息位,剩余信道作为冻结位。Arikan论文中给出了巴氏参数的递归求解办法,这使得我们能够很方便的通过matlab实现信息位的选取。

极化码的matlab仿真(2)——编码

% 将巴氏参数计算过程封装在函数B_para之中方便调用,Z为数组,作为实参传递进来。数组中只有第一个元素。
% Z第一个元素可以通过计算得到,计算公式为Z(1) = 2*(p*(1-p))^0.5;
function y = B_para(Z)
for i = 1 : log2(N) %迭代次数,N为码长
Z_pre = Z; %z_pre为上一层信道巴氏参数
for j = 1 : 2^(i-1) %本层运算使用的下标
Z(2*j-1) = 2*Z_pre(j) - Z_pre(j)^2;
Z(2*j) = Z_pre(j)^2; %递推公式
end
end  
y = z;    % y作为实参从函数中传递出去,y就是最终的巴氏参数

得到巴氏参数序列后,下面的操作就是将此序列进行排序,并根据码率确定信息位和固定位。

[Z_in_order,index] = sort( y );            %将巴氏参数从小到大排列
signal_index = sort( index( 1:S ) );   %前S位作为信息位
frozen_index = sort( index( s+1:end ) ); %后面的作为冻结位

得到了生成矩阵,确定了信息位冻结位,下面要做的就是进行编码。

for j=1:block    %对每一个码块都要进行编码处理
encoded(1,((j-1)*N+1):(j*N)) = signal(((j-1)*S+1):(j*S))*G(signal_index,:) + frozen(((j-1)*F+1):(j*F))*G(frozen_index,:);
end                  %进行编码
encode = mod(encode,2);  %对2取模
encode = 2 * encode - 1;     %符号化
encode = encode + noise; %叠加噪声

数组 encode 就是我们得到的编码矩阵。

编码matlab实现就是这样,为了照顾和知识点讲解同步,各个部分代码并未进行完整、严谨的书写。本系列最后我会整合与各章节中分散的代码,为大家带来可运行的代码,并将M文件附在文后。敬请期待。

下一节我们要探讨的是polar code中非常重要的译码部分——连续消除译码(SC译码)。