Asymmetric numeral systems

From HandWiki
Short description: Entropy coding methods

Asymmetric numeral systems (ANS)[1][2] is a family of entropy encoding methods introduced by Jarosław (Jarek) Duda[3] from Jagiellonian University, used in data compression since 2014[4] due to improved performance compared to previous methods.[5] ANS combines the compression ratio of arithmetic coding (which uses a nearly accurate probability distribution), with a processing cost similar to that of Huffman coding. In the tabled ANS (tANS) variant, this is achieved by constructing a finite-state machine to operate on a large alphabet without using multiplication.

Among others, ANS is used in the Facebook Zstandard compressor[6][7] (also used e.g. in Linux kernel,[8] Android[9] operating system, was published as RFC 8478 for MIME[10] and HTTP[11]), Apple LZFSE compressor,[12] Google Draco 3D compressor[13] (used e.g. in Pixar Universal Scene Description format[14]) and PIK image compressor,[15] CRAM DNA compressor[16] from SAMtools utilities,[17] Dropbox DivANS compressor,[18] Microsoft DirectStorage BCPack texture compressor,[19] and JPEG XL[20] image compressor.

The basic idea is to encode information into a single natural number [math]\displaystyle{ x }[/math]. In the standard binary number system, we can add a bit [math]\displaystyle{ s \in \{0, 1\} }[/math] of information to [math]\displaystyle{ x }[/math] by appending [math]\displaystyle{ s }[/math] at the end of [math]\displaystyle{ x }[/math], which gives us [math]\displaystyle{ x' = 2x + s }[/math]. For an entropy coder, this is optimal if [math]\displaystyle{ \Pr(0) = \Pr(1) = 1/2 }[/math]. ANS generalizes this process for arbitrary sets of symbols [math]\displaystyle{ s \in S }[/math] with an accompanying probability distribution [math]\displaystyle{ (p_s)_{s \in S} }[/math]. In ANS, if the information from [math]\displaystyle{ s }[/math] is appended to [math]\displaystyle{ x }[/math] to result in [math]\displaystyle{ x' }[/math], then [math]\displaystyle{ x' \approx x \cdot p_s^{-1} }[/math]. Equivalently, [math]\displaystyle{ \log_2(x') \approx \log_2(x) + \log_2(1/p_s) }[/math], where [math]\displaystyle{ \log_2(x) }[/math] is the number of bits of information stored in the number [math]\displaystyle{ x }[/math], and [math]\displaystyle{ \log_2(1/p_s) }[/math] is the number of bits contained in the symbol [math]\displaystyle{ s }[/math].

