设为首页收藏本站language 语言切换
查看: 1431|回复: 1
收起左侧

基于Spark的分布式深度学习框架BigDL开源了!

[复制链接]
发表于 2017-1-23 12:15:12 | 显示全部楼层 |阅读模式

导读
Intel开源了基于Apache Spark的分布式深度学习框架BigDL,BigDL借助现有的Spark集群来运行深度学习计算,并简化存储在Hadoop中的大数据集的数据加载。

在Xeon服务器上运行的测试结果表明,BigDL比其他开源框架Caffe、Torch和TensorFlow有显著的性能提升。BigDL速度可与主流GPU匹敌,而且能够扩展到数十个Xeon服务器,BigDL库支持Spark 1.5、1.6和2.0版本,并容许将深度学习嵌入在现有的Spark程序中,BigDL库中有把Spark RDDs转换为BigDL DataSet的方法,并且可以直接与Spark ML Pipelines一起使用。
对于模型训练,BigDL使用了在多个执行器中执行单个Spark任务的同步小批量SGD(Stochastic Gradient Descent),每个执行器运行一个多线程引擎并处理一部分微批次数据,在当前版本中,所有的训练和验证数据都存储到存储器中。

BigDL使用Scala开发,并参考了Torch的模型;像Torch一样,它有一个使用Intel MKL库进行计算的Tensor类,Intel MKL(Math Kernel Library)是由一系列为计算优化过的小程序所组成的库,这些小程序从FFT(快速傅立叶变换)到矩阵乘法均有涉及,常用于深度学习模型训练,Module是另一个从Torch借鉴而来的概念,它的灵感来自Torch的nn package,Module代表单独的神经网络层、Table和Criterion。
BigDL提供了一个AWS EC2镜像和一些示例,比如使用卷积神经网络进行文本分类,还有图像分类以及如何将在Torch或Caffe中预训练过的模型加载到Spark中进行预测计算,来自社区的请求主要包括提供对Python的支持和MKL-DNN(MKL的深度学习扩展)。

免费提供最新Linux技术教程书籍,为开源技术爱好者努力做得更多更好,开源站点:http://www.linuxprobe.com/

您需要登录后才可以回帖 登录 | 论坛注册

本版积分规则

QQ|Archiver|手机版|小黑屋|sitemap|鸿鹄论坛 ( 京ICP备14027439号 )  

GMT+8, 2025-3-10 08:41 , Processed in 0.067721 second(s), 23 queries , Redis On.  

  Powered by Discuz!

  © 2001-2025 HH010.COM

快速回复 返回顶部 返回列表