# Intuitive/heuristic explanation of Polya's urn

Suppose we have an urn with one red ball and one blue ball. At each step, we take out a single ball from the urn and note its color; we then put that ball back into the urn, along with an additional ball of the same color.

This is Polya's urn, and one of the basic facts about it is the following: the number of red balls after $n$ draws is uniform over $\{1, \ldots, n+1\}$.

This is very surprising to me. While its not hard to show this by direct calculation, I wonder if anyone can give an intuitive/heuristic explanation why this distribution should be uniform.

There are quite a few questions on Polya's urn on math.stackexchange, but none of them seem to be asking exactly this. The closest is this question, where there are some nice explanations for why, assuming as above that we start with one red and one blue ball, the probability of drawing a red ball at the $k$'th step is $1/2$ for every $k$ (it follows by symmetry).

• Ah, draws vs. balls, gotcha :) – Colm Bhandal Sep 18 '15 at 22:15
• Not sure why the quoted question doesn't answer yours. The definition of uniform probability is that the probability is constant from draw to draw. The answer to the other question shows that the probability is the same for every draw. What more are you after? – Paul Sinclair Sep 18 '15 at 22:59
• @PaulSinclair -- the fact that the probability to draw a red ball on draw $k$ is $1/2$ does not imply that the distribution of red balls is uniform after $n$ draws. – Michael G Sep 18 '15 at 23:00
• Hmm. It seems I must echo Colm Bhandal's post. – Paul Sinclair Sep 18 '15 at 23:09

By induction: Assume that the proposition holds true for (say) $m=5$ , where $m=n+1$ is the total number of balls, hence the urns may have $k=1\cdots 4$ red balls with uniform probability. The experiment then can be pictured as this: we have 4 urns, each with 5 balls, the first with $1$ red ball, the second with 2, etc. We pick one of these 4 urns, with same probability, and then we pick a ball with uniform probability from the urn. This is the same as saying that we pick a ball at random, always uniformly, from the total $4\times 5=20$ balls. The following picture helps (I hope) to envision the possible alternatives.

On the left, we have the 4 urns, and the 20 balls, from which we will draw one. On the right, the same balls are arranged with a displacement along the diagonal. In this way we can identify all the balls that, if selected, will lead to each of the new urns (diagram below). Because we have the same number of total balls for each new urn, each one wil have the same probability.

• a very cool pictorial illustration! – Michael G Sep 19 '15 at 1:24
• Two comments: First, it seems to me that a key condition for these results is that the urn’s history is unknown. Knowing the results of any of the draws can skew the probabilities and ball distributions away from uniformity. Second, this reminds me of Schrödinger’s cat and quantum-mechanical superposition of states. Unless you look in the urn and count the balls, you can consider it to be in all of its possible states at once. – amd Sep 19 '15 at 8:48

The best I can do here is an inductive argument:

• Clearly, after $0$ draws, there is always exactly $1$ red ball with a probability of $1 = \frac1{0+1}$ i.e. certainty of being there.

• For the inductive case, we notice that drawing $n + 1$ balls first begins with drawing $n$ balls, then another. After drawing the $n$ balls, there is uniform chance of any number of red balls in the urn from $1$ to $n+1$, by induction. That is, for any $1 \leq k \leq n+1$, the probability of $k$ red balls is $\frac1{n+1}$. Now, what is the probability of having $k$ red balls after $n + 1$ draws? Let's consider some cases:

1. If $2 \leq k \leq n + 1$, we have two more cases to consider with non-zero probability: $k$ balls after $n$ draws, and then draw a blue ball, or we have $k - 1$ after $n$ draws, and then we draw a red ball. Now, after we have already drawn $n$ times, there are $n+2$ balls in total in the urn. The chances then of drawing a blue ball, given $k$ red balls in the urn is $\frac{n + 2 - k}{n + 2}$. The chances of drawing a red ball given $k - 1$ red balls is $\frac{k - 1}{n + 2}$. Both numbers of red balls after $n$ draws are equally likely by the inductive hypothesis, with probability $\frac1{n+1}$. The overall probability is then $\frac1{n + 1}(\frac{n + 2 - k}{n + 2} + \frac{k - 1}{n + 2}) = \frac1{n + 2}$.

2. If $k = 1$, then we must never draw a red ball. So we must have exactly $1$ red ball remaining after $n$ draws, and then draw a blue ball. By induction, the probability of $1$ ball after $n$ draws is $\frac1{n+1}$. Drawing a blue ball after this has probability $\frac{n+1}{n+2}$. Multiplying the two probabilities easily gives us the desired $\frac{1}{n+2}$.

3. If $k = n+2$, then the problem is symmetric, considering $k=1$ for blue balls instead of red balls and so the previous argument applies.

It seems as though Polya chose this question intentionally to challenge our very intuitions, which are so often wrong. He seems to have crafted the question also to force us to use a clever argument of induction, topped with the icing of an argument based on symmetry. Naturally, we are inclined to think of some sort of geometrically decreasing probability distribution across $n$. Once symmetry is realised, we must update this: perhaps the distribution approaches a normal distribution as $n$ gets large? These are certainly intuitions that I had when I saw this question. But I don't trust my intuition. And I am right not to. This is a good demonstration of where intuition fails, at least initially, before we really soak in the problem. So we look instead for a rigorous demonstration, and the most elegant one at that.

As Einstein once said "Everything Should Be Made as Simple as Possible, But Not Simpler". Erdos used to talk about the most elegant proofs as being "from the book". I won't blacken either of their names by claiming the above is "as simple as possible" or "from the book", but it's certainly the best I can think of. Intuitively, I don't feel you can "compress" the argument much further, without losing its essence. That intuition I can't explain, it's just a hunch.

Having said that, the key steps in the argument, I believe, are as follows. The first is hitting upon the right idea of induction: i.e. induction on $n$, not $k$. There is a conceptual "aha" moment when you realise that drawing $n + 1$ balls is the same as first drawing $n$ balls, and then drawing another. The second "aha" moment is when you follow your previous idea for induction, and notice that all values of $k$ balls after $n + 1$ draws, except the corner cases of $n + 2$ and $1$, can happen in exactly two ways: either $k$ balls are chosen from $n$, and then no more, or $k - 1$ are chosen, and $1$ more. We follow this hunch, work out the maths, and hey presto, the correct answer pops out nicely. Then we're left with the corner cases. The case for $1$ also works, hoorah! The final touch of elegance is to use a symmetric argument on the case of $n+2$, rather than work it out from scratch.

There is a shorter way of explaining this, quasi-intuitively, but it loses the crispness of the above argument. Still, it is a nice auxiliary intuition to complement the above. Again, imagine the inductive proof. Imagine all the possibly values for red balls: $1, 2, 3, \dots, n+2$. In the next step, each choice "gives" part of its probability away to the next one up. Clearly larger values give more than smaller ones: because there is more chance of drawing another red once you already have red. But they also get more, because they get a value from the number below. Still, overall they lose. Each gets $\frac{k - 1}{n + 2}$ and gives $\frac{k}{n + 2}$ So $1$ "gives away" a value of $\frac1{n + 2}$ to $2$. $2$ "gets" this, and gives away $\frac2{n + 2}$ to $3$ etc. Eventually, when all values are passed on like this, the distribution settles back at uniformity, except now with $\frac1{n + 2}$ instead of $\frac1{n + 1}$.