Sequential-RNN-Decoder

Updated: 05/02/2018, PyTorch version under construction. Later will all merge to PyTorch.

This repository contains source code necessary to reproduce the results presented in the following paper: Communication Algorithms via Deep Learning(https://openreview.net/pdf?id=ryazCMbR-) by Hyeji Kim, Yihan Jiang, Ranvir B. Rana, Sreeram Kannan, Sewoong Oh, Pramod Viswanath, accepted to ICLR 2018 as poster.

Dependency

Python (2.7.10+)
numpy (1.14.1)
Keras (2.0)
scikit-commpy (0.3.0) For Commpy, we use a modified version of the original commpy, which is in the folder with name commpy. You don't need to install commpy via pip. The original commpy has a few bugs which is fixed in our version.
h5py (2.7.0)
tensorflow (1.5)

Benchmarks

We have benchmarks for evaluating BER/BLER for convolutional code, turbo code. The curves from paper are from MATLAB simulation, the python curve is for reference. We find the python and MATLAB implementation has same performance. When running large number of blocks (>1000), you might need to use multiprocess to speed up simulation, change -num_cpu to the number you like.

To evaluate BER/BLER for convolutional code, by default the codec is rate 1/2 (7,5) convolutioanl code with feedback = 7. (5 means f(x) = 1 + x^2, 7 means f(x) = 1 + x + x^2):

$ python conv_codes_benchmark.py -num_block 100 -block_len 100 -snr_test_start -1.5 -snr_test_end 2.0 -snr_points 8 -num_cpu 1

To evaluate BER/BLER for turbo code:.

$ python turbo_codes_benchmark.py -num_block 100 -block_len 100 -snr_test_start -1.5 -snr_test_end 2.0 -snr_points 8 -num_cpu 1

You can change to LTE turbo codec by

$ python turbo_codes_benchmark.py -enc1 11 -enc2 13 -M 3 -feedback 11 -num_block 100 -block_len 100 -snr_test_start -1.5 -snr_test_end 2.0 -snr_points 8 -num_cpu 1

By default the number of decoding iteration is 6, you can change via change argument '-num_dec_iteration'.

RNN for Convolutioanl Code

use conv_decoder.py to train Viterbi-like RNNs.
use bcjr_rnn_train.py to train BCJR-like RNNs.

Neural Turbo Decoder

Note: Currently debugging non-AWGN decoders. Recommend to run on GPU.

To evaluate Neural Turbo Decoder run default setting by: python turbo_neural_decoder_eval.py -h, to sepecify the parameters for testing. By default use the following:

$ python turbo_neural_decoder_eval.py -num_block 100 -block_len 100 -snr_test_start -1.5 -snr_test_end 2.0 -snr_points 8 -model_path ./models/turbo_models/awgn_bl100_1014.h5
To train Neural Turbo Decoder: python turbo_neural_decoder_train.py -h, to specify the parameters for training. Not using GPU will take long time (48+ hours) for training.

$ python turbo_neural_decoder_train.py -num_block 1000 -block_len 100 -train_snr -1.5 -init_nw_model ./models/turbo_models/awgn_bl100_1014.h5

Interpreting the RNN

Under construction.

Questions?

Please email Yihan Jiang ([email protected]) for any questions.

songzi2013 / sequential-rnn-decoder Goto Github PK