An Extension of Enumerative Sphere Shaping for Arbitrary Channel Input Distributions
Frederik Ritter, Andrej Rode, Laurent Schmalen
TL;DR
This work extends Enumerative Sphere Shaping to support arbitrary discrete channel input distributions by introducing distribution-dependent weights in a trellis framework, enabling non-uniform shaping on discrete constellations. By selecting weights via a divergence-optimal approach (w^{(k)} = -log P_A(a^{(k)})), the method minimizes the informational divergence to the target input distribution, improving the mutual information bound. Simulation results on a non-Gaussian optical-channel model show a measurable rate gain over CCDM and the ability to adapt shaping via the trellis' maximum weight level, offering practical rate adaptation benefits. The combination of generalized ESS with probabilistic amplitude shaping demonstrates robust performance improvements and provides an open-source implementation for broader adoption.
Abstract
A non-uniform channel input distribution is key for achieving the capacity of arbitrary channels. However, message bits are generally assumed to follow a uniform distribution which must first be transformed to a non-uniform distribution by using a distribution matching algorithm. One such algorithm is enumerative sphere shaping (ESS). Compared to algorithms such as constant composition distribution matching (CCDM), ESS can utilize more channel input symbol sequences, allowing it to achieve a comparably low rate loss. However, the distribution of channel input symbols produced by ESS is fixed, restricting the utility of ESS to channels with Gaussian-like capacity-achieving input distributions. In this paper, we generalize ESS to produce arbitrary discrete channel input distributions, making it usable on most channels. Crucially, our generalization replaces fixed weights used internally by ESS with weights depending on the desired channel input distribution. We present numerical simulations using generalized ESS with probabilistic amplitude shaping (PAS) to transmit sequences of 256 symbols over a simplified model of an unamplified coherent optical link, a channel with a distinctly non-Gaussian capacity-achieving input distribution. In these simulations, we found that generalized ESS improves the maximum transmission rate by 0.0425 bit/symbol at a frame error rate below 10^{-4} compared to CCDM.
