Differentiable neural computer explained

In artificial intelligence, a differentiable neural computer (DNC) is a memory augmented neural network architecture (MANN), which is typically (but not by definition) recurrent in its implementation. The model was published in 2016 by Alex Graves et al. of DeepMind.

Applications

DNC indirectly takes inspiration from Von-Neumann architecture, making it likely to outperform conventional architectures in tasks that are fundamentally algorithmic that cannot be learned by finding a decision boundary.

So far, DNCs have been demonstrated to handle only relatively simple tasks, which can be solved using conventional programming. But DNCs don't need to be programmed for each problem, but can instead be trained. This attention span allows the user to feed complex data structures such as graphs sequentially, and recall them for later use. Furthermore, they can learn aspects of symbolic reasoning and apply it to working memory. The researchers who published the method see promise that DNCs can be trained to perform complex, structured tasks^{[1] [2]} and address big-data applications that require some sort of reasoning, such as generating video commentaries or semantic text analysis.^[3]

DNC can be trained to navigate rapid transit systems, and apply that network to a different system. A neural network without memory would typically have to learn about each transit system from scratch. On graph traversal and sequence-processing tasks with supervised learning, DNCs performed better than alternatives such as long short-term memory or a neural turing machine.^[4] With a reinforcement learning approach to a block puzzle problem inspired by SHRDLU, DNC was trained via curriculum learning, and learned to make a plan. It performed better than a traditional recurrent neural network.

Architecture

DNC networks were introduced as an extension of the Neural Turing Machine (NTM), with the addition of memory attention mechanisms that control where the memory is stored, and temporal attention that records the order of events. This structure allows DNCs to be more robust and abstract than a NTM, and still perform tasks that have longer-term dependencies than some predecessors such as Long Short Term Memory (LSTM). The memory, which is simply a matrix, can be allocated dynamically and accessed indefinitely. The DNC is differentiable end-to-end (each subcomponent of the model is differentiable, therefore so is the whole model). This makes it possible to optimize them efficiently using gradient descent.^{[5] [6] [7]}

The DNC model is similar to the Von Neumann architecture, and because of the resizability of memory, it is Turing complete.^[8]

Traditional DNC

DNC, as originally published^[1]

Independent variables
x _{t}	Input vector
z _{t}	Target vector
Controller
\boldsymbol\chi _{t} = [ x _{t;} r ^{1 ;} _{t -1} … ; r ^{R]} _{t -1}	Controller input matrix
Deep (layered) LSTM	\forall 0 \leq l\leq L
i ^{l} _{t} = \sigma(W ^{l} _{i} [\boldsymbol\chi _{t;} h ^{l;} _{t -1} h ^{l -1} _{t} ] + b ^{l)} _{i}	Input gate vector
o ^{l} _{t} = \sigma(W ^{l} _{o} [\boldsymbol\chi _{t;} h ^{l;} _{t -1} h ^{l -1} _{t} ] + b ^{l)} _{o}	Output gate vector
f ^{l} _{t} = \sigma(W ^{l} _{f} [\boldsymbol\chi _{t;} h ^{l;} _{t -1} h ^{l -1} _{t} ] + b ^{l)} _{f}	Forget gate vector
s ^{l} _{t} = f ^{l} _{t} s ^{l} _{t -1} + i ^{l} _{s} [\boldsymbol\chi _{t;} h ^{l;} _{t -1} h ^{l -1} _{t} ] + b ^{l)} _{s}	State gate vector, s _{0} = 0
h ^{l} _{t} = o ^{l} _{t} \tanh( s ^{l)} _{t}	Hidden gate vector, h _{0=0;} h ^{0=0 \forall t} _{t}
y _{t=W} _{y[h} ^{L] + W} ^{r[ r} ^{R]} ^{t}	DNC output vector
Read & Write heads
\xi _{t} = W _{\xi[h} ^{L]} ^{t}	Interface parameters
= [ k ^{r, 1} _{t} ; … ; k ^{r,R} _{t} ;\hat{\beta} ^{r, 1} _{t} ; … ;\hat{\beta} ^{r,R} _{t} ; k ^{w}; \hat{e} _{t} }_t;\mathbf_t;\hat;\cdots;\hat;\hat_t^a;\hat_t^w;\hat_t^1;\cdots;\hat_t^R]
Read heads	\forall 1 \leq i\leq R
k ^{r,i} _{t}	Read keys
\beta ^{r,i} _{t} = oneplus (\hat{\beta} ^{r,i} _{t} )	Read strengths
f ^{i)} _{t}	Free gates
\boldsymbol\pi ^{i)} _{t}	Read modes, \boldsymbol\pi ^{i\in R} _{t} ^{3}
Write head
k ^{w} _{t}	Write key
\beta ^{w} _{t}	Write strength
e _{t=\sigma(\hat{e} }_t)	Erase vector
v _{t}	Write vector
g ^{a)} _{t}	Allocation gate
g ^{w)} _{t}	Write gate
Memory
M _{t=M} _{t-1} \circ(E - w ^{\intercal} _{t}	Memory matrix, Matrix of ones E\in R ^{N x}
u _{t=(u} _{t-1} + w ^{w - u} _{t -1} \circ w ^{w)\circ\boldsymbol\psi} _{t}	Usage vector
p _{t=\left(1-\sum} _{iw} ^{w[i]\right) p} ^{t -1} + w ^{w} _{t}	Precedence weighting, p _{0=0}
L _{t=(1} - I )\left[( 1- w ^{j)L} _{t -1} [i,j] + w ^{j\right]} _{t -1}	Temporal link matrix, L _{0=0}
w ^{a a} _{t + ( 1- g} ^{w]} ^{t}	Write weighting
w ^{r,i} _{t} = \boldsymbol\pi ^{r,i} _{t} + \boldsymbol\pi ^{i} _{t}	Read weighting
r ^{r,i} _{t}	Read vectors
l{C}(M, k ,\beta)[i]= (\exp\{ l{D)/(( k ,M[i, ⋅ ])\beta\}}{\sum) _{j\exp\{ l{D}( k ,M[j, ⋅ ])\beta\}}}	Content-based addressing, Lookup key k , key strength \beta
\phi _{t}	Indices of u _{t} , sorted in ascending order of usage
a _{t[\phi} _{t[j]]=(1-u} _{t[\phi} _{t[j]])\prod} ^{j -1} ^{i =1} u _{t[\phi} _{t[i]]}	Allocation weighting
c ^{w = l{C}(M} _{t -1} , k ^{w)} _{t}	Write content weighting
c ^{r,i} _{t} = l{C}(M _{t-1} , k ^{r,i} _{t} ,\beta ^{r,i} _{t} )	Read content weighting
f ^{i = L} _{t w} ^{r,i} ^{t -1}	Forward weighting
b ^{r,i} _{t -1}	Backward weighting
\boldsymbol\psi _{t=\prod} ^{r,i} ^{t -1} \right)	Memory retention vector
Definitions
W , b	Weight matrix, bias vector
0 , 1 , I	Zeros matrix, ones matrix, identity matrix
\circ	Element-wise multiplication
l{D}( u , v )= (u ⋅ v)/(\| u \|\| v \|)	Cosine similarity
\sigma(x) =1 /( 1+ e ^{-x} )	Sigmoid function
oneplus (x) =1+log ( 1+ e ^{x)}	Oneplus function
softmax ( x ) _{j} = (e ^{x _{j}})/(\sum ^{K} _{k =1} e ^{x _{k}})    for j = 1, ..., K.	Softmax function

