Asymmetric numeral systems explained

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] Google Chrome browser,[9] Android[10] operating system, was published as RFC 8478 for MIME[11] and HTTP[12]), Apple LZFSE compressor,[13] Google Draco 3D compressor[14] (used e.g. in Pixar Universal Scene Description format[15]) and PIK image compressor,[16] CRAM DNA compressor[17] from SAMtools utilities,[18] NVIDIA nvCOMP high speed compression library,[19] Dropbox DivANS compressor,[20] Microsoft DirectStorage BCPack texture compressor,[21] and JPEG XL[22] image compressor.

The basic idea is to encode information into a single natural number

x

. In the standard binary number system, we can add a bit

s\in\{0,1\}

of information to

x

by appending

s

at the end of

x

, which gives us

x'=2x+s

. For an entropy coder, this is optimal if

\Pr(0)=\Pr(1)=1/2

. ANS generalizes this process for arbitrary sets of symbols

s\inS

with an accompanying probability distribution

(ps)s

. In ANS, if the information from

s

is appended to

x

to result in

x'

, then

x'x

-1
p
s
. Equivalently,

log2(x')log2(x)+log2(1/ps)

, where

log2(x)

is the number of bits of information stored in the number

x

, and

log2(1/ps)

is the number of bits contained in the symbol

s

.

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

s

into the information already stored in the current number

x

, we go to number

x'=C(x,s)x/p

being the position of the

x

-th appearance from the

s

-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

x

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

\lceillog2(1000)\rceil10

bits here instead of the original 1000 bits.

Generally, such length

n

sequences containing

pn

zeros and

(1-p)n

ones, for some probability

p\in(0,1)

, are called combinations. Using Stirling's approximation we get their asymptotic number being

{n\choosepn}2nh(p)forlargenandh(p)=-plog2(p)-(1-p)log2(1-p),

called Shannon entropy.

Hence, to choose one such sequence we need approximately

nh(p)

bits. It is still

n

bits if

p=1/2

, however, it can also be much smaller. For example, we need only

n/2

bits for

p=0.11

.

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

p

contains

log2(1/p)

bits of information. ANS encodes information into a single natural number

x

, interpreted as containing

log2(x)

bits of information. Adding information from a symbol of probability

p

increases this informational content to

log2(x)+log2(1/p)=log2(x/p)

. Hence, the new number containing both information should be

x'x/p

.

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:

Consider a more general source with k letters, with rational probabilities

n1/N,...,nk/N

. 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

r1,...,rk

by rational numbers

n1/N,...,nk/N

with a small denominator

N

.

Basic concepts of ANS

Imagine there is some information stored in a natural number

x

, for example as bit sequence of its binary expansion. To add information from a binary variable

s

, we can use coding function

x'=C(x,s)=2x+s

, which shifts all bits one position up, and place the new bit in the least significant position. Now decoding function

D(x')=(\lfloorx'/2\rfloor,mod(x',2))

allows one to retrieve the previous

x

and this added bit:

D(C(x,s))=(x,s),C(D(x'))=x'

. We can start with

x=1

initial state, then use the

C

function on the successive bits of a finite bit sequence to obtain a final

x

number storing this entire sequence. Then using

D

function multiple times until

x=1

allows one to retrieve the bit sequence in reversed order.

The above procedure is optimal for the uniform (symmetric) probability distribution of symbols

\Pr(0)=\Pr(1)=1/2

. ANS generalize it to make it optimal for any chosen (asymmetric) probability distribution of symbols:

\Pr(s)=ps

. While

s

in the above example was choosing between even and odd

C(x,s)

, in ANS this even/odd division of natural numbers is replaced with division into subsets having densities corresponding to the assumed probability distribution

\{ps\}s

: up to position

x

, there are approximately

xps

occurrences of symbol

s

.

The coding function

C(x,s)

returns the

x

-th appearance from such subset corresponding to symbol

s

. The density assumption is equivalent to condition

x'=C(x,s)x/ps

. Assuming that a natural number

x

contains

log2(x)

bits of information,

log2(C(x,s))log2(x)+log2(1/ps)

. Hence the symbol of probability

ps

is encoded as containing

≈ log2(1/ps)

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

\Pr(1)=p

,

\Pr(0)=1-p

. Up to position

x

we want approximately

px

analogues of odd numbers (for

s=1

). We can choose this number of appearances as

\lceilxp\rceil

, getting

s=\lceil(x+1)p\rceil-\lceilxp\rceil

. This variant is called uABS and leads to the following decoding and encoding functions:[23]

Decoding:s = ceil((x+1)*p) - ceil(x*p) // 0 if fract(x*p) < 1-p, else 1if 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_xif s = 1 then new_x = floor(x/p) // C(x,1) = new_x

For

p=1/2

it amounts to the standard binary system (with 0 and 1 inverted), for a different

p

it becomes optimal for this given probability distribution. For example, for

p=0.3

these formulas lead to a table for small values of

x

:

C(x,s)

01234567891011121314151617181920

s=0

012345678910111213

s=1

0123456
The symbol

s=1

corresponds to a subset of natural numbers with density

p=0.3

, which in this case are positions

\{0,3,6,10,13,16,20,23,26,\ldots\}

. As

1/4<0.3<1/3

, these positions increase by 3 or 4. Because

p=3/10

here, the pattern of symbols repeats every 10 positions.

The coding

C(x,s)

can be found by taking the row corresponding to a given symbol

s

, and choosing the given

x

in this row. Then the top row provides

C(x,s)

. For example,

C(7,0)=11

from the middle to the top row.

Imagine we would like to encode the sequence '0100' starting from

x=1

. First

s=0

takes us to

x=2

, then

s=1

to

x=6

, then

s=0

to

x=9

, then

s=0

to

x=14

. By using the decoding function

D(x')

on this final

x

, we can retrieve the symbol sequence. Using the table for this purpose,

x

in the first row determines the column, then the non-empty row and the written value determine the corresponding

s

and

x

.

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

2n

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

2n

denominator, where is chosen (usually 8-12 bits):

psf[s]/2n

for some natural

f[s]

numbers (sizes of subranges).

Denote

mask=2n-1

, and a cumulative distribution function:

\operatorname{CDF}[s]=\sumi<sf[i]=f[0]++f[s-1].

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

y\in[0,2n-1]

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 . To avoid using large number arithmetic, in practice stream variants are used: which enforce

x\in[L,bL-1]

by renormalization: sending the least significant bits of to or from the bitstream (usually and are powers of 2).

In rANS variant is for example 32 bit. For 16 bit renormalization,

x\in[216,232-1]

, decoder refills the least significant bits from the bitstream when needed:

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

Tabled variant (tANS)

tANS variant puts the entire behavior (including renormalization) for

x\in[L,2L-1]

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 transitionwriteSymbol(t.symbol); //decoded symbol

The step of the encoding loop:s = ReadSymbol;nbBits = (x + ns[s]) >> r; // # of bits for renormalizationwriteBits(x, nbBits); // send the least significant bits to bitstreamx = encodingTable[start[s] + (x >> nbBits)];

A specific tANS coding is determined by assigning a symbol to every

[L,2L-1]

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.

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,). 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".[24] 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[25] to the US Patent office seeking a rejection. The USPTO rejected its application in 2018, and Google subsequently abandoned the patent.[26]

In June 2019 Microsoft lodged a patent application called "Features of range asymmetric number system encoding and decoding".[27] 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.",[28] seeking to overturn the final rejection under the "After Final Consideration Pilot 2.0" program.[29] After reconsideration, the USPTO granted the application on January 25, 2022.[30]

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. Web site: Dr Jarosław Duda (Jarek Duda) . Institute of Theoretical Physics . Jagiellonian University in Krakow . 2021-08-02.
  4. Web site: List of compressors using ANS, implementations and other materials . Duda . Jarek . October 6, 2019 . October 6, 2019.
  5. Web site: Google Accused of Trying to Patent Public Domain Technology . . September 11, 2017.
  6. https://code.facebook.com/posts/1658392934479273/smaller-and-faster-data-compression-with-zstandard/ Smaller and faster data compression with Zstandard
  7. https://code.fb.com/core-data/zstandard/ 5 ways Facebook improved compression at scale with Zstandard
  8. https://www.phoronix.com/scan.php?page=news_item&px=Linux-4.14-Zstd-Pull Zstd Compression For Btrfs & Squashfs Set For Linux 4.14, Already Used Within Facebook
  9. https://developer.chrome.com/blog/new-in-chrome-123 New in Chrome 123 (Content-Encoding)
  10. Web site: Zstd in Android P release . 2019-05-29 . 2020-08-26 . https://web.archive.org/web/20200826023340/https://android.googlesource.com/kernel/common/+/refs/heads/android-4.14-p-release/fs/btrfs/zstd.c . dead .
  11. https://datatracker.ietf.org/doc/draft-kucherawy-dispatch-zstd/ Zstandard Compression and The application/zstd Media Type (email standard)
  12. https://www.iana.org/assignments/http-parameters/http-parameters.xhtml Hypertext Transfer Protocol (HTTP) Parameters
  13. https://www.infoq.com/news/2016/07/apple-lzfse-lossless-opensource Apple Open-Sources its New Compression Algorithm LZFSE
  14. https://github.com/google/draco Google Draco 3D compression library
  15. https://opensource.googleblog.com/2019/11/google-and-pixar-add-draco-compression.html Google and Pixar add Draco Compression to Universal Scene Description (USD) Format
  16. https://github.com/google/pik Google PIK: new lossy image format for the internet
  17. https://samtools.github.io/hts-specs/CRAMv3.pdf CRAM format specification (version 3.0)
  18. Chen W, Elliott LT. Compression for population genetic data through finite-state entropy. . J Bioinform Comput Biol . 2021 . 19. 5. 2150026 . 34590992 . 10.1142/S0219720021500268 . free .
  19. https://developer.nvidia.com/nvcomp High Speed Data Compression Using NVIDIA GPUs
  20. https://blogs.dropbox.com/tech/2018/06/building-better-compression-together-with-divans/ Building better compression together with DivANS
  21. https://docs.microsoft.com/en-us/gaming/gdk/_content/gc/system/overviews/directstorage/directstorage-overview Microsoft DirectStorage overview
  22. Committee Draft of JPEG XL Image Coding System . 1908.03565 . 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 . eess.IV . 2019.
  23. http://mattmahoney.net/dc/dce.html#Section_33 Data Compression Explained
  24. Web site: Mixed boolean-token ans coefficient coding . 14 June 2021.
  25. Web site: Protest to Google . Institute of Theoretical Physics. Jagiellonian University in Krakow Poland . Professor Jarosław Duda.
  26. Web site: After Patent Office Rejection, It is Time For Google To Abandon Its Attempt to Patent Use of Public Domain Algorithm . 30 August 2018 . EFF.
  27. Web site: Features of range asymmetric number system encoding and decoding . 14 June 2021.
  28. Web site: Third time's a harm? Microsoft tries to get twice-rejected compression patent past skeptical examiners . The Register . 14 June 2021.
  29. Web site: After Final Consideration Pilot 2.0 . United States Patent and Trademark Office . 14 June 2021.
  30. Web site: Features of range asymmetric number system encoding and decoding . 16 February 2022.

External links