My watch list
my.bionity.com

Oja's rule

Oja's Learning Rule, or simply Oja's rule, is a model of how neurons in the brain or in artificial neural networks change connection strength, or learn, over time. It is a modification of the standard Hebb's Rule (see Hebbian learning) that, through multiplicative normalization, solves all stability problems and generates an algorithm for principal components analysis. This is a computational form of an effect which is believed to happen in biological neurons.

Theory

Oja's rule requires a number of simplifications to derive, but in its final form it is demonstrably stable, unlike Hebb's rule. It is a single-neuron special case of the Generalized Hebbian Algorithm. However, Oja's rule can also be generalized in other ways to varying degrees of stability and success.

Formula

Oja's rule defines the change in presynaptic weights $\textbf{w}$ given the output response y of a neuron to its inputs $\textbf{x}$, or

$\,\Delta \textbf{w} = \textbf{w}(n+1)-\textbf{w}(n) = \eta y(n) (\textbf{x}(n) - y(n)\textbf{w}(n))$,

where η is the learning rate which can also change with time. Note that the bold symbols are vectors and n defines a discrete time iteration. The rule can also be made for continuous iterations as

$\,\frac{d\textbf{w}}{d t} = \eta y(t) (\textbf{x}(t) - y(t)\textbf{w}(t))$.

Derivation

The simplest learning rule known is Hebb's rule, which states in conceptual terms that neurons that fire together, wire together. In component form as a difference equation, it is written

$\,w_i (n+1) = w_i (n) + \eta y(\textbf{x}(n)) x_i (n)$,

where $y(\textbf{x}(n))$ is again the output, this time explicitly dependent on its input vector $\textbf{x}$.

Hebb's rule has synaptic weights approaching infinity with a positive learning rate. We can stop this by normalizing the weights so that each weight's magnitude is restricted between 0, corresponding to no weight, and 1, corresponding to being the only input neuron with any weight. Mathematically, this takes the form

$\,w_i (n+1) = \frac{w_i (n) + \eta y(\textbf{x}(n)) x_i (n)}{(\sum_{j=1}^m [w_j (n) + \eta y(\textbf{x}(n)) x_j (n)]^p)^{1/p}}$.

Note that in Oja's original paper[1], p = 2, corresponding to a sum in quadrature, which is the familiar normalization rule. However, any type of normalization, even linear, will give the same result without loss of generality.

Our next step is to expand this into a Taylor series for a small learning rate $| \eta | \ll 1$, giving

$\,w_i (n+1) = \frac{w_i (n)}{(\sum_j w_j^p)^{1/p}} + \eta \left( \frac{y(n) x_i (n)}{(\sum_j w_j^p)^{1/p}} - \frac{w_i (n)\sum_j y(n) x_j (n) w_j (n)}{(\sum_j w_j^p)^{1 + 1/p}} \right) + O(\eta^2)$.

For small η, our higher order terms (O2)) go to zero (see Big-O notation). We again make the specification of a linear neuron, that is, the output of the neuron is equal to the sum of the product of each input and its synaptic weight, or

$\,y(\textbf{x}(n)) = \sum_{j=1}^m x_j (n) w_j (n)$.

We also specify that our weights normalize to 1, which will be a necessary condition for stability, so

$\,\| \textbf{w} \| = (\sum_{j=1}^m w_j^p)^{1/p} = 1$,

which, when substituted into our expansion, gives Oja's rule, or

$\,w_i (n+1) = w_i (n) + \eta y(n) (x_i (n) - w_i (n) y(n))$.

Stability and PCA

In analyzing the convergence of a single neuron evolving by Oja's rule, one extracts the first principal component, or feature, of a data set. Furthermore, with extensions using the Generalized Hebbian Algorithm, one can create a multi-Oja neural network that can extract as many features as desired, allowing for principal components analysis.

A principal component $\,a_j$ is extracted from a dataset $\textbf{x}$ through some associated vector $\textbf{q}_j$, or $a_j = \textbf{q}_j \cdot \textbf{x}$, and we can restore our original dataset by taking

$\textbf{x}=\sum_j a_j \textbf{q}_j$.

In the case of a single neuron trained by Oja's rule, we find the weight vector converges to $\textbf{q}_1$, or the first principal component, as time or number of iterations approaches infinity. We can also define, given a set of input vectors $\,X_i$, that its correlation matrix $\,R_{ij}=X_i X_j$ has an associated eigenvector given by $\textbf{q}_j$ with eigenvalue $\,\lambda_j$. The variance of outputs of our Oja neuron $\sigma^2(n)=\langle y^2 (n) \rangle$ then converges with time iterations to the principal eigenvalue, or

$\lim_{n\rightarrow\infty} \sigma^2(n) = \lambda_1$.

These results are derived using Lyapunov function analysis, and they show that Oja's neuron necessarily converges on strictly the first principal component if certain conditions are met in our original learning rule. Most importantly, our learning rate $\,\eta$ is allowed to vary with time, but only such that its sum is divergent but its power sum is convergent, that is

$\sum_{n=1}^\infty \eta(n) = \infty, ~ \sum_{n=1}^\infty \eta(n)^p < \infty, ~ p>1$.

Our output activation function $y(\textbf{x}(n))$ is also allowed to be nonlinear and nonstatic, but it must be continuously differentiable in both $\textbf{x}$ and $\textbf{w}$ and have derivatives bounded in time. [2]

Applications

Oja's rule was originally described in Oja's 1982 paper[1], but the principle of self-organization to which it is applied is first attributed to Alan Turing in 1952[2]. PCA has also had a long history of use before Oja's rule formalized its use in network computation in 1989. The model can thus be applied to any problem of self-organizing mapping, in particular those in which feature extraction is of primary interest. Therefore, Oja's rule has an important place in image and speech processing. It is also useful as it expands easily to higher dimensions of processing, thus being able to integrate multiple outputs quickly. A canonical example is its use in binocular vision[3].

Biology and Oja's Subspace rule

There is clear evidence for both long-term potentiation and long-term depression in biological neural networks, along with a normalization effect in both input weights and neuron outputs. However, while as of yet there is no direct experimental evidence of Oja's rule active in a biological neural network, a biophysical derivation of a generalization of the rule is possible. Such a derivation requires retrograde signalling from the postsynaptic neuron, which is biologically plausible (see neural backpropagation), and takes the form of

$\Delta w_{ij} \propto \langle x_i y_j \rangle - \epsilon \left\langle (c_\textrm{pre} * \sum_k w_{ik} y_k)\cdot (c_\textrm{post} * y_j)\right\rangle$,

where as before wij is the synaptic weight between the ith input and jth output neurons, x is the input, y is the postsynaptic output, and we define ε to be a constant analogous the learning rate, and cpre and cpost are presynaptic and postsynaptic functions that model the weakening of signals over time. Note that the angle brackets denote the average and the * operator is a convolution. By taking the pre- and post-synaptic functions into frequency space and combining integration terms with the convolution, we find that this gives an arbitrary-dimensional generalization of Oja's rule known as Oja's Subspace[4], namely

$\Delta w \propto C x\cdot w - w\cdot C y$[5].