For the encoding rule, the set of natural numbers is split into disjoint subsets corresponding to different symbols – like into even and odd numbers, but with densities corresponding to the probability distribution of the symbols to encode. Then to add information from symbol [math]\displaystyle{ s }[/math] into the information already stored in the current number [math]\displaystyle{ x }[/math], we go to number [math]\displaystyle{ x' = C(x, s) \approx x/p }[/math] being the position of the [math]\displaystyle{ x }[/math]-th appearance from the [math]\displaystyle{ s }[/math]-th subset.

There are alternative ways to apply it in practice – direct mathematical formulas for encoding and decoding steps (uABS and rANS variants), or one can put the entire behavior into a table (tANS variant). Renormalization is used to prevent [math]\displaystyle{ x }[/math] going to infinity – transferring accumulated bits to or from the bitstream.

Entropy coding

Suppose a sequence of 1,000 zeros and ones would be encoded, which would take 1000 bits to store directly. However, if it is somehow known that it only contains 1 zero and 999 ones, it would be sufficient to encode the zero's position, which requires only [math]\displaystyle{ \lceil \log_2(1000)\rceil \approx 10 }[/math] bits here instead of the original 1000 bits.

Generally, such length [math]\displaystyle{ n }[/math] sequences containing [math]\displaystyle{ pn }[/math] zeros and [math]\displaystyle{ (1-p)n }[/math] ones, for some probability [math]\displaystyle{ p\in(0,1) }[/math], are called combinations. Using Stirling's approximation we get their asymptotic number being

[math]\displaystyle{ {n \choose pn }\approx 2^{nh(p)} \text{ for large } n \text{ and } h(p) =-p \log_2(p)-(1-p)\log_2(1-p), }[/math]

called Shannon entropy.

Hence, to choose one such sequence we need approximately [math]\displaystyle{ n h(p) }[/math] bits. It is still [math]\displaystyle{ n }[/math] bits if [math]\displaystyle{ p=1/2 }[/math], however, it can also be much smaller. For example, we need only [math]\displaystyle{ \approx n/2 }[/math] bits for [math]\displaystyle{ p = 0.11 }[/math].

An entropy coder allows the encoding of a sequence of symbols using approximately the Shannon entropy bits per symbol. For example, ANS could be directly used to enumerate combinations: assign a different natural number to every sequence of symbols having fixed proportions in a nearly optimal way.

In contrast to encoding combinations, this probability distribution usually varies in data compressors. For this purpose, Shannon entropy can be seen as a weighted average: a symbol of probability [math]\displaystyle{ p }[/math] contains [math]\displaystyle{ \log_2(1/p) }[/math] bits of information. ANS encodes information into a single natural number [math]\displaystyle{ x }[/math], interpreted as containing [math]\displaystyle{ \log_2(x) }[/math] bits of information. Adding information from a symbol of probability [math]\displaystyle{ p }[/math] increases this informational content to [math]\displaystyle{ \log_2(x)+\log_2(1/p)=\log_2(x/p) }[/math]. Hence, the new number containing both information should be [math]\displaystyle{ x'\approx x/p }[/math].

Motivating examples

Consider a source with 3 letters A, B, C, with probability 1/2, 1/4, 1/4. It is simple to construct the optimal prefix code in binary: A = 0, B = 10, C = 11. Then, a message is encoded as ABC -> 01011.

We see that an equivalent method for performing the encoding is as follows:

  • Start with number 1, and perform an operation on the number for each input letter.
  • A = multiply by 2; B = multiply by 4, add 2; C = multiply by 4, add 3.
  • Express the number in binary, then remove the first digit 1.

Consider a more general source with k letters, with rational probabilities [math]\displaystyle{ n_1/N, ..., n_k/N }[/math]. Then performing arithmetic coding on the source requires only exact arithmetic with integers.

In general, ANS is an approximation of arithmetic coding that approximates the real probabilities [math]\displaystyle{ r_1, ..., r_k }[/math] by rational numbers [math]\displaystyle{ n_1/N, ..., n_k/N }[/math] with a small denominator [math]\displaystyle{ N }[/math].

Basic concepts of ANS

Comparison of the concept of arithmetic coding (left) and ANS (right). Both can be seen as generalizations of standard numeral systems, optimal for uniform probability distribution of digits, into optimized for some chosen probability distribution. Arithmetic or range coding corresponds to adding new information in the most significant position, while ANS generalizes adding information in the least significant position. Its coding rule is "x goes to x-th appearance of subset of natural numbers corresponding to currently encoded symbol". In the presented example, sequence (01111) is encoded into a natural number 18, which is smaller than 47 obtained by using standard binary system, due to better agreement with frequencies of sequence to encode. The advantage of ANS is storing information in a single natural number, in contrast to two defining a range.

Imagine there is some information stored in a natural number [math]\displaystyle{ x }[/math], for example as bit sequence of its binary expansion. To add information from a binary variable [math]\displaystyle{ s }[/math], we can use coding function [math]\displaystyle{ x'=C(x,s) = 2x + s }[/math], which shifts all bits one position up, and place the new bit in the least significant position. Now decoding function [math]\displaystyle{ D(x')=(\lfloor x'/2 \rfloor, \mathrm{mod}(x',2)) }[/math] allows one to retrieve the previous [math]\displaystyle{ x }[/math] and this added bit: [math]\displaystyle{ D(C(x,s))=(x,s),\ C(D(x'))=x' }[/math]. We can start with [math]\displaystyle{ x=1 }[/math] initial state, then use the [math]\displaystyle{ C }[/math] function on the successive bits of a finite bit sequence to obtain a final [math]\displaystyle{ x }[/math] number storing this entire sequence. Then using [math]\displaystyle{ D }[/math] function multiple times until [math]\displaystyle{ x=1 }[/math] allows one to retrieve the bit sequence in reversed order.

The above procedure is optimal for the uniform (symmetric) probability distribution of symbols [math]\displaystyle{ \Pr(0)=\Pr(1)=1/2 }[/math]. ANS generalize it to make it optimal for any chosen (asymmetric) probability distribution of symbols: [math]\displaystyle{ \Pr(s)=p_s }[/math]. While [math]\displaystyle{ s }[/math] in the above example was choosing between even and odd [math]\displaystyle{ C(x,s) }[/math], in ANS this even/odd division of natural numbers is replaced with division into subsets having densities corresponding to the assumed probability distribution [math]\displaystyle{ \{p_s\}_s }[/math]: up to position [math]\displaystyle{ x }[/math], there are approximately [math]\displaystyle{ x p_s }[/math] occurrences of symbol [math]\displaystyle{ s }[/math].

The coding function [math]\displaystyle{ C(x,s) }[/math] returns the [math]\displaystyle{ x }[/math]-th appearance from such subset corresponding to symbol [math]\displaystyle{ s }[/math]. The density assumption is equivalent to condition [math]\displaystyle{ x'=C(x,s) \approx x / p_s }[/math]. Assuming that a natural number [math]\displaystyle{ x }[/math] contains [math]\displaystyle{ \log_2(x) }[/math] bits of information, [math]\displaystyle{ \log_2(C(x,s)) \approx \log_2(x) + \log_2(1/ p_s) }[/math]. Hence the symbol of probability [math]\displaystyle{ p_s }[/math] is encoded as containing [math]\displaystyle{ \approx\log_2(1/ p_s) }[/math] bits of information as it is required from entropy coders.

Variants

Uniform binary variant (uABS)

Let us start with the binary alphabet and a probability distribution [math]\displaystyle{ \Pr(1) = p }[/math], [math]\displaystyle{ \Pr(0)=1-p }[/math]. Up to position [math]\displaystyle{ x }[/math] we want approximately [math]\displaystyle{ p\cdot x }[/math] analogues of odd numbers (for [math]\displaystyle{ s = 1 }[/math]). We can choose this number of appearances as [math]\displaystyle{ \lceil x\cdot p \rceil }[/math], getting [math]\displaystyle{ s = \lceil (x+1)\cdot p \rceil - \lceil x\cdot p \rceil }[/math]. This variant is called uABS and leads to the following decoding and encoding functions:[21]

Decoding:

s = ceil((x+1)*p) - ceil(x*p)  // 0 if fract(x*p) < 1-p, else 1
if s = 0 then new_x = x - ceil(x*p)   // D(x) = (new_x, 0), this is the same as new_x = floor(x*(1-p))
if s = 1 then new_x = ceil(x*p)  // D(x) = (new_x, 1)

Encoding:

if s = 0 then new_x = ceil((x+1)/(1-p)) - 1 // C(x,0) = new_x
if s = 1 then new_x = floor(x/p)  // C(x,1) = new_x

For [math]\displaystyle{ p=1/2 }[/math] it amounts to the standard binary system (with 0 and 1 inverted), for a different [math]\displaystyle{ p }[/math] it becomes optimal for this given probability distribution. For example, for [math]\displaystyle{ p=0.3 }[/math] these formulas lead to a table for small values of [math]\displaystyle{ x }[/math]:

[math]\displaystyle{ C(x,s) }[/math] 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
[math]\displaystyle{ s=0 }[/math] 0 1 2 3 4 5 6 7 8 9 10 11 12 13
[math]\displaystyle{ s=1 }[/math] 0 1 2 3 4 5 6

The symbol [math]\displaystyle{ s=1 }[/math] corresponds to a subset of natural numbers with density [math]\displaystyle{ p=0.3 }[/math], which in this case are positions [math]\displaystyle{ \{0,3,6,10,13,16,20,23,26,\ldots\} }[/math]. As [math]\displaystyle{ 1/4\lt 0.3 \lt 1/3 }[/math], these positions increase by 3 or 4. Because [math]\displaystyle{ p=3/10 }[/math] here, the pattern of symbols repeats every 10 positions.

The coding [math]\displaystyle{ C(x,s) }[/math] can be found by taking the row corresponding to a given symbol [math]\displaystyle{ s }[/math], and choosing the given [math]\displaystyle{ x }[/math] in this row. Then the top row provides [math]\displaystyle{ C(x,s) }[/math]. For example, [math]\displaystyle{ C(7,0)=11 }[/math] from the middle to the top row.

Imagine we would like to encode the sequence '0100' starting from [math]\displaystyle{ x=1 }[/math]. First [math]\displaystyle{ s=0 }[/math] takes us to [math]\displaystyle{ x=2 }[/math], then [math]\displaystyle{ s=1 }[/math] to [math]\displaystyle{ x=6 }[/math], then [math]\displaystyle{ s=0 }[/math] to [math]\displaystyle{ x=9 }[/math], then [math]\displaystyle{ s=0 }[/math] to [math]\displaystyle{ x=14 }[/math]. By using the decoding function [math]\displaystyle{ D(x') }[/math] on this final [math]\displaystyle{ x }[/math], we can retrieve the symbol sequence. Using the table for this purpose, [math]\displaystyle{ x }[/math] in the first row determines the column, then the non-empty row and the written value determine the corresponding [math]\displaystyle{ s }[/math] and [math]\displaystyle{ x }[/math].

Range variants (rANS) and streaming

The range variant also uses arithmetic formulas, but allows operation on a large alphabet. Intuitively, it divides the set of natural numbers into size [math]\displaystyle{ 2^n }[/math] ranges, and split each of them in identical way into subranges of proportions given by the assumed probability distribution.

We start with quantization of probability distribution to [math]\displaystyle{ 2^n }[/math] denominator, where n is chosen (usually 8-12 bits): [math]\displaystyle{ p_s \approx f[s]/2^n }[/math] for some natural [math]\displaystyle{ f[s] }[/math] numbers (sizes of subranges).

Denote [math]\displaystyle{ \text{mask} = 2^n-1 }[/math], and a cumulative distribution function:

[math]\displaystyle{ \operatorname{CDF}[s] = \sum_{i\lt s} f[i] =f[0]+ \cdots +f[s-1]. }[/math]

Note here that the CDF[s] function is not a true CDF in that the current symbol's probability is not included in the expression's value. Instead, the CDF[s] represents the total probability of all previous symbols. Example: Instead of the normal definition of CDF[0] = f[0], it is evaluated as CDF[0] = 0, since there are no previous symbols.

For [math]\displaystyle{ y\in[0,2^n-1] }[/math] denote function (usually tabled)

symbol(y) = s  such that  CDF[s] <= y < CDF[s+1]

Now coding function is:

C(x,s) = (floor(x / f[s]) << n) + (x % f[s]) + CDF[s]

Decoding: s = symbol(x & mask)

D(x) = (f[s] * (x >> n) + (x & mask ) - CDF[s], s)

This way we can encode a sequence of symbols into a large natural number x. To avoid using large number arithmetic, in practice stream variants are used: which enforce [math]\displaystyle{ x\in [L, b \cdot L-1] }[/math] by renormalization: sending the least significant bits of x to or from the bitstream (usually L and b are powers of 2).

In rANS variant x is for example 32 bit. For 16 bit renormalization, [math]\displaystyle{ x\in [2^{16},2^{32}-1] }[/math], decoder refills the least significant bits from the bitstream when needed:

if(x < (1 << 16)) x = (x << 16) + read16bits()

Tabled variant (tANS)

Simple example of 4 state ANS automaton for Pr(a) = 3/4, Pr(b) = 1/4 probability distribution. Symbol b contains −lg(1/4) = 2 bits of information and so it always produces two bits. In contrast, symbol a contains −lg(3/4) ~ 0.415 bits of information, hence sometimes it produces one bit (from state 6 and 7), sometimes 0 bits (from state 4 and 5), only increasing the state, which acts as buffer containing fractional number of bits: lg(x). The number of states in practice is for example 2048, for 256 size alphabet (to directly encode bytes).

tANS variant puts the entire behavior (including renormalization) for [math]\displaystyle{ x\in[L,2L-1] }[/math] into a table which yields a finite-state machine avoiding the need of multiplication.

Finally, the step of the decoding loop can be written as:

t = decodingTable(x);  
x = t.newX + readBits(t.nbBits); //state transition
writeSymbol(t.symbol); //decoded symbol

The step of the encoding loop:

s = ReadSymbol();
nbBits = (x + ns[s]) >> r;  // # of bits for renormalization
writeBits(x, nbBits);  // send the least significant bits to bitstream
x = encodingTable[start[s] + (x >> nbBits)];

A specific tANS coding is determined by assigning a symbol to every [math]\displaystyle{ [L,2L-1] }[/math] position, their number of appearances should be proportional to the assumed probabilities. For example, one could choose "abdacdac" assignment for Pr(a)=3/8, Pr(b)=1/8, Pr(c)=2/8, Pr(d)=2/8 probability distribution. If symbols are assigned in ranges of lengths being powers of 2, we would get Huffman coding. For example, a->0, b->100, c->101, d->11 prefix code would be obtained for tANS with "aaaabcdd" symbol assignment.

Example of generation of tANS tables for m = 3 size alphabet and L = 16 states, then applying them for stream decoding. First we approximate probabilities using fraction with denominator being the number of states. Then we spread these symbols in nearly uniform way, optionally the details may depend on cryptographic key for simultaneous encryption. Then we enumerate the appearances starting with value being their amount for a given symbol. Then we refill the youngests bits from the stream to return to the assumed range for x (renormalization).

Remarks

As for Huffman coding, modifying the probability distribution of tANS is relatively costly, hence it is mainly used in static situations, usually with some Lempel–Ziv scheme (e.g. ZSTD, LZFSE). In this case, the file is divided into blocks - for each of them symbol frequencies are independently counted, then after approximation (quantization) written in the block header and used as static probability distribution for tANS.

In contrast, rANS is usually used as a faster replacement for range coding (e.g. CRAM, LZNA, Draco,[13]). It requires multiplication, but is more memory efficient and is appropriate for dynamically adapting probability distributions.

Encoding and decoding of ANS are performed in opposite directions, making it a stack for symbols. This inconvenience is usually resolved by encoding in backward direction, after which decoding can be done forward. For context-dependence, like Markov model, the encoder needs to use context from the perspective of later decoding. For adaptivity, the encoder should first go forward to find probabilities which will be used (predicted) by decoder and store them in a buffer, then encode in backward direction using the buffered probabilities.

The final state of encoding is required to start decoding, hence it needs to be stored in the compressed file. This cost can be compensated by storing some information in the initial state of encoder. For example, instead of starting with "10000" state, start with "1****" state, where "*" are some additional stored bits, which can be retrieved at the end of the decoding. Alternatively, this state can be used as a checksum by starting encoding with a fixed state, and testing if the final state of decoding is the expected one.

Patent controversy

The author of the novel ANS algorithm and its variants tANS and rANS specifically intended his work to be available freely in the public domain, for altruistic reasons. He has not sought to profit from them and took steps to ensure they would not become a "legal minefield", or restricted by, or profited from by others. In 2015, Google published a US and then worldwide patent for "Mixed boolean-token ans coefficient coding".[22] At the time, Professor Duda had been asked by Google to help it with video compression, so was intimately aware of this domain, having the original author assisting them.

Duda was not pleased by (accidentally) discovering Google's patent intentions, given he had been clear he wanted it as public domain, and had assisted Google specifically on that basis. Duda subsequently filed a third-party application[23] to the US Patent office seeking a rejection. The USPTO rejected its application in 2018, and Google subsequently abandoned the patent.[24]

In June 2019 Microsoft lodged a patent application called "Features of range asymmetric number system encoding and decoding".[25] The USPTO issued a final rejection of the application on October 27, 2020. Yet on March 2, 2021, Microsoft gave a USPTO explanatory filing stating "The Applicant respectfully disagrees with the rejections.",[26] seeking to overturn the final rejection under the "After Final Consideration Pilot 2.0" program.[27] After reconsideration, the USPTO granted the application on January 25, 2022.[28]

See also

References

  1. J. Duda, K. Tahboub, N. J. Gadil, E. J. Delp, The use of asymmetric numeral systems as an accurate replacement for Huffman coding, Picture Coding Symposium, 2015.
  2. J. Duda, Asymmetric numeral systems: entropy coding combining speed of Huffman coding with compression rate of arithmetic coding, arXiv:1311.2540, 2013.
  3. "Dr Jarosław Duda (Jarek Duda)". Jagiellonian University in Krakow. http://th.if.uj.edu.pl/~dudaj/. 
  4. Duda, Jarek (October 6, 2019). "List of compressors using ANS, implementations and other materials". https://encode.su/threads/2078-List-of-Asymmetric-Numeral-Systems-implementations. 
  5. "Google Accused of Trying to Patent Public Domain Technology". Bleeping Computer. September 11, 2017. https://www.bleepingcomputer.com/news/google/google-accused-of-trying-to-patent-public-domain-technology/. 
  6. Smaller and faster data compression with Zstandard, Facebook, August 2016.
  7. 5 ways Facebook improved compression at scale with Zstandard, Facebook, December 2018.
  8. Zstd Compression For Btrfs & Squashfs Set For Linux 4.14, Already Used Within Facebook, Phoronix, September 2017.
  9. "Zstd in Android P release". https://android.googlesource.com/kernel/common/+/refs/heads/android-4.14-p-release/fs/btrfs/zstd.c. 
  10. Zstandard Compression and The application/zstd Media Type (email standard).
  11. Hypertext Transfer Protocol (HTTP) Parameters, IANA.
  12. Apple Open-Sources its New Compression Algorithm LZFSE, InfoQ, July 2016.
  13. 13.0 13.1 Google Draco 3D compression library.
  14. Google and Pixar add Draco Compression to Universal Scene Description (USD) Format .
  15. Google PIK: new lossy image format for the internet.
  16. CRAM format specification (version 3.0).
  17. Chen W, Elliott LT (2021). "Compression for population genetic data through finite-state entropy.". J Bioinform Comput Biol 19 (5): 2150026. doi:10.1142/S0219720021500268. PMID 34590992. 
  18. Building better compression together with DivANS.
  19. Microsoft DirectStorage overview.
  20. Rhatushnyak, Alexander; Wassenberg, Jan; Sneyers, Jon; Alakuijala, Jyrki; Vandevenne, Lode; Versari, Luca; Obryk, Robert; Szabadka, Zoltan; Kliuchnikov, Evgenii; Comsa, Iulia-Maria; Potempa, Krzysztof; Bruse, Martin; Firsching, Moritz; Khasanova, Renata; Ruud van Asseldonk; Boukortt, Sami; Gomez, Sebastian; Fischbacher, Thomas (2019). "Committee Draft of JPEG XL Image Coding System". arXiv:1908.03565 [eess.IV].
  21. Data Compression Explained, Matt Mahoney
  22. "Mixed boolean-token ans coefficient coding". https://patents.google.com/patent/US20170164007. 
  23. "Protest to Google". Professor Jarosław Duda. http://th.if.uj.edu.pl/~dudaj/protestGoogle.pdf. 
  24. "After Patent Office Rejection, It is Time For Google To Abandon Its Attempt to Patent Use of Public Domain Algorithm". EFF. 30 August 2018. https://www.eff.org/deeplinks/2018/08/after-patent-office-rejection-it-time-google-abandon-its-attempt-patent-use-public. 
  25. "Features of range asymmetric number system encoding and decoding". https://patents.google.com/patent/US20200413106A1. 
  26. "Third time's a harm? Microsoft tries to get twice-rejected compression patent past skeptical examiners". The Register. https://www.theregister.com/2021/03/13/microsoft_ans_patent/. 
  27. "After Final Consideration Pilot 2.0". https://www.uspto.gov/patents/initiatives/after-final-consideration-pilot-20. 
  28. "Features of range asymmetric number system encoding and decoding". https://patents.google.com/patent/US20200413106A1. 

External links