Deep belief network

Short description: Type of artificial neural network

Schematic overview of a deep belief net. Arrows represent directed connections in the graphical model that the net represents.

In machine learning, a deep belief network (DBN) is a generative graphical model, or alternatively a class of deep neural network, composed of multiple layers of latent variables ("hidden units"), with connections between the layers but not between units within each layer.

When trained on a set of examples without supervision, a DBN can learn to probabilistically reconstruct its inputs. The layers then act as feature detectors. After this learning step, a DBN can be further trained with supervision to perform classification.

DBNs can be viewed as a composition of simple, unsupervised networks such as restricted Boltzmann machines (RBMs) or autoencoders, where each sub-network's hidden layer serves as the visible layer for the next. An RBM is an undirected, generative energy-based model with a "visible" input layer and a hidden layer and connections between but not within layers. This composition leads to a fast, layer-by-layer unsupervised training procedure, where contrastive divergence is applied to each sub-network in turn, starting from the "lowest" pair of layers (the lowest visible layer is a training set).

The observation that DBNs can be trained greedily, one layer at a time, led to one of the first effective deep learning algorithms. Overall, there are many attractive implementations and uses of DBNs in real-life applications and scenarios (e.g., electroencephalography, drug discovery).

Training

The training method for RBMs proposed by Geoffrey Hinton for use with training "Product of Experts" models is called contrastive divergence (CD). CD provides an approximation to the maximum likelihood method that would ideally be applied for learning the weights. In training a single RBM, weight updates are performed with gradient descent via the following equation: [math]\displaystyle{ w_{ij}(t+1) = w_{ij}(t) + \eta\frac{\partial \log(p(v))}{\partial w_{ij}} }[/math]

where, [math]\displaystyle{ p(v) }[/math] is the probability of a visible vector, which is given by [math]\displaystyle{ p(v) = \frac{1}{Z}\sum_he^{-E(v,h)} }[/math]. [math]\displaystyle{ Z }[/math] is the partition function (used for normalizing) and [math]\displaystyle{ E(v,h) }[/math] is the energy function assigned to the state of the network. A lower energy indicates the network is in a more "desirable" configuration. The gradient [math]\displaystyle{ \frac{\partial \log(p(v))}{\partial w_{ij}} }[/math] has the simple form [math]\displaystyle{ \langle v_ih_j\rangle_\text{data} - \langle v_ih_j\rangle_\text{model} }[/math] where [math]\displaystyle{ \langle\cdots\rangle_p }[/math] represent averages with respect to distribution [math]\displaystyle{ p }[/math]. The issue arises in sampling [math]\displaystyle{ \langle v_ih_j\rangle_\text{model} }[/math] because this requires extended alternating Gibbs sampling. CD replaces this step by running alternating Gibbs sampling for [math]\displaystyle{ n }[/math] steps (values of [math]\displaystyle{ n = 1 }[/math] perform well). After [math]\displaystyle{ n }[/math] steps, the data are sampled and that sample is used in place of [math]\displaystyle{ \langle v_ih_j\rangle_\text{model} }[/math]. The CD procedure works as follows:

1. Initialize the visible units to a training vector.
2. Update the hidden units in parallel given the visible units: [math]\displaystyle{ p(h_j = 1 \mid \textbf{V}) = \sigma(b_j + \sum_i v_iw_{ij}) }[/math]. [math]\displaystyle{ \sigma }[/math] is the sigmoid function and [math]\displaystyle{ b_j }[/math] is the bias of [math]\displaystyle{ h_j }[/math].
3. Update the visible units in parallel given the hidden units: [math]\displaystyle{ p(v_i = 1 \mid \textbf{H}) = \sigma(a_i + \sum_j h_jw_{ij}) }[/math]. [math]\displaystyle{ a_i }[/math] is the bias of [math]\displaystyle{ v_i }[/math]. This is called the "reconstruction" step.
4. Re-update the hidden units in parallel given the reconstructed visible units using the same equation as in step 2.
5. Perform the weight update: [math]\displaystyle{ \Delta w_{ij} \propto \langle v_ih_j\rangle_\text{data} - \langle v_ih_j\rangle_\text{reconstruction} }[/math].

Once an RBM is trained, another RBM is "stacked" atop it, taking its input from the final trained layer. The new visible layer is initialized to a training vector, and values for the units in the already-trained layers are assigned using the current weights and biases. The new RBM is then trained with the procedure above. This whole process is repeated until the desired stopping criterion is met.

Although the approximation of CD to maximum likelihood is crude (does not follow the gradient of any function), it is empirically effective.

A restricted Boltzmann machine (RBM) with fully connected visible and hidden units. Note there are no hidden-hidden or visible-visible connections.

See also

• Bayesian network
• Convolutional deep belief network
• Deep learning
• Energy based model
• Stacked Restricted Boltzmann Machine

External links

• "Deep Belief Networks". Deep Learning Tutorials. http://deeplearning.net/tutorial/DBN.html. 
• "Deep Belief Network Example". Deeplearning4j Tutorials. http://deeplearning4j.org/deepbeliefnetwork.html.