Learning to Decode Linear Codes Using Deep Learning

Code

• github.com/nvlabs/gnn-decoder
  tf★ 68
• github.com/tomerraviv95/bayesian-learning-in-receivers-for-decoding
  pytorch★ 2
• github.com/NVlabs/sionna/blob/main/examples/Weighted_BP_Algorithm.ipynb
  tf★ 0

Abstract

A novel deep learning method for improving the belief propagation algorithm is proposed. The method generalizes the standard belief propagation algorithm by assigning weights to the edges of the Tanner graph. These edges are then trained using deep learning techniques. A well-known property of the belief propagation algorithm is the independence of the performance on the transmitted codeword. A crucial property of our new method is that our decoder preserved this property. Furthermore, this property allows us to learn only a single codeword instead of exponential number of code-words. Improvements over the belief propagation algorithm are demonstrated for various high density parity check codes.

Tasks

Reproductions