arXiv · 1412.6249
Purine: A bi-graph based deep learning framework
Abstract
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. This eases researchers from coding for various parallelization schemes, and the same dispatcher can be used for solving variant graphs. 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.
Explore related subjects
Keep this discovery
Min Lin, Shuo Li, Xuan Luo, Shuicheng Yan. 2014-12-19. Purine: A bi-graph based deep learning framework. https://arxiv.org/abs/1412.6249
Cite the original work for its findings. Save a collection to share your selection of sources.