Pointwise mutual information

In statistics, probability theory and information theory, pointwise mutual information (PMI),^[1] or point mutual information, is a measure of association. It compares the probability of two events occurring together to what this probability would be if the events were independent.^[2]

PMI (especially in its positive pointwise mutual information variant) has been described as "one of the most important concepts in NLP", where it "draws on the intuition that the best way to weigh the association between two words is to ask how much more the two words co-occur in [a] corpus than we would have expected them to appear by chance."^[2]

The concept was introduced in 1961 by Robert Fano under the name of "mutual information", but today that term is instead used for a related measure of dependence between random variables:^[2] The mutual information (MI) of two discrete random variables refers to the average PMI of all possible events.

The PMI of a pair of outcomes x and y belonging to discrete random variables X and Y quantifies the discrepancy between the probability of their coincidence given their joint distribution and their individual distributions, assuming independence. Mathematically:^[2]

${\displaystyle \operatorname {pmi} (x;y)\equiv \log _{2}{\frac {p(x,y)}{p(x)p(y)}}=\log _{2}{\frac {p(x|y)}{p(x)}}=\log _{2}{\frac {p(y|x)}{p(y)}}}$

(with the latter two expressions being equal to the first by Bayes' theorem). The mutual information (MI) of the random variables X and Y is the expected value of the PMI (over all possible outcomes).

The measure is symmetric (${\displaystyle \operatorname {pmi} (x;y)=\operatorname {pmi} (y;x)}$). It can take positive or negative values, but is zero if X and Y are independent. Note that even though PMI may be negative or positive, its expected outcome over all joint events (MI) is non-negative. PMI maximizes when X and Y are perfectly associated (i.e. ${\displaystyle p(x|y)}$ or ${\displaystyle p(y|x)=1}$), yielding the following bounds:

${\displaystyle -\infty \leq \operatorname {pmi} (x;y)\leq \min \left[-\log p(x),-\log p(y)\right].}$

Finally, ${\displaystyle \operatorname {pmi} (x;y)}$ will increase if ${\displaystyle p(x|y)}$ is fixed but ${\displaystyle p(x)}$ decreases.

Here is an example to illustrate:

Using this table we can marginalize to get the following additional table for the individual distributions:

With this example, we can compute four values for ${\displaystyle \operatorname {pmi} (x;y)}$. Using base-2 logarithms:

${\displaystyle \operatorname {pmi} (x=0;y=0)=-1}$

${\displaystyle \operatorname {pmi} (x=0;y=1)=0.222392}$

${\displaystyle \operatorname {pmi} (x=1;y=0)=1.584963}$

${\displaystyle \operatorname {pmi} (x=1;y=1)=-1.584963}$

(For reference, the mutual information ${\displaystyle \operatorname {I} (X;Y)}$ would then be 0.2141709.)

Pointwise Mutual Information has many of the same relationships as the mutual information. In particular,

${\displaystyle {\begin{aligned}\operatorname {pmi} (x;y)&=&h(x)+h(y)-h(x,y)\\&=&h(x)-h(x\mid y)\\&=&h(y)-h(y\mid x)\end{aligned}}}$

Where ${\displaystyle h(x)}$ is the self-information, or ${\displaystyle -\log _{2}p(x)}$.

Several variations of PMI have been proposed, in particular to address what has been described as its "two main limitations":^[3]

1. PMI can take both positive and negative values and has no fixed bounds, which makes it harder to interpret.^[3]
2. PMI has "a well-known tendency to give higher scores to low-frequency events", but in applications such as measuring word similarity, it is preferable to have "a higher score for pairs of words whose relatedness is supported by more evidence."^[3]

The positive pointwise mutual information (PPMI) measure is defined by setting negative values of PMI to zero:^[2]

${\displaystyle \operatorname {ppmi} (x;y)\equiv \max \left(\log _{2}{\frac {p(x,y)}{p(x)p(y)}},0\right)}$

This definition is motivated by the observation that "negative PMI values (which imply things are co-occurring less often than we would expect by chance) tend to be unreliable unless our corpora are enormous" and also by a concern that "it's not clear whether it's even possible to evaluate such scores of 'unrelatedness' with human judgment".^[2] It also avoids having to deal with ${\displaystyle -\infty }$ values for events that never occur together (${\displaystyle p(x,y)=0}$), by setting PPMI for these to 0.^[2]

Pointwise mutual information can be normalized between [-1,+1] resulting in -1 (in the limit) for never occurring together, 0 for independence, and +1 for complete co-occurrence.^[4]

${\displaystyle \operatorname {npmi} (x;y)={\frac {\operatorname {pmi} (x;y)}{h(x,y)}}}$

Where ${\displaystyle h(x,y)}$ is the joint self-information ${\displaystyle -\log _{2}p(x,y)}$.

The PMI^k measure (for k=2, 3 etc.), which was introduced by Béatrice Daille around 1994, and as of 2011 was described as being "among the most widely used variants", is defined as^[5][3]

${\displaystyle \operatorname {pmi} ^{k}(x;y)\equiv \log _{2}{\frac {p(x,y)^{k}}{p(x)p(y)}}=\operatorname {pmi} (x;y)-(-(k-1)\log _{2}p(x,y))}$

In particular, ${\displaystyle pmi^{1}(x;y)=pmi(x;y)}$. The additional factors of ${\displaystyle p(x,y)}$ inside the logarithm are intended to correct the bias of PMI towards low-frequency events, by boosting the scores of frequent pairs.^[3] A 2011 case study demonstrated the success of PMI³ in correcting this bias on a corpus drawn from English Wikipedia. Taking x to be the word "football", its most strongly associated words y according to the PMI measure (i.e. those maximizing ${\displaystyle pmi(x;y)}$) were domain-specific ("midfielder", "cornerbacks", "goalkeepers") whereas the terms ranked most highly by PMI³ were much more general ("league", "clubs", "england").^[3]

Total correlation is an extension of mutual information to multi-variables. Analogously to the definition of total correlation, the extension of PMI to multi-variables is "specific correlation."^[6] The SI of the results of random variables ${\displaystyle {\boldsymbol {x}}=(x_{1},x_{2},\ldots {},x_{n})}$ is expressed as the following:

${\displaystyle \mathrm {SI} (x_{1},x_{2},\ldots ,x_{n})\equiv \log {\frac {p(x_{1},x_{2},\ldots ,x_{n})}{\prod _{i=1}^{n}p(x_{i})}}=\log p({\boldsymbol {x}})-\log \prod _{i=1}^{n}p\left(x_{i}\right)}$

Like mutual information,^[7] point mutual information follows the chain rule, that is,

${\displaystyle \operatorname {pmi} (x;yz)=\operatorname {pmi} (x;y)+\operatorname {pmi} (x;z|y)}$

This is proven through application of Bayes' theorem:

${\displaystyle {\begin{aligned}\operatorname {pmi} (x;y)+\operatorname {pmi} (x;z|y)&{}=\log {\frac {p(x,y)}{p(x)p(y)}}+\log {\frac {p(x,z|y)}{p(x|y)p(z|y)}}\\&{}=\log \left[{\frac {p(x,y)}{p(x)p(y)}}{\frac {p(x,z|y)}{p(x|y)p(z|y)}}\right]\\&{}=\log {\frac {p(x|y)p(y)p(x,z|y)}{p(x)p(y)p(x|y)p(z|y)}}\\&{}=\log {\frac {p(x,yz)}{p(x)p(yz)}}\\&{}=\operatorname {pmi} (x;yz)\end{aligned}}}$

PMI could be used in various disciplines e.g. in information theory, linguistics or chemistry (in profiling and analysis of chemical compounds).^[8] In computational linguistics, PMI has been used for finding collocations and associations between words. For instance, countings of occurrences and co-occurrences of words in a text corpus can be used to approximate the probabilities ${\displaystyle p(x)}$ and ${\displaystyle p(x,y)}$ respectively. The following table shows counts of pairs of words getting the most and the least PMI scores in the first 50 millions of words in Wikipedia (dump of October 2015)^{[citation needed]} filtering by 1,000 or more co-occurrences. The frequency of each count can be obtained by dividing its value by 50,000,952. (Note: natural log is used to calculate the PMI values in this example, instead of log base 2)

Good collocation pairs have high PMI because the probability of co-occurrence is only slightly lower than the probabilities of occurrence of each word. Conversely, a pair of words whose probabilities of occurrence are considerably higher than their probability of co-occurrence gets a small PMI score.

• Fano, R M (1961). "chapter 2". Transmission of Information: A Statistical Theory of Communications. MIT Press, Cambridge, MA. ISBN 978-0262561693. {{cite book}}: ISBN / Date incompatibility (help)

• Demo at Rensselaer MSR Server (PMI values normalized to be between 0 and 1)