BrainSlug: Transparent Acceleration of Deep Learning Through Depth-First Parallelism

Nicolas Weber, Florian Schmidt, Mathias Niepert, Felipe Huici
NEC Laboratories Europe
arXiv:1804.08378 [cs.DC], (23 Apr 2018)


   title={BrainSlug: Transparent Acceleration of Deep Learning Through Depth-First Parallelism},

   author={Weber, Nicolas and Schmidt, Florian and Niepert, Mathias and Huici, Felipe},






Download Download (PDF)   View View   Source Source   



Project page: BrainSlug: Transparent Neural Network Acceleration (http://www.brainslug.info/)


Neural network frameworks such as PyTorch and TensorFlow are the workhorses of numerous machine learning applications ranging from object recognition to machine translation. While these frameworks are versatile and straightforward to use, the training of and inference in deep neural networks is resource (energy, compute, and memory) intensive. In contrast to recent works focusing on algorithmic enhancements, we introduce BrainSlug, a framework that transparently accelerates neural network workloads by changing the default layer-by-layer processing to a depth-first approach, reducing the amount of data required by the computations and thus improving the performance of the available hardware caches. BrainSlug achieves performance improvements of up to 41.1% on CPUs and 35.7% on GPUs. These optimizations come at zero cost to the user as they do not require hardware changes and only need tiny adjustments to the software.

Rating: 5.0/5. From 2 votes.
Please wait...

Recent source codes

* * *

* * *

HGPU group © 2010-2018 hgpu.org

All rights belong to the respective authors

Contact us: