Added to Favorites

Popular Searches

In information theory and machine learning, information gain is an alternative synonym for Kullback–Leibler divergence. ## General definition

## Formal definition

{|Ex> bullet H({xin Ex|value(x,a)=v}) ## Drawbacks

Although information gain is usually a good measure for deciding the relevance of an attribute, it is not perfect. A notable problem occurs when information gain is applied to attributes that can take on a large number of distinct values. For example, suppose that we are building a decision tree for some data describing a business's customers. Information gain is often used to decide which of the attributes are the most relevant, so they can be tested near the root of the tree. One of the input attributes might be the customer's credit card number. This attribute has a high information gain, because it uniquely identifies each customer, but we do not want to include it in the decision tree: deciding how to treat a customer based on their credit card number is unlikely to generalize to customers we haven't seen before.## Constructing a decision tree using information gain

## External sources

In particular, the information gain about a random variable X obtained from an observation that a random variable A takes the value A=a is the Kullback-Leibler divergence D_{KL}(p(x|a) || p(x|I) ) of the prior distribution p(x|I) for x from the posterior distribution p(x|a) for x given a.

The expected value of the information gain is the mutual information I(X;A) of X and A — i.e. the reduction in the entropy of X achieved by learning the state of the random variable A.

In machine learning this concept can be used to define a preferred sequence of attributes to investigate to most rapidly narrow down the state of X. Such a sequence (which depends on the outcome of the investigation of previous attributes at each stage) is called a decision tree. Usually an attribute with high information gain should be preferred to other attributes.

In general terms, the expected information gain is the change in information entropy from a prior state to a state that takes some information as given:

$IG(Ex,a)\; =\; H(Ex)\; -\; H(Ex|a)$

Let $Attr$ be the set of all attributes and $Ex$ the set of all training examples, $value(x,a)$ with $xin\; Ex$ defines the value of a specific example $x$ for attribute $ain\; Attr$, $H$ specifies the entropy. The information gain for an attribute $ain\; Attr$ is defined as follows:

$IG(Ex,a)=H(Ex)-sum\_\{vin\; values(a)\}\; frac$

The information gain is equal to the total entropy for an attribute if for each of the attribute values a unique classification can be made for the result attribute. In this case the relative entropies subtracted from the total entropy are 0.

Information gain ratio is sometimes used instead. This biases the decision tree against considering attributes with a large number of distinct values.

A decision tree can be constructed top-down using the information gain in the following way:

- begin at the root node
- determine the attribute with the highest information gain which is not used in an ancestor node
- add a child node for each possible value of that attribute
- attach all examples to the child node where the attribute values of the examples are identical to the attribute value attached to the node
- if all examples attached to the child node can be classified uniquely add that classification to that node and mark it as leaf node
- go back to step two if there is at least one more unused attribute left, otherwise add the classification of most of the examples attached to the child node

- [1] Mitchell, Tom M., Machine Learning. The Mc-Graw-Hill Companies, Inc., 1997

Wikipedia, the free encyclopedia © 2001-2006 Wikipedia contributors (Disclaimer)

This article is licensed under the GNU Free Documentation License.

Last updated on Wednesday July 23, 2008 at 07:19:04 PDT (GMT -0700)

View this article at Wikipedia.org - Edit this article at Wikipedia.org - Donate to the Wikimedia Foundation

This article is licensed under the GNU Free Documentation License.

Last updated on Wednesday July 23, 2008 at 07:19:04 PDT (GMT -0700)

View this article at Wikipedia.org - Edit this article at Wikipedia.org - Donate to the Wikimedia Foundation

Copyright © 2015 Dictionary.com, LLC. All rights reserved.