We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

math

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Parallel Complexity of Forward and Backward Propagation

Authors: Maxim Naumov
Abstract: We show that the forward and backward propagation can be formulated as a solution of lower and upper triangular systems of equations. For standard feedforward (FNNs) and recurrent neural networks (RNNs) the triangular systems are always block bi-diagonal, while for a general computation graph (directed acyclic graph) they can have a more complex triangular sparsity pattern. We discuss direct and iterative parallel algorithms that can be used for their solution and interpreted as different ways of performing model parallelism. Also, we show that for FNNs and RNNs with $k$ layers and $\tau$ time steps the backward propagation can be performed in parallel in O($\log k$) and O($\log k \log \tau$) steps, respectively. Finally, we outline the generalization of this technique using Jacobians that potentially allows us to handle arbitrary layers.
Comments: 18 pages
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Numerical Analysis (math.NA)
MSC classes: 68T05 (Primary) 65F99, 15B99 (Secondary)
ACM classes: I.2.6; I.5.0
Cite as: arXiv:1712.06577 [cs.LG]
  (or arXiv:1712.06577v1 [cs.LG] for this version)

Submission history

From: Maxim Naumov [view email]
[v1] Mon, 18 Dec 2017 18:46:51 GMT (14kb)

Link back to: arXiv, form interface, contact.