Työväentutkimus Vuosikirja
5G Polar Codes: ML-Assisted Decoding Performance Comparison on GPUs
Anton Bogun
Abstract
Polar code decoding is an algorithmic challenge that is highly sensitive to the latency and throughput. It is used in 5G NR control channels and demands good noise decoding perfor- mance. Decoders such as successive cancellation (SC) and its list variant (SCL) achieve good bit error rates (BER) throughout the signal-to-noise ratio (SNR) regimes, but suffer from high latency and challenges in parallel implementation due to their sequential nature. The belief propagation (BP) decoder is an alternative representation of the polar graph, which allows parallel decoding over the whole codeword, but requires multiple iterations to reach acceptable BER. The hardware-friendly min-sum (MS) algorithmic simplification uses cheap instructions instead of expensive transcendental operations for higher throughput and lower latency, but further negatively impacts the noise decoding performance. This thesis considers the normalized variation of min-sum (NMS), which adds trainable scales per edge in the BP graph, applied to the 5G N=128 rate 1/2 polar code. The weights are trained by treating the unrolled 5 iteration BP graph as a neural network, which achieves a better-or- equal noise performance compared to the true BP decoder, a 0.5 dB average improvement over the untrained MS, and a 5x lower BER than true BP at 6 dB SNR, making the performance roughly comparable to SC in high SNR regions. These improvements are implemented as a single optimized GPU kernel with further optimiza- tions for better throughput, and run on an NVIDIA H100 GPU. The final achieved throughput peaks at 181 million codewords per second (11.6 information Gbit/s), which is around 25x the throughput of the XLA-compiled Sionna framework running the same decoder, and around 190x of a 192-thread CPU implementation.