Pólya urn model

Questions of interest are the evolution of the urn population and the sequence of colors of the balls drawn out.

After ${\displaystyle n}$ draws, the probability that the urn contains ${\displaystyle (x+n_{1})}$ white balls and ${\displaystyle (y+n_{2})}$ black balls is ${\displaystyle {\binom {n}{n_{1}}}{\frac {x^{{\bar {n}}_{1}}y^{{\bar {n}}_{2}}}{(x+y)^{\bar {n}}}}}$, where the overbar denotes rising factorial. This can be proved by drawing the Pascal's triangle of all possible configurations.

More generally, if the urn starts with ${\displaystyle a_{i}}$ balls of color ${\displaystyle i}$, with ${\displaystyle i=1,2,...,k}$, then after ${\displaystyle n}$ draws, the probability that the urn contains ${\displaystyle (a_{i}+n_{i})}$ balls of color ${\displaystyle i}$ is

$${\displaystyle {\binom {n}{n_{1},\cdots ,n_{k}}}{\frac {\prod _{i=1}^{k}a_{i}^{{\bar {n}}_{i}}}{(\sum _{i}a_{i})^{\bar {n}}}}}$$

where we use the multinomial coefficient.

Conditional on the urn ending up with ${\displaystyle (a_{i}+n_{i})}$ balls of color ${\displaystyle i}$ after ${\displaystyle n}$ draws, there are ${\displaystyle {\binom {n}{n_{1},\cdots ,n_{k}}}}$ different trajectories that could have led to such an end-state. The conditional probability of each trajectory is the same: ${\displaystyle {\binom {n}{n_{1},\cdots ,n_{k}}}^{-1}}$.

One of the reasons for interest in this particular rather elaborate urn model (i.e. with duplication and then replacement of each ball drawn) is that it provides an example in which the count (initially x black and y white) of balls in the urn is not concealed, which is able to approximate the correct updating of subjective probabilities appropriate to a different case in which the original urn content is concealed while ordinary sampling with replacement is conducted (without the Pólya ball-duplication). Because of the simple "sampling with replacement" scheme in this second case, the urn content is now static, but this greater simplicity is compensated for by the assumption that the urn content is now unknown to an observer. A Bayesian analysis of the observer's uncertainty about the urn's initial content can be made, using a particular choice of (conjugate) prior distribution. Specifically, suppose that an observer knows that the urn contains only identical balls, each coloured either black or white, but they do not know the absolute number of balls present, nor the proportion that are of each colour. Suppose that they hold prior beliefs about these unknowns: for them the probability distribution of the urn content is well approximated by some prior distribution for the total number of balls in the urn, and a beta prior distribution with parameters (x,y) for the initial proportion of these which are black, this proportion being (for them) considered approximately independent of the total number. Then the process of outcomes of a succession of draws from the urn (with replacement but without the duplication) has approximately the same probability law as does the above Pólya scheme in which the actual urn content was not hidden from them. The approximation error here relates to the fact that an urn containing a known finite number m of balls of course cannot have an exactly beta-distributed unknown proportion of black balls, since the domain of possible values for that proportion are confined to being multiples of ${\displaystyle 1/m}$, rather than having the full freedom to assume any value in the continuous unit interval, as would an exactly beta distributed proportion. This slightly informal account is provided for reason of motivation, and can be made more mathematically precise.

This basic Pólya urn model has been generalized in many ways.

• beta-binomial distribution: The distribution of the number of successful draws (trials), e.g. number of extractions of white ball, given ${\displaystyle n}$ draws from a Pólya urn.
• Beta negative binomial distribution: The distribution of number of white balls observed until a fixed number black balls are observed.
• Dirichlet-multinomial distribution (also known as the multivariate Pólya distribution): The distribution over the number of balls of each color, given ${\displaystyle n}$ draws from a Pólya urn where there are ${\displaystyle k}$ different colors instead of only two.
• Dirichlet negative multinomial distribution: The distribution over the number of balls of each color until a fixed number of stopping colored balls are observed.
• martingales, the Beta-binomial distribution and the beta distribution: Let w and b be the number of white and black balls initially in the urn, and ${\displaystyle w+n_{w}}$ the number of white balls currently in the urn after n draws. Then the sequence of values ${\displaystyle {\frac {w+n_{w}}{w+b+n}}}$ for ${\displaystyle n=1,2,3,\dots }$ is a normalized version of the Beta-binomial distribution. It is a martingale and converges to the beta distribution when n → ∞.
• Dirichlet process, Chinese restaurant process, Hoppe urn: Imagine a modified Pólya urn scheme as follows. We start with an urn with ${\displaystyle \alpha }$ black balls. When drawing a ball from the urn, if we draw a black ball, put the ball back along with a new ball of a new non-black color randomly generated from a uniform distribution over an infinite set of available colours, and consider the newly generated color to be the "value" of the draw. Otherwise, put the ball back along with another ball of the same color, as for the standard Pólya urn scheme. The colors of an infinite sequence of draws from this modified Pólya urn scheme follow a Chinese restaurant process. If, instead of generating a new color, we draw a random value from a given base distribution and use that value to label the ball, the labels of an infinite sequence of draws follow a Dirichlet process.[1]
• Moran model: An urn model used to model genetic drift in theoretical population genetics. This is closely similar to the Pólya urn model except that, in addition to adding a new ball of the same color, a randomly drawn ball is removed from the urn. The number of balls in the urn thus remains constant. Continued sampling then leads ultimately to an urn with all balls of one color, the probability of each color being the proportion of that color in the original urn. There are variants of the Moran model that insist that the ball removed from the urn be a different ball from one originally sampled in that step, and variants that do the removal of a ball immediately after the new ball is placed in the urn, so that the new ball is one of the balls available to be removed. This makes a small difference in the time taken to reach the state in which all balls are the same color. The Moran process models genetic drift in a population with overlapping generations.

