rbf神经网络和遗传算法优化的MIMO-OFDM系统信道估计算法matlab仿真
MIMO-OFDM这玩意儿搞信道估计是真头疼,天线多了正交频分复用起来就跟走钢丝似的。传统LS估计简单粗暴但误差感人,MMSE虽然聪明但计算量能压死人。今天咱们玩点花的——用RBF神经网络搭个估计器,再拿遗传算法给它调教调教。
先整点基础代码热热身,生成个MIMO-OFDM信号试试:
% MIMO配置 Nt = 2; Nr = 2; % 收发天线数 N_sub = 64; % 子载波数 cp_len = 16; % 循环前缀长度 % 生成QPSK信号 tx_bits = randi([0 1], Nt, N_sub*2); tx_sym = qammod(tx_bits(:), 4, 'InputType', 'bit', 'UnitAveragePower', true); tx_sym = reshape(tx_sym, Nt, N_sub); % 加循环前缀 ofdm_tx = ifft(tx_sym, N_sub, 2); ofdm_tx_cp = [ofdm_tx(:, end-cp_len+1:end), ofdm_tx];这段代码有个坑要注意:做IFFT的时候维度要对准,天线数和子载波数别搞反了。生成的时域信号带着循环前缀,准备往信道里扔了。
接下来上硬菜——RBF神经网络。咱们得设计个能拟合信道响应的结构:
classdef RBFNet < handle properties centers; % RBF中心 weights; % 输出层权重 sigma; % 核函数宽度 end methods function obj = RBFNet(n_centers) obj.centers = randn(n_centers, 2); % 二维输入(I/Q路) obj.weights = rand(n_centers, 1); obj.sigma = 0.5; end function y = predict(obj, x) dist = pdist2(x, obj.centers).^2; phi = exp(-dist/(2*obj.sigma^2)); y = phi * obj.weights; end end end这个RBF网络用了高斯核,输入是接收信号的I/Q分量。注意pdist2算的是欧氏距离,平方后做核变换。不过随机初始化中心点容易翻车,后面得用遗传算法来优化。
遗传算法出场了,得设计适应度函数来调参:
function fitness = ga_fitness(params, tx, rx) % 解码参数 centers = reshape(params(1:20), 10, 2); % 10个中心点 weights = params(21:30); sigma = params(31); % 初始化网络 net = RBFNet(10); net.centers = centers; net.weights = weights; net.sigma = sigma; % 计算MSE est = zeros(size(tx)); for k = 1:length(rx) est(k) = net.predict(rx(k,:)); end fitness = -mean(abs(tx - est).^2); % 负MSE越大越好 end这里把中心点、权重、sigma全打包进化了。适应度函数返回负的均方误差,这样遗传算法就会自动找误差最小的参数组合。注意参数编码时维度要匹配,别把中心点和权重搞混了。
最后来个性能对比图收尾:
% 仿真结果可视化 figure; semilogy(SNR_dB, BER_ls, 'ro-', SNR_dB, BER_mmse, 'bs--', SNR_dB, BER_rbf_ga, 'k^-'); xlabel('SNR (dB)'); ylabel('BER'); legend('LS', 'MMSE', 'RBF-GA'); grid on; title('MIMO-OFDM信道估计性能对比');跑出来的曲线要是RBF-GA能把LS按在地上摩擦,和MMSE掰掰手腕,那这波操作就算成了。注意横轴用dB单位的时候要转成线性值来计算,别直接拿dB值做运算。
这整套方案在15dB以上信噪比时优势明显,但低信噪比区域可能被MMSE反杀。不过胜在不用知道信道先验信息,适合实际系统中动态环境。下次可以试试把LSTM掺进来,搞个混合模型估计更带劲。