about
Enabling Massive Deep Neural Networks with the GraphBLAS[pdf] (arxiv.org)
3 points by gfredtech on Aug 11, 2017 | hide | past | pdf | discuss on HN

In plain words: A neural network's math is rewritten to run on a library built for sparse matrices, which store only the nonzero weights so giant networks fit in memory. Compared with the usual dense math library, it ran faster as the weight matrices grew sparser.

Abstract · Enabling Massive Deep Neural Networks with the GraphBLAS

Deep Neural Networks (DNNs) have emerged as a core tool for machine learning. The computations performed during DNN training and inference are dominated by operations on the weight matrices describing the DNN. As DNNs incorporate more stages and more nodes per stage, these weight matrices may be required to be sparse because of memory limitations. The GraphBLAS.org math library standard was developed to provide high performance manipulation of sparse weight matrices and input/output vectors. For sufficiently sparse matrices, a sparse matrix library requires significantly less memory than the corresponding dense matrix implementation. This paper provides a brief description of the mathematics underlying the GraphBLAS. In addition, the equations of a typical DNN are rewritten in a form designed to use the GraphBLAS. An implementation of the DNN is given using a preliminary GraphBLAS C library. The performance of the GraphBLAS implementation is measured relative to a standard dense linear algebra library implementation. For various sizes of DNN weight matrices, it is shown that the GraphBLAS sparse implementation outperforms a BLAS dense implementation as the weight matrix becomes sparser.

Jeremy Kepner, Manoj Kumar, José Moreira, Pratap Pattnaik, Mauricio Serrano, Henry Tufo
arXiv:1708.02937 · cs.DC, cs.LG · submitted Aug 9, 2017
abstract · pdf · html · 10 pages, 7 figures, to appear in the 2017 IEEE High Performance Extreme Computing (HPEC) conference

add comment on HN