Polya's Urn is a quintessential example of an exchangeable process.

Suppose we have an urn containing ${\displaystyle \gamma }$ white balls and ${\displaystyle \alpha }$ black balls. We proceed to draw balls at random from the urn. On the ${\displaystyle i}$-th draw, we define a random variable, ${\displaystyle X_{i}}$, by ${\displaystyle X_{i}=1}$ if the ball is black and ${\displaystyle X_{i}=0}$ otherwise. We then return the ball to the urn, with an additional ball of the same colour. For a given ${\displaystyle i}$, if we have that ${\displaystyle X_{j}=1}$ for many ${\displaystyle j<i}$, then it is more likely that ${\displaystyle X_{i}=1}$, because more black balls have been added to the urn. Therefore, these variables are not independent of each other.

The sequence ${\displaystyle X_{1},X_{2},X_{3},\dots }$ does, however, exhibit the weaker property of exchangeability.[2] Recall that a (finite or infinite) sequence of random variables is called exchangeable if its joint distribution is invariant under permutations of indices.

To show exchangeability of the sequence ${\displaystyle X_{1},X_{2},X_{3},\dots }$, assume that ${\displaystyle n}$ balls are picked from the urn, and out of these ${\displaystyle n}$ balls, ${\displaystyle k}$ balls are black and ${\displaystyle n-k}$ are white. On the first draw the number of balls in the urn is ${\displaystyle \gamma +\alpha }$; on the second draw it is ${\displaystyle \gamma +\alpha +1}$ and so on. On the ${\displaystyle i}$-th draw, the number of balls will be ${\displaystyle \gamma +\alpha +i-1}$. The probability that we draw all ${\displaystyle k}$ black balls first, and then all ${\displaystyle n-k}$ white balls is given by${\displaystyle \mathbb {P} \left(X_{1}=1,\dots ,X_{k}=1,X_{k+1}=0,\dots ,X_{n}=0\right)={\frac {\alpha }{\gamma +\alpha }}\times {\frac {\alpha +1}{\gamma +\alpha +1}}\times \cdots \times {\frac {\alpha +k-1}{\gamma +\alpha +k-1}}\times {\frac {\gamma }{\gamma +\alpha +k}}\times {\frac {\gamma +1}{\gamma +\alpha +k+1}}\times \cdots \times {\frac {\gamma +n-k-1}{\gamma +\alpha +n-1}}}$

Now we must show that if the order of black and white balls is permuted, there is no change to the probability. As in the expression above, even after permuting the draws, the ${\displaystyle i}$th denominator will always be ${\displaystyle \gamma +\alpha +i-1}$, since this is the number of balls in the urn at that round.

If we see ${\displaystyle j}$-th black ball in round ${\displaystyle t}$, the probability ${\displaystyle X_{t}=1}$ will be equal to ${\displaystyle {\frac {\alpha +j-1}{\gamma +\alpha +t-1}}}$, i.e. the numerator will be equal to ${\displaystyle \alpha +j-1}$. With the same argument, we can calculate the probability for white balls. Therefore, for any sequence ${\displaystyle x_{1},x_{2},x_{3},\dots }$ in which ${\displaystyle 1}$ occurs ${\displaystyle k}$ times and ${\displaystyle 0}$ occurs ${\displaystyle n-k}$ times (i.e. a sequence with ${\displaystyle k}$ black balls and ${\displaystyle n-k}$ white balls drawn in some order) the final probability will be equal to the following expression, where we take advantage of commutativity of multiplication in the numerator:

