Purine: A bi-graph based deep learning framework

Min Lin, Shuo Li, Xuan Luo, Shuicheng Yan
Graduate School of integrated Sciences and Engineering, National University of Singapore
arXiv:1412.6249 [cs.NE], (19 Dec 2014)



Download Download (PDF)   View View   Source Source   



In this paper, we introduce a novel deep learning framework, termed Purine. In Purine, a deep network is expressed as a bipartite graph (bi-graph), which is composed of interconnected operators and data tensors. With the bi-graph abstraction, networks are easily solvable with event-driven task dispatcher. We then demonstrate that different parallelism schemes over GPUs and/or CPUs on single or multiple PCs can be universally implemented by graph composition. Scheduled by the task dispatcher, memory transfers are fully overlapped with other computations, which greatly reduce the communication overhead and help us achieve approximate linear acceleration. This eases researchers from coding for various parallelization schemes, and the same dispatcher can be used for solving variant graphs.
No votes yet.
Please wait...

* * *

* * *

HGPU group © 2010-2021 hgpu.org

All rights belong to the respective authors

Contact us: