A Novel Decoding Scheme for Polar Code Using Convolutional Neural Network

Polar code is the first capacity-achieving channel coding scheme which has been selected for the next generation of wireless communication standard(5G). However, the traditional decoding schemes such as CA-SCL and BP algorithms cannot take into account both BER and latency performance. Inspired by the deep learning technology, we propose a scheme that combine convolutional neural network (CNN) with traditional Belief Propagation (BP) decoding. By exploiting noise correlation for channel decoding under colored noise environment, the proposed scheme can obtain a more accurate estimation of channel noise. We simulate the performance of the proposed scheme under BPSK and 4QAM modulation. For QAM modulation, a scheme that extend the conventional real-valued CNN technique to the complex-valued one is given. The results demonstrate that bit error rates (BER) performance can be improved after removing the estimated noise. Compare to the traditional Polar decoding schemes, the proposed scheme has better trade-off between BER and latency performance under colored noise environment.