11.2 Missing Data

When data is missing some values for some features, the missing data cannot be ignored. Example 10.13 gives an example where ignoring missing data leads to wrong conclusions. Making inference from missing data is a causality problem, as it cannot be solved by observation, but requires a causal model.

A missingness graph, or m-graph for short, is used to model data where some values might be missing. Start with a belief network model of the domain. In the m-graph, all variables in the original graph exist with the same parents. For each variable V that could be observed with some values missing, the m-graph contains two extra variables:

  • •

    M⁢_⁢V, a Boolean variable that is true when V’s value is missing. The parents of this node can be whatever variables the missingness is assumed to depend on.

  • •

    A variable V∗, with domain d⁢o⁢m⁢(V)∪{m⁢i⁢s⁢s⁢i⁢n⁢g}, where m⁢i⁢s⁢s⁢i⁢n⁢g is a new value (not in the domain of V). The only parents of V∗ are V and M⁢_⁢V. The conditional probability table contains only 0 and 1, with the 1s being

    P⁢(V∗=m⁢i⁢s⁢s⁢i⁢n⁢g∣M⁢_⁢V=t⁢r⁢u⁢e)=1
    P⁢(V∗=v∣M⁢_⁢V=f⁢a⁢l⁢s⁢e∧V=v)=1.

If the value of V is observed to be v, then V∗=v is conditioned on. If the value for V is missing, V∗=m⁢i⁢s⁢s⁢i⁢n⁢g is conditioned on. Note that V∗ is always observed and conditioned on, and V is never conditioned on, in this augmented model. When modeling a domain, the parents of M⁢_⁢V specify what the missingness depends on.

Example 11.5.

Example 10.13 gives a problematic case of a drug that just makes people sicker and so drop out, giving missing data. A graphical model for it is shown in Figure 11.5.

Refer to caption
Figure 11.5: Missingness graph for Example 11.5

Assume T⁢a⁢k⁢e⁢_⁢d⁢r⁢u⁢g is Boolean and the domains of S⁢i⁢c⁢k⁢_⁢b⁢e⁢f⁢o⁢r⁢e and S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r are {well, s⁢i⁢c⁢k, very_sick}. Then the domain of S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r∗ is {well, s⁢i⁢c⁢k, v⁢e⁢r⁢y⁢_⁢s⁢i⁢c⁢k, missing}. The variable M⁢_⁢S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r is Boolean.

Suppose there is a dataset from which to learn, with S⁢i⁢c⁢k⁢_⁢b⁢e⁢f⁢o⁢r⁢e and T⁢a⁢k⁢e⁢_⁢d⁢r⁢u⁢g observed for each example, and some examples have S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r observed and some have it missing. To condition the m-graph on an example, all of the variables except S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r are conditioned on. S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r∗ has the value of S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r when it is observed, and has value m⁢i⁢s⁢s⁢i⁢n⁢g otherwise.

You might think that you can learn the missing data using expectation maximization (EM), with S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r as a hidden variable. There are, however, many probability distributions that are consistent with the data. All of the missing cases could have value w⁢e⁢l⁢l for S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r, or they all could be v⁢e⁢r⁢y⁢_⁢s⁢i⁢c⁢k; you can’t tell from the data. EM can converge to any one of these distributions that are consistent with the data. Thus, although EM may converge, it does not converge to something that makes predictions that can be trusted.

To determine appropriate model parameters, one should find some data about the relationship between S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r and M⁢_⁢S⁢i⁢c⁢k⁢_⁢a⁢f⁢t⁢e⁢r. When doing a human study, the designers of the study need to try to find out why people dropped out of the study. These cases cannot just be ignored.

A distribution is recoverable or identifiable from missing data if the distribution can be accurately measured from the data, even with parts of the data missing. Whether a distribution is recoverable is a property of the underlying graph. A distribution that is not recoverable cannot be reconstructed from observational data, no matter how large the dataset. The distribution in Example 11.5 is not recoverable.

Data is missing completely at random (MCAR) if V and M⁢_⁢V are independent. If the data is missing completely at random, the examples with missing values can be ignored. This is a strong assumption that rarely occurs in practice, but is often implicitly assumed when missingness is ignored.

A weaker assumption is that a variable Y is missing at random (MAR), which occurs when Y is independent of M⁢_⁢Y given the observed variables Vo. That is, when P⁢(Y∣Vo,M⁢_⁢Y)=P⁢(Y∣Vo). This occurs when the reason the data is missing can be observed. The distribution over Y and the observed variables is recoverable by P⁢(Y,Vo)=P⁢(Y∣Vo,M⁢_⁢Y=f⁢a⁢l⁢s⁢e)⁢P⁢(Vo). Thus, the non-missing data is used to estimate P⁢(Y∣Vo) and all of the data is used to estimate P⁢(Vo).

Example 11.6.

Suppose you have a dataset of education and income, where the income values are often missing, and have modeled that income depends on education. You want to learn the joint probability of I⁢n⁢c⁢o⁢m⁢e and E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n.

If income is missing completely at random, shown in Figure 11.6(a), the missing data can be ignored when learning the probabilities:

P⁢(I⁢n⁢c⁢o⁢m⁢e,E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n)=P⁢(I⁢n⁢c⁢o⁢m⁢e∗,E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n∣M⁢_⁢I⁢n⁢c⁢o⁢m⁢e=f⁢a⁢l⁢s⁢e)

since M⁢_⁢I⁢n⁢c⁢o⁢m⁢e is independent of I⁢n⁢c⁢o⁢m⁢e and E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n.

Refer to caption
Figure 11.6: Missing data. E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n is observed but I⁢n⁢c⁢o⁢m⁢e might have missing values: (a) completely at random, (b) missing at random, (c) missing not at random

If income is missing at random, shown in Figure 11.6(b), the missing data cannot be ignored when learning the probabilities, however

P( Income,Education)
=P⁢(I⁢n⁢c⁢o⁢m⁢e∣E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n)∗P⁢(E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n)
=P⁢(I⁢n⁢c⁢o⁢m⁢e∣E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n∧M⁢_⁢I⁢n⁢c⁢o⁢m⁢e=f⁢a⁢l⁢s⁢e)∗P⁢(E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n)
=P⁢(I⁢n⁢c⁢o⁢m⁢e∗∣E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n∧M⁢_⁢I⁢n⁢c⁢o⁢m⁢e=f⁢a⁢l⁢s⁢e)∗P⁢(E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n).

Both of these can be estimated from the data. The first probability can ignore the examples with I⁢n⁢c⁢o⁢m⁢e missing, and the second cannot.

If I⁢n⁢c⁢o⁢m⁢e is missing not at random, as shown in Figure 11.6(c), which is similar to Figure 11.5, the probability P⁢(I⁢n⁢c⁢o⁢m⁢e,E⁢d⁢u⁢c⁢a⁢t⁢i⁢o⁢n) cannot be learned from data, because there is no way to determine whether those who don’t report income are those with very high income or very low income. While algorithms like EM converge, what they learn is fiction, converging to one of the many possible hypotheses about how the data could be missing.

The main points to remember are:

  • •

    You cannot learn from missing data without making modeling assumptions.

  • •

    Some distributions are not recoverable from missing data, and some are. It depends on the independence structure of the underlying graph.

  • •

    If the distribution is not recoverable, a learning algorithm still may be able to learn parameters, but the resulting distributions should not be trusted.