Efficient representation of large-alphabet probability distributions

A Adler, J Tang, Y Polyanskiy - IEEE Journal on Selected Areas …, 2022 - ieeexplore.ieee.org
A Adler, J Tang, Y Polyanskiy
IEEE Journal on Selected Areas in Information Theory, 2022ieeexplore.ieee.org
A number of engineering and scientific problems require representing and manipulating
probability distributions over large alphabets, which we may think of as long vectors of reals
summing to 1. In some cases it is required to represent such a vector with only bits per entry.
A natural choice is to partition the interval into uniform bins and quantize entries to each bin
independently. We show that a minor modification of this procedure–applying an entrywise
non-linear function (compander) prior to quantization–yields an extremely effective …
A number of engineering and scientific problems require representing and manipulating probability distributions over large alphabets, which we may think of as long vectors of reals summing to 1. In some cases it is required to represent such a vector with only bits per entry. A natural choice is to partition the interval into uniform bins and quantize entries to each bin independently. We show that a minor modification of this procedure– applying an entrywise non-linear function (compander) prior to quantization– yields an extremely effective quantization method. For example, for and -sized alphabets, the quality of representation improves from a loss (under KL divergence) of bits/entry to bits/entry. Compared to floating point representations, our compander method improves the loss from to bits/entry. These numbers hold for both real-world data (word frequencies in books and DNA -mer counts) and for synthetic randomly generated distributions. Theoretically, we analyze a minimax optimality criterion and show that the closed-form compander is (asymptotically as ) optimal for quantizing probability distributions over a -letter alphabet. Non-asymptotically, such a compander (substituting for for simplicity) has KL-quantization loss bounded by . Interestingly, a similar minimax criterion for the quadratic loss on the hypercube shows optimality of the standard uniform quantizer. This suggests that the ArcSinh quantizer is as fundamental for KL-distortion as the uniform quantizer for quadratic distortion.
ieeexplore.ieee.org