Extensions

Refinements include sparse memory addressing, which reduces time and space complexity by thousands of times. This can be achieved by using an approximate nearest neighbor algorithm, such as Locality-sensitive hashing, or a random k-d tree like Fast Library for Approximate Nearest Neighbors from UBC.^[9] Adding Adaptive Computation Time (ACT) separates computation time from data time, which uses the fact that problem length and problem difficulty are not always the same.^[10] Training using synthetic gradients performs considerably better than Backpropagation through time (BPTT).^[11] Robustness can be improved with use of layer normalization and Bypass Dropout as regularization.^[12]

See also

• Differentiable programming

External links

• A bit-by-bit guide to the equations governing differentiable neural computers
• DeepMind's Differentiable Neural Network Thinks Deeply

Notes and References

1. Graves. Alex. Wayne. Greg. Reynolds. Malcolm. Harley. Tim. Danihelka. Ivo. Grabska-Barwińska. Agnieszka. Colmenarejo. Sergio Gómez. Grefenstette. Edward. Ramalho. Tiago. 2016-10-12. Hybrid computing using a neural network with dynamic external memory. Nature. en. 538. 7626. 10.1038/nature20101. 1476-4687. 471–476. 27732574. 2016Natur.538..471G. 205251479.
2. Web site: Differentiable neural computers DeepMind. DeepMind. 12 October 2016 . 2016-10-19.
3. Jaeger. Herbert. 2016-10-12. Artificial intelligence: Deep neural reasoning. Nature. en. 538. 7626. 10.1038/nature19477. 1476-4687. 467–468. 27732576. 2016Natur.538..467J. free.
4. Web site: DeepMind's Differentiable Neural Network Thinks Deeply. James. Mike. www.i-programmer.info. 2016-10-20.
5. News: DeepMind's AI learned to ride the London Underground using human-like reason and memory. Burgess. Matt. WIRED UK. en-GB. 2016-10-19.
6. News: DeepMind AI 'Learns' to Navigate London Tube. PCMAG. 2016-10-19.
7. Web site: DeepMind's differentiable neural computer helps you navigate the subway with its memory. Mannes. John. TechCrunch. 13 October 2016 . 2016-10-19.
8. Web site: RNN Symposium 2016: Alex Graves - Differentiable Neural Computer. YouTube.
9. 1610.09027. Scaling Memory-Augmented Neural Networks with Sparse Reads and Writes. Jack W Rae. Jonathan J Hunt. Harley. Tim. Danihelka. Ivo. Senior. Andrew. Wayne. Greg. Graves. Alex. Timothy P Lillicrap. cs.LG. 2016.
10. 1603.08983. Adaptive Computation Time for Recurrent Neural Networks. Graves. Alex. cs.NE. 2016.
11. 1608.05343. Decoupled Neural Interfaces using Synthetic Gradients. Jaderberg. Max. Wojciech Marian Czarnecki. Osindero. Simon. Vinyals. Oriol. Graves. Alex. Silver. David. Kavukcuoglu. Koray. cs.LG. 2016.
12. 1807.02658. Robust and Scalable Differentiable Neural Computer for Question Answering. Franke. Jörg. Niehues. Jan. Waibel. Alex. cs.CL. 2018.