silas/imdd_silas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Few more scripts to evaluate new ML-MLSE Equalizer - which is not better!!! :-(

Browse Source
This commit is contained in:
silas (home)
2025-11-03 08:17:39 +01:00
parent 09f345aa91
commit 39bc8243fc
7 changed files with 443 additions and 319 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
projects/ML_based_MLSE/model.m
View File

@@ -18,13 +18,12 @@ laser_linewidth = 1e6;
% Channel
link_length = 0;
alpha = 0;
doub_mode = db_mode.no_db;
cols = linspecer(6);
rop = [-5];
rop = [-8];
bwl = [0.5:0.1:1.5];
fsym = [212:16:256].*1e9;
fsym = [160:16:256].*1e9;
% nonlin_mod = [0.5:0.01:0.75];
nonlin_mod = ones(size(fsym)).*0.5;
@@ -100,6 +99,7 @@ for r = 1:length(fsym)
%%
mu_lms = 0.0005;
pf_ncoeffs = 2;
eq_ = FFE("epochs_tr",5,"epochs_dd",5,"len_tr",2^13,"mu_dd",mu_lms,"mu_tr",mu_lms,"order",50,"sps",2,"dd_mode",1,"adaption_technique","lms");
pf_ = Postfilter("ncoeff",pf_ncoeffs,"useBurg",1);
@@ -116,49 +116,63 @@ for r = 1:length(fsym)
% Sequence Est
mlse_ = MLSE("duobinary_output",0,'M',M,'trellis_states',PAMmapper(M,0).levels,'scale_mode',0,'trellis_exclusion',0,'trellis_state_mode',2,'debug',0,'DIR',pf_.coefficients);
[y_mlse] = mlse_.process(y_white,Symbols);
Vit_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_mlse);
[~, errors, ber_mlse_normal, errpos] = calc_ber(Vit_bits.signal, Tx_bits.signal, "skip_front", 10, "skip_end", 10, "returnErrorLocation", 1);
mlse_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_mlse);
[~, errors, ber_mlse_normal, errpos] = calc_ber(mlse_bits.signal, Tx_bits.signal, "skip_front", 10, "skip_end", 10, "returnErrorLocation", 1);
fprintf('MLSE: %.2e \n',ber_mlse_normal);
showLevelHistogram(y_ffe,Symbols,"displayname",'ffe','fignum',111);
% showLevelHistogram(y_white,Symbols,"displayname",'ffe','fignum',111);
%% optimize length
mu_lms = 0.15;
eq = ML_MLSE("epochs_tr",5,"epochs_dd",5,"len_tr",2^14,...
"mu_dd",mu_lms,"mu_tr",mu_lms,"order",5,"sps",2,...
"traceback_depth",128,"L",2,"delta",0);
[y_ml_mlse,Vit_signal] = eq.process(Rx_sig_2sps,Symbols);
y_ml_mlse_ = y_ml_mlse;
y_ml_mlse_.signal = circshift(y_ml_mlse.signal,0);
Vit_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_ml_mlse_);
[~, errors, ber, errpos] = calc_ber(Vit_bits.signal, Tx_bits.signal, "skip_front", 10, "skip_end", 10, "returnErrorLocation", 1);
fprintf('ML MLSE: %.2e \n',ber);
bursts = count_error_bursts(errpos, 10);
e = zeros(size(Vit_bits.signal));
e = zeros(size(mlse_bits.signal));
e(errpos) = 1;
figure(8)
stem(e)
%% RUN ML-Based MLSE
mu_lms = 0.15;
ml_mlse_equalizer = ML_MLSE("epochs_tr",50,"epochs_dd",10,"len_tr",Rx_sig_2sps.length-100,...
"mu_dd",mu_lms,"mu_tr",mu_lms,"order",15,"sps",2,...
"traceback_depth",128,"L",3,"delta",5);
%%
ml_mlse_equalizer.epochs_tr = 50;
ml_mlse_equalizer.epochs_dd = 1;
[y_ml_mlse,Vit_signal] = ml_mlse_equalizer.process(Rx_sig_2sps,Symbols);
ml_mlse_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_ml_mlse);
[~, errors, ber, errpos] = calc_ber(ml_mlse_bits.signal, Tx_bits.signal, "skip_front", 10, "skip_end", 10, "returnErrorLocation", 1);
fprintf('ML MLSE BER: %.2e \n',ber);
bursts = count_error_bursts(errpos, 10);
e = zeros(size(ml_mlse_bits.signal));
e(errpos) = 1;
figure(8)
stem(e)
figure()
plot(ml_mlse_equalizer.ber)
beautifyBERplot
%% optimize delta
deltas = [-1:4];
ber = zeros(1,length(deltas));
parfor m = 1:numel(deltas)
mu_lms = 0.2;
eq = ML_MLSE("epochs_tr",2,"epochs_dd",5,"len_tr",2^13,...
ml_mlse_equalizer = ML_MLSE("epochs_tr",2,"epochs_dd",5,"len_tr",2^13,...
"mu_dd",mu_lms,"mu_tr",mu_lms,"order",4,"sps",1,...
"traceback_depth",128,"L",3,"delta",deltas(m));
[y_ml_mlse,Vit_signal] = eq.process(y_ffe,Symbols);
[y_ml_mlse,Vit_signal] = ml_mlse_equalizer.process(y_ffe,Symbols);
y_ml_mlse_ = y_ml_mlse;
y_ml_mlse_.signal = circshift(y_ml_mlse.signal,0);
Vit_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_ml_mlse_);
[~, errors, ber(m), errpos] = calc_ber(Vit_bits.signal, Tx_bits.signal, "skip_front", 0, "skip_end", 0, "returnErrorLocation", 1);
mlse_bits = PAMmapper(M, 0, "eth_style", 0).demap(y_ml_mlse_);
[~, errors, ber(m), errpos] = calc_ber(mlse_bits.signal, Tx_bits.signal, "skip_front", 0, "skip_end", 0, "returnErrorLocation", 1);
fprintf('ML MLSE: %.2e \n',ber(m));
end