$${\displaystyle {\begin{aligned}\mathbb {P} (X_{1}=x_{1},X_{2}=x_{2},...,X_{n}=x_{n})&={\frac {\prod _{i=1}^{k}\left(\alpha +i-1\right)\times \prod _{i=1}^{n-k}\left(\gamma +i-1\right)}{\prod _{i=1}^{n}\left(\gamma +\alpha +i-1\right)}}\\&={\frac {\left(\alpha +k-1\right)!\times \left(\gamma +n-k-1\right)!\times \left(\alpha +\gamma -1\right)!}{\left(\alpha -1\right)!\times \left(\gamma -1\right)!\left(\alpha +\gamma +n-1\right)!}}\end{aligned}}}$$

This probability is not related to the order of seeing black and white balls and only depends on the total number of white balls and the total number of black balls.[2]

According to the De Finetti's theorem, there must be a unique prior distribution such that the joint distribution of observing the sequence is a Bayesian mixture of the Bernoulli probabilities. It can be shown that this prior distribution is a beta distribution with parameters ${\displaystyle \beta \left(\cdot$ ;\,\alpha ,\,\gamma \right)} . In De Finetti's theorem, if we replace ${\displaystyle \pi (\cdot )}$ with ${\displaystyle \beta \left(\cdot$ ;\,\alpha ,\,\gamma \right)} , then we get the previous equation:[2]

$${\displaystyle {\begin{aligned}p(X_{1}=x_{1},X_{2}=x_{2},...,X_{n}=x_{n})&=\int \theta ^{\left({\sum _{i=1}^{n}x_{i}}\right)}\times \left(1-\theta \right)^{\left(n-{\sum _{i=1}^{n}x_{i}}\right)}\,\beta \left(\theta$$ ;\alpha ,\,\gamma \right)d\left(\theta \right)\\&=\int \theta ^{\left({\sum _{i=1}^{n}x_{i}}\right)}\times \left(1-\theta \right)^{\left(n-{\sum _{i=1}^{n}x_{i}}\right)}\,{\dfrac {(\alpha +\gamma -1)!}{(\alpha -1)!\,(\gamma -1)!}}\theta ^{\alpha -1}(1-\theta )^{\gamma -1}d\left(\theta \right)\\&=\int \theta ^{\left({\alpha -1+\sum _{i=1}^{n}x_{i}}\right)}\times \left(1-\theta \right)^{\left(n+\gamma -1-{\sum _{i=1}^{n}x_{i}}\right)}\,{\dfrac {(\alpha +\gamma -1)!}{(\alpha -1)!\,(\gamma -1)!}}d\left(\theta \right)\\&=\int \theta ^{\left({\alpha +k-1}\right)}\times \left(1-\theta \right)^{\left(n-k-1+\gamma \right)}\,{\dfrac {(\alpha +\gamma -1)!}{(\alpha -1)!\,(\gamma -1)!}}d\left(\theta \right)\\&={\dfrac {(\alpha +\gamma -1)!}{(\alpha -1)!\,(\gamma -1)!}}\int \theta ^{\left({\alpha +k-1}\right)}\times \left(1-\theta \right)^{\left(n-k+\gamma -1\right)}\,d\left(\theta \right)\\&={\dfrac {(\alpha +\gamma -1)!}{(\alpha -1)!\,(\gamma -1)!}}{\dfrac {\Gamma (\gamma +n-k)\Gamma (\alpha +k)}{\Gamma (\alpha +\gamma +n)}}\\&={\dfrac {\left(\alpha +k-1\right)!\times \left(\gamma +n-k-1\right)!\times \left(\alpha +\gamma -1\right)!}{\left(\alpha -1\right)!\times \left(\gamma -1\right)!\left(\alpha +\gamma +n-1\right)!}}\end{aligned}}}

In this equation ${\displaystyle k=\sum _{i=1}^{n}x_{i}}$.

• Pitman–Yor process
• Moran process
• Yule process
• De Finetti's theorem
• Chinese restaurant process

• F. Alajaji and T. Fuja, "A Communication Channel Modeled on Contagion," IEEE Transactions on Information Theory, Vol. 40, pp. 2035–2041, November 1994.
• A. Banerjee, P. Burlina and F. Alajaji, "Image Segmentation and Labeling Using the Pólya Urn Model," IEEE Transactions on Image Processing, Vol. 8, No. 9, pp. 1243–1253, September 1999.

• N.L. Johnson and S.Kotz, (1977) "Urn Models and Their Application." John Wiley.
• Hosam Mahmoud, (2008) "Pólya Urn Models." Chapman and Hall/CRC. ISBN 978-1420059830.