Alias method

From HandWiki
Short description: Family of algorithms for sampling from discrete probability distributions
A circle on the left has 5 lines to 5 boxes in a column labeled "Acceptance". The first and second box are solid and each have the number 1 in them. The second box is half full and has the number 0.5 in it. The fourth box is solid with a 1 and the fifth box is three quarters full with a 0.75. Each box has an arrow from the filled region to its index, i.e., the first box points to a 1, the second box to a two, etc. There is a second column of five boxes labeled "Alias", each corresponding to one of the first boxes. Three are empty, but the third has a 2 in it and the fifth has a 1 in it. There is an arrow from the empty part of the third box in the first column to the third box in the second column and similarly for the fifth boxes.
A diagram of an alias table that represents the probability distribution〈0.25, 0.3, 0.1, 0.2, 0.15〉

In computing, the alias method is a family of efficient algorithms for sampling from a discrete probability distribution, published in 1974 by A. J. Walker.[1][2] That is, it returns integer values 1 ≤ in according to some arbitrary probability distribution pi. The algorithms typically use O(n log n) or O(n) preprocessing time, after which random values can be drawn from the distribution in O(1) time.[3]

Operation

Internally, the algorithm consults two tables, a probability table Ui and an alias table Ki (for 1 ≤ in). To generate a random outcome, a fair dice is rolled to determine an index into the two tables. Based on the probability stored at that index, a biased coin is then flipped, and the outcome of the flip is used to choose between a result of i and Ki.[4]

More concretely, the algorithm operates as follows:

  1. Generate a uniform random variate 0 ≤ x < 1.
  2. Let i = ⌊nx⌋ + 1 and y = nx + 1 − i. (This makes i uniformly distributed on {1, 2, ..., n} and y uniformly distributed on [0, 1).)
  3. If y < Ui, return i. This is the biased coin flip.
  4. Otherwise, return Ki.

An alternative formulation of the probability table, proposed by Marsaglia et al.[5] as the "square histogram" method, uses the condition x < Vi in the third step (where Vi = (Ui + i − 1)/n) instead of computing y.

Table generation

The distribution may be padded with additional probabilities pi = 0 to increase n to a convenient value, such as a power of two.

To generate the table, first initialize Ui = npi. While doing this, divide the table entries into three categories:

  • The "overfull" group, where Ui > 1,
  • The "underfull" group, where Ui < 1 and Ki has not been initialized, and
  • The "exactly full" group, where Ui = 1 or Ki has been initialized.

If Ui = 1, the corresponding value Ki will never be consulted and is unimportant, but a value of Ki = i is sensible.

As long as not all table entries are exactly full, repeat the following steps:

  1. Arbitrarily choose an overfull entry Ui > 1 and an underfull entry Uj < 1. (If one of these exists, the other must, as well.)
  2. Allocate the unused space in entry j to outcome i, by setting Kj = i.
  3. Remove the allocated space from entry i by changing Ui = Ui − (1 − Uj) = Ui + Uj − 1.
  4. Entry j is now exactly full.
  5. Assign entry i to the appropriate category based on the new value of Ui.

Each iteration moves at least one entry to the "exactly full" category (and the last moves two), so the procedure is guaranteed to terminate after at most n−1 iterations. Each iteration can be done in O(1) time, so the table can be set up in O(n) time.

Vose[3]:974 points out that floating-point rounding errors may cause the guarantee referred to in step 1 to be violated. If one category empties before the other, the remaining entries may have Ui set to 1 with negligible error. The solution accounting for floating point is sometimes called the Walker-Vose method or the Vose alias method.

The Alias structure is not unique.

As the lookup procedure is slightly faster if y < Ui (because Ki does not need to be consulted), one goal during table generation is to maximize the sum of the Ui. Doing this optimally turns out to be NP hard,[5]:6 but a greedy algorithm comes reasonably close: rob from the richest and give to the poorest. That is, at each step choose the largest Ui and the smallest Uj. Because this requires sorting the Ui, it requires O(n log n) time.

Efficiency

Although the alias method is very efficient if generating a uniform deviate is itself fast, there are cases where it is far from optimal in terms of random bit usage. This is because it uses a full-precision random variate x each time, even when only a few random bits are needed.

One case arises when the probabilities are particularly well balanced, so many Ui = 1 and Ki is not needed. Generating y is a waste of time. For example if p1 = p2 = ​12, then a 32-bit random variate x could be used to make 32 choices, but the alias method will only generate one.

Another case arises when the probabilities are strongly unbalanced, so many Ui ≈ 0. For example if p1 = 0.999 and p2 = 0.001, then the great majority of the time, only a few random bits are required to determine that case 1 applies. In such cases, the table method described by Marsaglia et al.[5]:1–4 is more efficient. If we make many choices with the same probability we can on average require much less than one unbiased random bit. Using arithmetic coding techniques arithmetic we can approach the limit given by the binary entropy function.

Literature

Implementations

References

  1. Walker, A. J. (April 1974). "New fast method for generating discrete random numbers with arbitrary frequency distributions". Electronics Letters 10 (8): 127. doi:10.1049/el:19740097. Bibcode1974ElL....10..127W. 
  2. Walker, A. J. (September 1977). "An Efficient Method for Generating Discrete Random Variables with General Distributions". ACM Transactions on Mathematical Software 3 (3): 253–256. doi:10.1145/355744.355749. 
  3. 3.0 3.1 Vose, Michael D. (September 1991). "A linear algorithm for generating random numbers with a given distribution". IEEE Transactions on Software Engineering 17 (9): 972–975. doi:10.1109/32.92917. http://web.eecs.utk.edu/~vose/Publications/random.pdf. 
  4. "Darts, Dice, and Coins: Sampling from a Discrete Distribution". 29 December 2011. http://www.keithschwarz.com/darts-dice-coins/. 
  5. 5.0 5.1 5.2 Marsaglia, George; Tsang, Wai Wan; Wang, Jingbo (2004-07-12), "Fast Generation of Discrete Random Variables", Journal of Statistical Software 11 (3): 1–11, doi:10.18637/jss.v011.i03