English

DeepSpark: A Spark-Based Distributed Deep Learning Framework for Commodity Clusters

Machine Learning 2016-10-04 v3

Abstract

The increasing complexity of deep neural networks (DNNs) has made it challenging to exploit existing large-scale data processing pipelines for handling massive data and parameters involved in DNN training. Distributed computing platforms and GPGPU-based acceleration provide a mainstream solution to this computational challenge. In this paper, we propose DeepSpark, a distributed and parallel deep learning framework that exploits Apache Spark on commodity clusters. To support parallel operations, DeepSpark automatically distributes workloads and parameters to Caffe/Tensorflow-running nodes using Spark, and iteratively aggregates training results by a novel lock-free asynchronous variant of the popular elastic averaging stochastic gradient descent based update scheme, effectively complementing the synchronized processing capabilities of Spark. DeepSpark is an on-going project, and the current release is available at http://deepspark.snu.ac.kr.

Keywords

Cite

@article{arxiv.1602.08191,
  title  = {DeepSpark: A Spark-Based Distributed Deep Learning Framework for Commodity Clusters},
  author = {Hanjoo Kim and Jaehong Park and Jaehee Jang and Sungroh Yoon},
  journal= {arXiv preprint arXiv:1602.08191},
  year   = {2016}
}
R2 v1 2026-06-22T12:58:18.866Z