Reinforcement Learning for Channel Coding: Learned Bit-Flipping Decoding

In this paper, we use reinforcement learning to find effective decoding strategies for binary linear codes. We start by reviewing several iterative decoding algorithms that involve a decision-making process at each step, including bit-flipping (BF) decoding, residual belief propagation, and anchor decoding. We then illustrate how such algorithms can be mapped to Markov decision processes allowing for data-driven learning of optimal decision strategies, rather than basing decisions on heuristics or intuition. As a case study, we consider BF decoding for both the binary symmetric and additive white Gaussian noise channel. Our results show that learned BF decoders can offer a range of performance-complexity trade-offs for the considered Reed-Muller and BCH codes, and achieve near-optimal performance in some cases. We also demonstrate learning convergence speed-ups when biasing the learning process towards correct decoding decisions, as opposed to relying only on random explorations and past knowledge.

[1]  Stephan ten Brink,et al.  On deep learning-based channel decoding , 2017, 2017 51st Annual Conference on Information Sciences and Systems (CISS).

[2]  Rong Li,et al.  Learning to Flip Successive Cancellation Decoding of Polar Codes with LSTM Networks , 2019, 2019 IEEE 30th Annual International Symposium on Personal, Indoor and Mobile Radio Communications (PIMRC).

[3]  Chris Watkins,et al.  Learning from delayed rewards , 1989 .

[4]  Krishna R. Narayanan,et al.  Iterative soft decoding of Reed-Solomon codes , 2004, IEEE Communications Letters.

[5]  L. G. Tallini,et al.  Neural nets for decoding error-correcting codes , 1995, IEEE Technical Applications Conference and Workshops. Northcon/95. Conference Record.

[6]  Jakob Hoydis,et al.  An Introduction to Deep Learning for the Physical Layer , 2017, IEEE Transactions on Cognitive Communications and Networking.

[7]  Yair Be'ery,et al.  Learning to decode linear codes using deep learning , 2016, 2016 54th Annual Allerton Conference on Communication, Control, and Computing (Allerton).

[8]  Ming Jiang,et al.  An improvement on the modified weighted bit flipping decoding algorithm for LDPC codes , 2005, IEEE Communications Letters.

[9]  Zhenyu Liu,et al.  A decoding algorithm for finite-geometry LDPC codes , 2005, IEEE Transactions on Communications.

[10]  Richard S. Sutton,et al.  Reinforcement Learning: An Introduction , 1998, IEEE Trans. Neural Networks.

[11]  Yoni Choukroun,et al.  Deep Learning for Decoding of Linear Codes - A Syndrome-Based Approach , 2018, 2018 IEEE International Symposium on Information Theory (ISIT).

[12]  Michael I. Jordan,et al.  RLlib: Abstractions for Distributed Reinforcement Learning , 2017, ICML.

[13]  Henry D. Pfister,et al.  Learned Belief-Propagation Decoding with Simple Scaling and SNR Adaptation , 2019, 2019 IEEE International Symposium on Information Theory (ISIT).

[14]  Henry D. Pfister,et al.  Approaching Miscorrection-Free Performance of Product Codes With Anchor Decoding , 2018, IEEE Transactions on Communications.

[15]  Henry D. Pfister,et al.  Decoding Reed-Muller Codes Using Minimum- Weight Parity Checks , 2018, 2018 IEEE International Symposium on Information Theory (ISIT).

[16]  O. Antoine,et al.  Theory of Error-correcting Codes , 2022 .

[17]  Shu Lin,et al.  Low-density parity-check codes based on finite geometries: A rediscovery and new results , 2001, IEEE Trans. Inf. Theory.

[18]  Shu Lin,et al.  Soft-decision decoding of linear block codes based on ordered statistics , 1994, IEEE Trans. Inf. Theory.

[19]  Shu Lin,et al.  Channel Codes: Classical and Modern , 2009 .

[20]  Ian McGraw,et al.  Residual Belief Propagation: Informed Scheduling for Asynchronous Message Passing , 2006, UAI.

[21]  Ying-Chang Liang,et al.  Applications of Deep Reinforcement Learning in Communications and Networking: A Survey , 2018, IEEE Communications Surveys & Tutorials.

[22]  Sreeram Kannan,et al.  Communication Algorithms via Deep Learning , 2018, ICLR.

[23]  Martin Bossert,et al.  Hard- and soft-decision decoding beyond the half minimum distance---An algorithm for linear codes , 1986, IEEE Trans. Inf. Theory.

[24]  Bin Chen,et al.  Decoding Staircase Codes with Marked Bits , 2018, 2018 IEEE 10th International Symposium on Turbo Codes & Iterative Information Processing (ISTC).

[25]  Marc P. C. Fossorier,et al.  A modified weighted bit-flipping decoding of low-density Parity-check codes , 2004, IEEE Communications Letters.

[26]  Keith M. Chugg,et al.  Random Redundant Soft-In Soft-Out Decoding of Linear Block Codes , 2006, 2006 IEEE International Symposium on Information Theory.

[27]  Ming Jiang,et al.  Improved weighted bit - fl ipping algorithm for decoding LDPC Codes , 2005 .

[28]  Ilya Dumer,et al.  Soft-decision decoding of Reed-Muller codes: recursive lists , 2006, IEEE Transactions on Information Theory.

[29]  Richard D. Wesel,et al.  LDPC Decoders with Informed Dynamic Scheduling , 2010, IEEE Transactions on Communications.