A Deterministic Algorithm for Computing the Weight Distribution of Polar Codes

TL;DR

This paper introduces a deterministic recursive algorithm to compute the entire weight distribution of polar codes efficiently, addressing an open problem and enabling practical calculations for codes of length 128.

Contribution

The authors develop a recursive procedure for weight distribution computation of polar codes and analyze complexity reductions using automorphism groups and code properties.

Findings

Recursive algorithm computes weight distribution in O(n^2) time.

Complexity reduction enables calculations for length n=128.

Self-dual Reed-Muller codes have the largest mixing factors among decreasing monomial codes.

Abstract

In this work, we present a deterministic algorithm for computing the entire weight distribution of polar codes. As the first step, we derive an efficient recursive procedure to compute the weight distribution that arises in successive cancellation decoding of polar codes along any decoding path. This solves the open problem recently posed by Polyanskaya, Davletshin, and Polyanskii. Using this recursive procedure, at code length n, we can compute the weight distribution of any polar cosets in time O(n^2). We show that any polar code can be represented as a disjoint union of such polar cosets; moreover, this representation extends to polar codes with dynamically frozen bits. However, the number of polar cosets in such representation scales exponentially with a parameter introduced herein, which we call the mixing factor. To upper bound the complexity of our algorithm for polar codes being decreasing monomial codes, we study the range of their mixing factors. We prove that among all decreasing monomial codes with rates at most 1/2, self-dual Reed-Muller codes have the largest mixing factors. To further reduce the complexity of our algorithm, we make use of the fact that, as decreasing monomial codes, polar codes have a large automorphism group. That automorphism group includes the block lower-triangular affine group (BLTA), which in turn contains the lower-triangular affine group (LTA). We prove that a subgroup of LTA acts transitively on certain subsets of decreasing monomial codes, thereby drastically reducing the number of polar cosets that we need to evaluate. This complexity reduction makes it possible to compute the weight distribution of polar codes at length n = 128.