Rate Compatible LDPC Neural Decoding Network: A Multi-Task Learning Approach

Yukun Cheng,Wei Chen,Lun Li,Bo Ai
2023-10-10
Abstract:Deep learning based decoding networks have shown significant improvement in decoding LDPC codes, but the neural decoders are limited by rate-matching operations such as puncturing or extending, thus needing to train multiple decoders with different code rates for a variety of channel conditions. In this correspondence, we propose a Multi-Task Learning based rate-compatible LDPC ecoding network, which utilizes the structure of raptor-like LDPC codes and can deal with multiple code rates. In the proposed network, different portions of parameters are activated to deal with distinct code rates, which leads to parameter sharing among tasks. Numerical experiments demonstrate the effectiveness of the proposed method. Training the specially designed network under multiple code rates makes the decoder compatible with multiple code rates without sacrificing frame error rate performance.
Signal Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the limitations of existing deep - learning - based decoders in handling codewords with different code rates in low - density parity - check code (LDPC) decoding. Specifically, traditional neural network decoders need to train multiple models for different code rates, which not only increases the storage cost but also raises the training complexity. In addition, these decoders perform poorly when faced with code - rate - matching operations (such as puncturing or spreading), limiting their flexibility in practical applications, especially in resource - constrained application scenarios, such as the Internet of Things (IoT). To overcome these problems, the paper proposes a rate - compatible (RC) LDPC neural decoding network based on multi - task learning (MTL). This network utilizes an LDPC code structure similar to Raptor codes and can dynamically adjust the network structure at different code rates through a parameter - sharing mechanism, thereby achieving support for multiple code rates. This method not only reduces storage costs and training complexity but also maintains frame error rate (FER) performance comparable to that of traditional decoding algorithms. The main contributions of the paper include: 1. **Proposing a new rate - compatible neural decoding network**: This network can adapt to multiple code rates without the need to train a separate model for each code rate. 2. **Utilizing the parameter - sharing mechanism**: By activating different neurons at different code rates, parameter sharing is achieved, reducing the storage requirements. 3. **Experimentally verifying the effectiveness of the method**: Through experiments on 5G NR LDPC codes, it is proven that the proposed RC neural decoder outperforms traditional decoding algorithms at different code rates and has performance comparable to that of separately trained neural decoders. In conclusion, this paper aims to develop an efficient LDPC neural decoding network that can adapt to multiple code rates in order to reduce storage and training costs while maintaining good decoding performance.