TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

雷锋网按:本文作者黄文坚,PPmoney 大数据算法总监,《 TensorFlow 实战》作者。本文节选自《 TensorFlow 实战》第二章。

在下周二(2月28日)雷锋网硬创公开课栏目将会推出黄文坚老师的公开课《深度学习之经典卷积神经网络的技术浅析》(点击了解详情),欢迎大家报名!

Google 近日发布了 TensorFlow 1.0 候选版,这第一个稳定版将是深度学习框架发展中的里程碑的一步。自 TensorFlow 于 2015 年底正式开源,距今已有一年多,这期间 TensorFlow 不断给人以惊喜。在这一年多时间,TensorFlow 已从初入深度学习框架大战的新星,成为了几近垄断的行业事实标准。

主流深度学习框架对比

深度学习研究的热潮持续高涨,各种开源深度学习框架也层出不穷,其中包括 TensorFlow、Caffe、Keras、CNTK、Torch7、MXNet、Leaf、Theano、DeepLearning4、Lasagne、Neon 等等。然而 TensorFlow 却杀出重围,在关注度和用户数上都占据绝对优势,大有一统江湖之势。表 2-1 所示为各个开源框架在GitHub上的数据统计(数据统计于 2017 年 1 月 3 日),可以看到 TensorFlow 在 star 数量、fork 数量、contributor 数量这三个数据上都完胜其他对手。

究其原因,主要是 Google 在业界的号召力确实强大,之前也有许多成功的开源项目,以及 Google 强大的人工智能研发水平,都让大家对 Google 的深度学习框架充满信心,以至于 TensorFlow 在 2015 年 11 月刚开源的第一个月就积累了 10000+ 的 star 。其次,TensorFlow 确实在很多方面拥有优异的表现,比如设计神经网络结构的代码的简洁度,分布式深度学习算法的执行效率,还有部署的便利性,都是其得以胜出的亮点。如果一直关注着 TensorFlow 的开发进度,就会发现基本上每星期 TensorFlow 都会有1万行以上的代码更新,多则数万行。产品本身优异的质量、快速的迭代更新、活跃的社区和积极的反馈,形成了良性循环,可以想见 TensorFlow 未来将继续在各种深度学习框架中独占鳌头。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

表2-1 各个开源框架在 GitHub 上的数据统计

观察表2-1还可以发现,Google、Microsoft、Facebook 等巨头都参与了这场深度学习框架大战,此外,还有毕业于伯克利大学的贾扬清主导开发的 Caffe,蒙特利尔大学 Lisa Lab 团队开发的 Theano,以及其他个人或商业组织贡献的框架。另外,可以看到各大主流框架基本都支持 Python,目前 Python 在科学计算和数据挖掘领域可以说是独领风骚。虽然有来自 R、Julia 等语言的竞争压力,但是 Python 的各种库实在是太完善了,Web 开发、数据可视化、数据预处理、数据库连接、爬虫等无所不能,有一个完美的生态环境。仅在数据挖据工具链上,Python 就有 NumPy、SciPy、Pandas、Scikit-learn、XGBoost 等组件,做数据采集和预处理都非常方便,并且之后的模型训练阶段可以和 TensorFlow 等基于 Python 的深度学习框架完美衔接。

表 2-1 和图 2-1 所示为对主流的深度学习框架 TensorFlow、Caffe、CNTK、Theano、Torch 在各个维度的评分,本书 2.2 节会对各个深度学习框架进行比较详细的介绍。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

表2-2 主流深度学习框架在各个维度的评分

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-1 主流深度学习框架对比图

各深度学习框架简介

在本节,我们先来看看目前各流行框架的异同,以及各自的特点和优势。

TensorFlow

TensorFlow 是相对高阶的机器学习库,用户可以方便地用它设计神经网络结构,而不必为了追求高效率的实现亲自写 C++或 CUDA 代码。它和 Theano 一样都支持自动求导,用户不需要再通过反向传播求解梯度。其核心代码和 Caffe 一样是用 C++编写的,使用 C++简化了线上部署的复杂度,并让手机这种内存和CPU资源都紧张的设备可以运行复杂模型(Python 则会比较消耗资源,并且执行效率不高)。除了核心代码的 C++接口,TensorFlow 还有官方的 Python、Go 和 Java 接口,是通过 SWIG(Simplified Wrapper and Interface Generator)实现的,这样用户就可以在一个硬件配置较好的机器中用 Python进行实验,并在资源比较紧张的嵌入式环境或需要低延迟的环境中用 C++部署模型。SWIG 支持给 C/C++代码提供各种语言的接口,因此其他脚本语言的接口未来也可以通过 SWIG 方便地添加。不过使用 Python 时有一个影响效率的问题是,每一个 mini-batch 要从 Python 中 feed 到网络中,这个过程在 mini-batch 的数据量很小或者运算时间很短时,可能会带来影响比较大的延迟。现在 TensorFlow 还有非官方的 Julia、Node.js、R 的接口支持,地址如下。

Julia: http://github.com/malmaud/TensorFlow.jl

Node.js: http://github.com/node-tensorflow/node-tensorflow

R: http://github.com/rstudio/tensorflow

TensorFlow 也有内置的 TF.Learn 和 TF.Slim 等上层组件可以帮助快速地设计新网络,并且兼容 Scikit-learn estimator 接口,可以方便地实现 evaluate、grid search、cross validation 等功能。同时 TensorFlow 不只局限于神经网络,其数据流式图支持非常自由的算法表达,当然也可以轻松实现深度学习以外的机器学习算法。事实上,只要可以将计算表示成计算图的形式,就可以使用 TensorFlow 。用户可以写内层循环代码控制计算图分支的计算,TensorFlow 会自动将相关的分支转为子图并执行迭代运算。TensorFlow 也可以将计算图中的各个节点分配到不同的设备执行,充分利用硬件资源。定义新的节点只需要写一个 Python 函数,如果没有对应的底层运算核,那么可能需要写 C++或者 CUDA 代码实现运算操作。

在数据并行模式上,TensorFlow 和 Parameter Server 很像,但 TensorFlow 有独立的 Variable node,不像其他框架有一个全局统一的参数服务器,因此参数同步更自由。TensorFlow 和 Spark 的核心都是一个数据计算的流式图,Spark 面向的是大规模的数据,支持 SQL 等操作,而 TensorFlow 主要面向内存足以装载模型参数的环境,这样可以最大化计算效率。

TensorFlow 的另外一个重要特点是它灵活的移植性,可以将同一份代码几乎不经过修改就轻松地部署到有任意数量 CPU 或 GPU 的 PC、服务器或者移动设备上。相比于 Theano,TensorFlow 还有一个优势就是它极快的编译速度,在定义新网络结构时,Theano 通常需要长时间的编译,因此尝试新模型需要比较大的代价,而 TensorFlow 完全没有这个问题。TensorFlow 还有功能强大的可视化组件 TensorBoard,能可视化网络结构和训练过程,对于观察复杂的网络结构和监控长时间、大规模的训练很有帮助。TensorFlow 针对生产环境高度优化,它产品级的高质量代码和设计都可以保证在生产环境中稳定运行,同时一旦 TensorFlow 广泛地被工业界使用,将产生良性循环,成为深度学习领域的事实标准。

除了支持常见的网络结构(卷积神经网络(Convolutional Neural Network,CNN)、循环神经网络(Recurent Neural Network,RNN))外,TensorFlow 还支持深度强化学习乃至其他计算密集的科学计算(如偏微分方程求解等)。TensorFlow 此前不支持 symbolic loop,需要使用 Python 循环而无法进行图编译优化,但最近新加入的 XLA 已经开始支持 JIT 和 AOT,另外它使用 bucketing trick 也可以比较高效地实现循环神经网络。TensorFlow 的一个薄弱地方可能在于计算图必须构建为静态图,这让很多计算变得难以实现,尤其是序列预测中经常使用的 beam search。

TensorFlow 的用户能够将训练好的模型方便地部署到多种硬件、操作系统平台上,支持 Intel 和 AMD 的 CPU,通过 CUDA 支持 NVIDIA 的 GPU (最近也开始通过 OpenCL 支持 AMD 的 GPU,但没有 CUDA 成熟),支持 Linux 和 Mac,最近在 0.12 版本中也开始尝试支持 Windows。在工业生产环境中,硬件设备有些是最新款的,有些是用了几年的老机型,来源可能比较复杂,TensorFlow 的异构性让它能够全面地支持各种硬件和操作系统。同时,其在 CPU 上的矩阵运算库使用了 Eigen 而不是 BLAS 库,能够基于 ARM 架构编译和优化,因此在移动设备(Android 和 iOS)上表现得很好。

TensorFlow 在最开始发布时只支持单机,而且只支持 CUDA 6.5 和 cuDNN v2,并且没有官方和其他深度学习框架的对比结果。在 2015 年年底,许多其他框架做了各种性能对比评测,每次 TensorFlow 都会作为较差的对照组出现。那个时期的 TensorFlow 真的不快,性能上仅和普遍认为很慢的 Theano 比肩,在各个框架中可以算是垫底。但是凭借 Google 强大的开发实力,很快支持了新版的 cuDNN (目前支持cuDNN v5.1),在单 GPU 上的性能追上了其他框架。表 2-3 所示为 https://github.com/soumith/convnet-benchmarks 给出的各个框架在 AlexNet 上单 GPU 的性能评测。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

表2-3 各深度学习框架在 AlexNet 上的性能对比

目前在单 GPU 的条件下,绝大多数深度学习框架都依赖于 cuDNN,因此只要硬件计算能力或者内存分配差异不大,最终训练速度不会相差太大。但是对于大规模深度学习来说,巨大的数据量使得单机很难在有限的时间完成训练。这时需要分布式计算使 GPU 集群乃至 TPU 集群并行计算,共同训练出一个模型,所以框架的分布式性能是至关重要的。TensorFlow 在 2016 年 4 月开源了分布式版本,使用 16 块 GPU 可达单 GPU 的 15 倍提速,在 50 块 GPU 时可达到 40 倍提速,分布式的效率很高。目前原生支持的分布式深度学习框架不多,只有 TensorFlow、CNTK、DeepLearning4J、MXNet 等。不过目前 TensorFlow 的设计对不同设备间的通信优化得不是很好,其单机的 reduction 只能用 CPU 处理,分布式的通信使用基于 socket 的 RPC,而不是速度更快的 RDMA,所以其分布式性能可能还没有达到最优。

Google 在 2016 年 2 月开源了 TensorFlow Serving,这个组件可以将 TensorFlow 训练好的模型导出,并部署成可以对外提供预测服务的 RESTful 接口,如图 2-2 所示。有了这个组件,TensorFlow 就可以实现应用机器学习的全流程:从训练模型、调试参数,到打包模型,最后部署服务,名副其实是一个从研究到生产整条流水线都齐备的框架。这里引用 TensorFlow 内部开发人员的描述:“ TensorFlow Serving 是一个为生产环境而设计的高性能的机器学习服务系统。它可以同时运行多个大规模深度学习模型,支持模型生命周期管理、算法实验,并可以高效地利用 GPU 资源,让 TensorFlow 训练好的模型更快捷方便地投入到实际生产环境”。除了 TensorFlow 以外的其他框架都缺少为生产环境部署的考虑,而 Google 作为广泛在实际产品中应用深度学习的巨头可能也意识到了这个机会,因此开发了这个部署服务的平台。TensorFlow Serving 可以说是一副王牌,将会帮 TensorFlow 成为行业标准做出巨大贡献。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-2 TensorFlow Serving 架构

TensorBoard 是 TensorFlow 的一组 Web 应用,用来监控 TensorFlow 运行过程,或可视化 Computation Graph。TensorBoard 目前支持五种可视化:标量(scalars)、图片(images)、音频(audio)、直方图(histograms)和计算图(Computation Graph)。TensorBoard 的 Events Dashboard 可以用来持续地监控运行时的关键指标,比如 loss、学习速率(learning rate)或是验证集上的准确率(accuracy);Image Dashboard 则可以展示训练过程中用户设定保存的图片,比如某个训练中间结果用 Matplotlib 等绘制(plot)出来的图片;Graph Explorer 则可以完全展示一个 TensorFlow 的计算图,并且支持缩放拖曳和查看节点属性。TensorBoard 的可视化效果如图 2-3 和图 2-4 所示。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-3 TensorBoard 的 loss 标量的可视化

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-4 TensorBoard 的模型结构可视化

TensorFlow 拥有产品级的高质量代码,有 Google 强大的开发、维护能力的加持,整体架构设计也非常优秀。相比于同样基于 Python 的老牌对手 Theano,TensorFlow 更成熟、更完善,同时 Theano 的很多主要开发者都去了 Google 开发 TensorFlow(例如书籍 Deep Learning 的作者 Ian Goodfellow,他后来去了 OpenAI)。Google 作为巨头公司有比高校或者个人开发者多得多的资源投入到 TensorFlow 的研发,可以预见,TensorFlow 未来的发展将会是飞速的,可能会把大学或者个人维护的深度学习框架远远甩在身后。

Caffe

官方网址:http://caffe.berkeleyvision.org/

GitHub:http://github.com/BVLC/caffe

Caffe 全称为 Convolutional Architecture for Fast Feature Embedding,是一个被广泛使用的开源深度学习框架(在 TensorFlow 出现之前一直是深度学习领域 GitHub star 最多的项目),目前由伯克利视觉学中心(Berkeley Vision and Learning Center,BVLC)进行维护。Caffe 的创始人是加州大学伯克利的 Ph.D.贾扬清,他同时也是TensorFlow的作者之一,曾工作于 MSRA、NEC 和 Google Brain,目前就职于 Facebook FAIR 实验室。Caffe 的主要优势包括如下几点。

  • 容易上手,网络结构都是以配置文件形式定义,不需要用代码设计网络。

  • 训练速度快,能够训练 state-of-the-art 的模型与大规模的数据。

  • 组件模块化,可以方便地拓展到新的模型和学习任务上。

Caffe 的核心概念是 Layer,每一个神经网络的模块都是一个 Layer。Layer 接收输入数据,同时经过内部计算产生输出数据。设计网络结构时,只需要把各个 Layer 拼接在一起构成完整的网络(通过写 protobuf 配置文件定义)。比如卷积的 Layer,它的输入就是图片的全部像素点,内部进行的操作是各种像素值与 Layer 参数的 convolution 操作,最后输出的是所有卷积核 filter 的结果。每一个 Layer 需要定义两种运算,一种是正向(forward)的运算,即从输入数据计算输出结果,也就是模型的预测过程;另一种是反向(backward)的运算,从输出端的 gradient 求解相对于输入的 gradient,即反向传播算法,这部分也就是模型的训练过程。实现新 Layer 时,需要将正向和反向两种计算过程的函数都实现,这部分计算需要用户自己写 C++或者 CUDA (当需要运行在 GPU 时)代码,对普通用户来说还是非常难上手的。正如它的名字 Convolutional Architecture for Fast Feature Embedding 所描述的,Caffe 最开始设计时的目标只针对于图像,没有考虑文本、语音或者时间序列的数据,因此 Caffe 对卷积神经网络的支持非常好,但对时间序列 RNN、LSTM 等支持得不是特别充分。同时,基于 Layer 的模式也对 RNN 不是非常友好,定义 RNN 结构时比较麻烦。在模型结构非常复杂时,可能需要写非常冗长的配置文件才能设计好网络,而且阅读时也比较费力。

Caffe 的一大优势是拥有大量的训练好的经典模型(AlexNet、VGG、Inception)乃至其他 state-of-the-art (ResNet等)的模型,收藏在它的 Model Zoo (http://github.com/BVLC/ caffe/wiki/Model-Zoo)。因为知名度较高,Caffe 被广泛地应用于前沿的工业界和学术界,许多提供源码的深度学习的论文都是使用 Caffe 来实现其模型的。在计算机视觉领域 Caffe 应用尤其多,可以用来做人脸识别、图片分类、位置检测、目标追踪等。虽然 Caffe 主要是面向学术圈和研究者的,但它的程序运行非常稳定,代码质量比较高,所以也很适合对稳定性要求严格的生产环境,可以算是第一个主流的工业级深度学习框架。因为 Caffe 的底层是基于 C++的,因此可以在各种硬件环境编译并具有良好的移植性,支持 Linux、Mac 和 Windows 系统,也可以编译部署到移动设备系统如 Android 和 iOS 上。和其他主流深度学习库类似,Caffe 也提供了 Python 语言接口 pycaffe,在接触新任务,设计新网络时可以使用其 Python 接口简化操作。不过,通常用户还是使用 Protobuf 配置文件定义神经网络结构,再使用 command line 进行训练或者预测。Caffe 的配置文件是一个 JSON 类型的 .prototxt 文件,其中使用许多顺序连接的 Layer 来描述神经网络结构。Caffe 的二进制可执行程序会提取这些 .prototxt 文件并按其定义来训练神经网络。理论上,Caffe 的用户可以完全不写代码,只是定义网络结构就可以完成模型训练了。Caffe 完成训练之后,用户可以把模型文件打包制作成简单易用的接口,比如可以封装成 Python 或 MATLAB 的 API 。不过在 .prototxt 文件内部设计网络节构可能会比较受限,没有像 TensorFlow 或者 Keras 那样在 Python 中设计网络结构方便、自由。更重要的是,Caffe 的配置文件不能用编程的方式调整超参数,也没有提供像 Scikit-learn 那样好用的 estimator 可以方便地进行交叉验证、超参数的 Grid Search 等操作。Caffe 在 GPU 上训练的性能很好(使用单块 GTX 1080 训练 AlexNet 时一天可以训练上百万张图片),但是目前仅支持单机多 GPU 的训练,没有原生支持分布式的训练。庆幸的是,现在有很多第三方的支持,比如雅虎开源的 CaffeOnSpark,可以借助 Spark 的分布式框架实现 Caffe 的大规模分布式训练。

Theano

官方网址:http://www.deeplearning.net/software/theano/

GitHub:http://github.com/Theano/Theano

Theano 诞生于2008年,由蒙特利尔大学 Lisa Lab 团队开发并维护,是一个高性能的符号计算及深度学习库。因其出现时间早,可以算是这类库的始祖之一,也一度被认为是深度学习研究和应用的重要标准之一。Theano 的核心是一个数学表达式的编译器,专门为处理大规模神经网络训练的计算而设计。它可以将用户定义的各种计算编译为高效的底层代码,并链接各种可以加速的库,比如 BLAS、CUDA 等。Theano 允许用户定义、优化和评估包含多维数组的数学表达式,它支持将计算装载到 GPU (Theano 在 GPU 上性能不错,但是 CPU 上较差)。与 Scikit-learn 一样,Theano 也很好地整合了 NumPy,对 GPU 的透明让 Theano 可以较为方便地进行神经网络设计,而不必直接写 CUDA 代码。Theano 的主要优势如下。

  • 集成 NumPy,可以直接使用 NumPy 的 ndarray,API 接口学习成本低。

  • 计算稳定性好,比如可以精准地计算输出值很小的函数(像 log(1+x))。

  • 动态地生成 C 或者 CUDA 代码,用以编译成高效的机器代码。

因为 Theano 非常流行,有许多人为它编写了高质量的文档和教程,用户可以方便地查找 Theano 的各种 FAQ,比如如何保存模型、如何运行模型等。不过 Theano 更多地被当作一个研究工具,而不是当作产品来使用。虽然 Theano 支持 Linux、Mac 和 Windows,但是没有底层 C++的接口,因此模型的部署非常不方便,依赖于各种 Python 库,并且不支持各种移动设备,所以几乎没有在工业生产环境的应用。Theano 在调试时输出的错误信息非常难以看懂,因此 DEBUG 时非常痛苦。同时,Theano 在生产环境使用训练好的模型进行预测时性能比较差,因为预测通常使用服务器 CPU(生产环境服务器一般没有 GPU,而且 GPU 预测单条样本延迟高反而不如 CPU),但是 Theano 在 CPU 上的执行性能比较差。

Theano 在单 GPU 上执行效率不错,性能和其他框架类似。但是运算时需要将用户的 Python 代码转换成 CUDA 代码,再编译为二进制可执行文件,编译复杂模型的时间非常久。此外,Theano 在导入时也比较慢,而且一旦设定了选择某块 GPU,就无法切换到其他设备。目前,Theano 在 CUDA 和 cuDNN 上不支持多 GPU,只在 OpenCL 和 Theano 自己的 gpuarray 库上支持多 GPU 训练,速度暂时还比不上 CUDA 的版本,并且 Theano 目前还没有分布式的实现。不过,Theano 在训练简单网络(比如很浅的 MLP)时性能可能比 TensorFlow 好,因为全部代码都是运行时编译,不需要像 TensorFlow 那样每次 feed mini-batch 数据时都得通过低效的 Python 循环来实现。

Theano 是一个完全基于 Python (C++/CUDA 代码也是打包为 Python 字符串)的符号计算库。用户定义的各种运算,Theano 可以自动求导,省去了完全手工写神经网络反向传播算法的麻烦,也不需要像 Caffe 一样为 Layer 写 C++或 CUDA 代码。Theano 对卷积神经网络的支持很好,同时它的符号计算 API 支持循环控制(内部名scan),让 RNN 的实现非常简单并且高性能,其全面的功能也让 Theano 可以支持大部分 state-of-the-art 的网络。Theano 派生出了大量基于它的深度学习库,包括一系列的上层封装,其中有大名鼎鼎的 Keras,Keras 对神经网络抽象得非常合适,以至于可以随意切换执行计算的后端(目前同时支持 Theano 和 TensorFlow)。Keras 比较适合在探索阶段快速地尝试各种网络结构,组件都是可插拔的模块,只需要将一个个组件(比如卷积层、激活函数等)连接起来,但是设计新模块或者新的 Layer 就不太方便了。除 Keras 外,还有学术界非常喜爱的 Lasagne,同样也是 Theano 的上层封装,它对神经内网络的每一层的定义都非常严谨。另外,还有 scikit-neuralnetwork、nolearn 这两个基于 Lasagne 的上层封装,它们将神经网络抽象为兼容 Scikit-learn 接口的 classifier 和 regressor,这样就可以方便地使用 Scikit-learn 中经典的 fit、transform、score 等操作。除此之外,Theano 的上层封装库还有 blocks、deepy、pylearn2 和 Scikit-theano,可谓是一个庞大的家族。如果没有 Theano,可能根本不会出现这么多好用的 Python 深度学习库。同样,如果没有 Python 科学计算的基石 NumPy,就不会有 SciPy、Scikit-learn 和 Scikit-image,可以说 Theano 就是深度学习界的 NumPy,是其他各类 Python 深度学习库的基石。虽然 Theano 非常重要,但是直接使用 Theano 设计大型的神经网络还是太烦琐了,用 Theano 实现 Google Inception 就像用 NumPy 实现一个支持向量机(SVM)。且不说很多用户做不到用 Theano 实现一个 Inception 网络,即使能做到但是否有必要花这个时间呢?毕竟不是所有人都是基础科学工作者,大部分使用场景还是在工业应用中。所以简单易用是一个很重要的特性,这也就是其他上层封装库的价值所在:不需要总是从最基础的 tensor 粒度开始设计网络,而是从更上层的 Layer 粒度设计网络。

Torch

官方网址:http://torch.ch/

GitHub:http://github.com/torch/torch7

Torch 给自己的定位是 LuaJIT 上的一个高效的科学计算库,支持大量的机器学习算法,同时以 GPU 上的计算优先。Torch 的历史非常悠久,但真正得到发扬光大是在 Facebook 开源了其深度学习的组件之后,此后包括 Google、Twitter、NYU、IDIAP、Purdue 等组织都大量使用 Torch。Torch 的目标是让设计科学计算算法变得便捷,它包含了大量的机器学习、计算机视觉、信号处理、并行运算、图像、视频、音频、网络处理的库,同时和 Caffe 类似,Torch 拥有大量的训练好的深度学习模型。它可以支持设计非常复杂的神经网络的拓扑图结构,再并行化到 CPU 和 GPU 上,在 Torch 上设计新的 Layer 是相对简单的。它和 TensorFlow 一样使用了底层 C++加上层脚本语言调用的方式,只不过 Torch 使用的是 Lua。Lua 的性能是非常优秀的(该语言经常被用来开发游戏),常见的代码可以通过透明的 JIT 优化达到 C 的性能的80%;在便利性上,Lua 的语法也非常简单易读,拥有漂亮和统一的结构,易于掌握,比写 C/C++简洁很多;同时,Lua 拥有一个非常直接的调用 C 程序的接口,可以简便地使用大量基于 C 的库,因为底层核心是 C 写的,因此也可以方便地移植到各种环境。Lua 支持 Linux、Mac,还支持各种嵌入式系统(iOS、Android、FPGA 等),只不过运行时还是必须有 LuaJIT 的环境,所以工业生产环境的使用相对较少,没有 Caffe 和 TensorFlow 那么多。

为什么不简单地使用 Python 而是使用 LuaJIT 呢?官方给出了以下几点理由。

  • LuaJIT 的通用计算性能远胜于 Python,而且可以直接在 LuaJIT 中操作 C 的 pointers。

  • Torch 的框架,包含 Lua 是自洽的,而完全基于 Python 的程序对不同平台、系统移植性较差,依赖的外部库较多。

  • LuaJIT 的 FFI 拓展接口非常易学,可以方便地链接其他库到 Torch 中。Torch 中还专门设计了 N-Dimension array type 的对象 Tensor,Torch 中的 Tensor 是一块内存的视图,同时一块内存可能有许多视图(Tensor)指向它,这样的设计同时兼顾了性能(直接面向内存)和便利性。同时,Torch 还提供了不少相关的库,包括线性代数、卷积、傅里叶变换、绘图和统计等,如图 2-5 所示。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-5 Torch 提供的各种数据处理的库

Torch 的 nn 库支持神经网络、自编码器、线性回归、卷积网络、循环神经网络等,同时支持定制的损失函数及梯度计算。Torc h因为使用了 LuaJIT,因此用户在 Lua 中做数据预处理等操作可以随意使用循环等操作,而不必像在 Python 中那样担心性能问题,也不需要学习 Python 中各种加速运算的库。不过,Lua 相比 Python 还不是那么主流,对大多数用户有学习成本。Torch 在 CPU 上的计算会使用 OpenMP、SSE 进行优化,GPU 上使用 CUDA、cutorch、cunn、cuDNN 进行优化,同时还有 cuda-convnet 的 wrapper。Torch 有很多第三方的扩展可以支持 RNN,使得 Torch 基本支持所有主流的网络。和 Caffe 类似的是,Torch 也是主要基于 Layer 的连接来定义网络的。Torch 中新的 Layer 依然需要用户自己实现,不过定义新 Layer 和定义网络的方式很相似,非常简便,不像 Caffe 那么麻烦,用户需要使用 C++或者 CUDA 定义新 Layer。同时,Torch 属于命令式编程模式,不像 Theano、TensorFlow 属于声明性编程(计算图是预定义的静态的结构),所以用它实现某些复杂操作(比如 beam search)比 Theano 和 TensorFlow 方便很多。

Lasagne

官网网址:http://lasagne.readthedocs.io/

GitHub:http://github.com/Lasagne/Lasagne

Lasagne 是一个基于 Theano 的轻量级的神经网络库。它支持前馈神经网络,比如卷积网络、循环神经网络、LSTM 等,以及它们的组合;支持许多优化方法,比如 Nesterov momentum、RMSprop、ADAM 等;它是 Theano 的上层封装,但又不像 Keras 那样进行了重度的封装,Keras 隐藏了 Theano 中所有的方法和对象,而 Lasagne 则是借用了 Theano 中很多的类,算是介于基础的 Theano 和高度抽象的 Keras 之间的一个轻度封装,简化了操作同时支持比较底层的操作。Lasagne 设计的六个原则是简洁、透明、模块化、实用、聚焦和专注。

Keras

官方网址:http://keras.io

GitHub:http://github.com/fchollet/keras

Keras 是一个崇尚极简、高度模块化的神经网络库,使用 Python 实现,并可以同时运行在 TensorFlow 和 Theano 上。它旨在让用户进行最快速的原型实验,让想法变为结果的这个过程最短。Theano 和 TensorFlow 的计算图支持更通用的计算,而 Keras 则专精于深度学习。Theano 和 TensorFlow 更像是深度学习领域的 NumPy,而 Keras 则是这个领域的 Scikit-learn。它提供了目前为止最方便的 API,用户只需要将高级的模块拼在一起,就可以设计神经网络,它大大降低了编程开销(code overhead)和阅读别人代码时的理解开销(cognitive overhead)。它同时支持卷积网络和循环网络,支持级联的模型或任意的图结构的模型(可以让某些数据跳过某些 Layer 和后面的 Layer 对接,使得创建 Inception 等复杂网络变得容易),从 CPU 上计算切换到 GPU 加速无须任何代码的改动。因为底层使用 Theano 或 TensorFlow,用 Keras 训练模型相比于前两者基本没有什么性能损耗(还可以享受前两者持续开发带来的性能提升),只是简化了编程的复杂度,节约了尝试新网络结构的时间。可以说模型越复杂,使用 Keras 的收益就越大,尤其是在高度依赖权值共享、多模型组合、多任务学习等模型上,Keras 表现得非常突出。Keras 所有的模块都是简洁、易懂、完全可配置、可随意插拔的,并且基本上没有任何使用限制,神经网络、损失函数、优化器、初始化方法、激活函数和正则化等模块都是可以自由组合的。Keras 也包括绝大部分 state-of-the-art 的 Trick,包括 Adam、RMSProp、Batch Normalization、PReLU、ELU、LeakyReLU 等。同时,新的模块也很容易添加,这让 Keras 非常适合最前沿的研究。Keras 中的模型也都是在 Python 中定义的,不像 Caffe、CNTK 等需要额外的文件来定义模型,这样就可以通过编程的方式调试模型结构和各种超参数。在 Keras 中,只需要几行代码就能实现一个 MLP,或者十几行代码实现一个 AlexNet,这在其他深度学习框架中基本是不可能完成的任务。Keras 最大的问题可能是目前无法直接使用多 GPU,所以对大规模的数据处理速度没有其他支持多 GPU 和分布式的框架快。Keras 的编程模型设计和 Torch 很像,但是相比 Torch,Keras 构建在 Python 上,有一套完整的科学计算工具链,而 Torch 的编程语言 Lua 并没有这样一条科学计算工具链。无论从社区人数,还是活跃度来看,Keras 目前的增长速度都已经远远超过了 Torch。

MXNet

官网网址:http://mxnet.io

GitHub:http://github.com/dmlc/mxnet

MXNet 是 DMLC(Distributed Machine Learning Community)开发的一款开源的、轻量级、可移植的、灵活的深度学习库,它让用户可以混合使用符号编程模式和指令式编程模式来最大化效率和灵活性,目前已经是 AWS 官方推荐的深度学习框架。MXNet 的很多作者都是中国人,其最大的贡献组织为百度,同时很多作者来自 cxxnet、minerva 和 purine2 等深度学习项目,可谓博采众家之长。它是各个框架中率先支持多 GPU 和分布式的,同时其分布式性能也非常高。MXNet 的核心是一个动态的依赖调度器,支持自动将计算任务并行化到多个 GPU 或分布式集群(支持 AWS、Azure、Yarn 等)。它上层的计算图优化算法可以让符号计算执行得非常快,而且节约内存,开启 mirror 模式会更加省内存,甚至可以在某些小内存 GPU 上训练其他框架因显存不够而训练不了的深度学习模型,也可以在移动设备(Android、iOS)上运行基于深度学习的图像识别等任务。此外,MXNet 的一个很大的优点是支持非常多的语言封装,比如 C++、Python、R、Julia、Scala、Go、MATLAB 和 JavaScript 等,可谓非常全面,基本主流的脚本语言全部都支持了。在 MXNet 中构建一个网络需要的时间可能比 Keras、Torch 这类高度封装的框架要长,但是比直接用 Theano 等要快。MXNet 的各级系统架构(下面为硬件及操作系统底层,逐层向上为越来越抽象的接口)如图2-6所示。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-6 MXNet 系统架构

DIGITS

官方网址:http://developer.nvidia.com/digits

GitHub: http://github.com/NVIDIA/DIGITS

DIGITS(Deep Learning GPU Training System)不是一个标准的深度学习库,它可以算是一个 Caffe 的高级封装(或者 Caffe 的 Web 版培训系统)。因为封装得非常重,以至于你不需要(也不能)在 DIGITS 中写代码,即可实现一个深度学习的图片识别模型。在 Caffe 中,定义模型结构、预处理数据、进行训练并监控训练过程是相对比较烦琐的,DIGITS 把所有这些操作都简化为在浏览器中执行。它可以算作 Caffe 在图片分类上的一个漂亮的用户可视化界面(GUI),计算机视觉的研究者或者工程师可以非常方便地设计深度学习模型、测试准确率,以及调试各种超参数。同时使用它也可以生成数据和训练结果的可视化统计报表,甚至是网络的可视化结构图。训练好的 Caffe 模型可以被 DIGITS 直接使用,上传图片到服务器或者输入 url 即可对图片进行分类。

CNTK

官方网址:http://cntk.ai

GitHub:http://github.com/Microsoft/CNTK

CNTK(Computational Network Toolkit)是微软研究院(MSR)开源的深度学习框架。它最早由 start the deep learning craze 的演讲人创建,目前已经发展成一个通用的、跨平台的深度学习系统,在语音识别领域的使用尤其广泛。CNTK 通过一个有向图将神经网络描述为一系列的运算操作,这个有向图中子节点代表输入或网络参数,其他节点代表各种矩阵运算。CNTK 支持各种前馈网络,包括 MLP、CNN、RNN、LSTM、Sequence-to-Sequence 模型等,也支持自动求解梯度。CNTK 有丰富的细粒度的神经网络组件,使得用户不需要写底层的 C++或 CUDA,就能通过组合这些组件设计新的复杂的 Layer。CNTK 拥有产品级的代码质量,支持多机、多 GPU 的分布式训练。

CNTK 设计是性能导向的,在 CPU、单 GPU、多 GPU,以及 GPU 集群上都有非常优异的表现。同时微软最近推出的 1-bit compression 技术大大降低了通信代价,让大规模并行训练拥有了很高的效率。CNTK 同时宣称拥有很高的灵活度,它和 Caffe 一样通过配置文件定义网络结构,再通过命令行程序执行训练,支持构建任意的计算图,支持 AdaGrad、RmsProp 等优化方法。它的另一个重要特性就是拓展性,CNTK 除了内置的大量运算核,还允许用户定义他们自己的计算节点,支持高度的定制化。CNTK 在 2016 年 9 月发布了对强化学习的支持,同时,除了通过写配置文件的方式定义网络结构,CNTK 还将支持其他语言的绑定,包括 Python、C++和 C#,这样用户就可以用编程的方式设计网络结构。CNTK 与 Caffe一样也基于 C++并且跨平台,大部分情况下,它的部署非常简单。PC 上支持 Linux、Mac 和 Windows,但是它目前不支持 ARM 架构,限制了其在移动设备上的发挥。图 2-7 所示为 CNTK 目前的总体架构图。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-7 CNTK的总体架构图

CNTK 原生支持多 GPU 和分布式,从官网公布的对比评测来看,性能非常不错。在多 GPU 方面,CNTK 相对于其他的深度学习库表现得更突出,它实现了 1-bit SGD 和自适应的 mini-batching。图 2-8 所示为 CNTK 官网公布的在 2015 年 12 月的各个框架的性能对比。在当时,CNTK 是唯一支持单机 8 块 GPU 的框架,并且在分布式系统中可以超越 8 块 GPU 的性能。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-8 CNTK与各个框架的性能对比

Deeplearning4J

官方网址:http://deeplearning4j.org/

GitHub: http://github.com/deeplearning4j/deeplearning4j

Deeplearning4J(简称DL4J)是一个基于 Java 和 Scala 的开源的分布式深度学习库,由 Skymind 于 2014 年 6 月发布,其核心目标是创建一个即插即用的解决方案原型。埃森哲、雪弗兰、博斯咨询和 IBM 等都是 DL4J 的客户。DL4J 拥有一个多用途的 n-dimensional array 的类,可以方便地对数据进行各种操作;拥有多种后端计算核心,用以支持 CPU 及 GPU 加速,在图像识别等训练任务上的性能与 Caffe 相当;可以与 Hadoop 及 Spark 自动整合,同时可以方便地在现有集群(包括但不限于AWS,Azure等)上进行扩展,同时 DL4J 的并行化是根据集群的节点和连接自动优化,不像其他深度学习库那样可能需要用户手动调整。DL4J 选择 Java 作为其主要语言的原因是,目前基于 Java 的分布式计算、云计算、大数据的生态非常庞大。用户可能拥有大量的基于 Hadoop 和 Spark 的集群,因此在这类集群上搭建深度学习平台的需求便很容易被 DL4J 满足。同时 JVM 的生态圈内还有数不胜数的 Library 的支持,而 DL4J 也创建了 ND4J,可以说是 JVM 中的 NumPy,支持大规模的矩阵运算。此外,DL4J 还有商业版的支持,付费用户在出现问题时可以通过电话咨询寻求支持。

Chainer

官方网址:http://chainer.org

GitHub:http://github.com/pfnet/chainer

Chainer 是由日本公司 Preferred Networks 于 2015 年 6 月发布的深度学习框架。Chainer 对自己的特性描述如下。

  • Powerful:支持 CUDA 计算,只需要几行代码就可以使用 GPU 加速,同时只需少许改动就可以运行在多 GPU 上。

  • Flexible:支持多种前馈神经网络,包括卷积网络、循环网络、递归网络,支持运行中动态定义的网络(Define-by-Run)。

  • Intuitive:前馈计算可以引入 Python 的各种控制流,同时反向传播时不受干扰,简化了调试错误的难度。

绝大多数的深度学习框架是基于“Define-and-Run”的,也就是说,需要首先定义一个网络,再向网络中 feed 数据(mini-batch)。因为网络是预先静态定义的,所有的控制逻辑都需要以 data 的形式插入网络中,包括像 Caffe 那样定义好网络结构文件,或者像 Theano、Torch、TensorFlow 等使用编程语言定义网络。而 Chainer 则相反,网络是在实际运行中定义的,Chainer 存储历史运行的计算结果,而不是网络的结构逻辑,这样就可以方便地使用 Python 中的控制流,所以无须其他工作就可以直接在网络中使用条件控制和循环。

Leaf

官方网址:http://autumnai.com/leaf/book

GitHub:http://github.com/autumnai/leaf

Leaf 是一个基于 Rust 语言的直观的跨平台的深度学习乃至机器智能框架,它拥有一个清晰的架构,除了同属 Autumn AI 的底层计算库 Collenchyma,Leaf 没有其他依赖库。它易于维护和使用,并且拥有非常高的性能。Leaf 自身宣传的特点是为 Hackers 定制的,这里的 Hackers 是指希望用最短的时间和最少的精力实现机器学习算法的技术极客。它的可移植性非常好,可以运行在 CPU、GPU 和 FPGA 等设备上,可以支持有任何操作系统的 PC、服务器,甚至是没有操作系统的嵌入式设备,并且同时支持 OpenCL 和 CUDA。Leaf 是 Autumn AI 计划的一个重要组件,后者的目标是让人工智能算法的效率提高 100 倍。凭借其优秀的设计,Leaf 可以用来创建各种独立的模块,比如深度强化学习、可视化监控、网络部署、自动化预处理和大规模产品部署等。

Leaf 拥有最简单的 API,希望可以最简化用户需要掌握的技术栈。虽然才刚诞生不久,Leaf 就已经跻身最快的深度学习框架之一了。图 2-9 所示为 Leaf 官网公布的各个框架在单 GPU 上训练 VGG 网络的计算时间(越小越好)的对比(这是和早期的 TensorFlow 对比,最新版的 TensorFlow 性能已经非常好了)。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-9 Leaf 和各深度学习框架的性能对比(深色为 forawrd,浅色为 backward)

DSSTNE

GitHub:http://github.com/amznlabs/amazon-dsstne

DSSTNE(Deep Scalable Sparse Tensor Network Engine)是亚马逊开源的稀疏神经网络框架,在训练非常稀疏的数据时具有很大的优势。DSSTNE 目前只支持全连接的神经网络,不支持卷积网络等。和 Caffe 类似,它也是通过写一个 JSON 类型的文件定义模型结构,但是支持非常大的 Layer(输入和输出节点都非常多);在激活函数、初始化方式及优化器方面基本都支持了 state-of-the-art 的方法,比较全面;支持大规模分布式的 GPU 训练,不像其他框架一样主要依赖数据并行,DSSTNE 支持自动的模型并行(使用数据并行需要在训练速度和模型准确度上做一定的 trade-off,模型并行没有这个问题)。

在处理特征非常多(上亿维)的稀疏训练数据时(经常在推荐、广告、自然语言处理任务中出现),即使一个简单的三个隐层的 MLP(Multi-Layer Perceptron)也会变成一个有非常多参数的模型(可能高达上万亿)。以传统的稠密矩阵的方式训练方法很难处理这么多的模型参数,更不必提超大规模的数据量,而 DSSTNE 有整套的针对稀疏数据的优化,率先实现了对超大稀疏数据训练的支持,同时在性能上做了非常大的改进。

在 DSSTNE 官方公布的测试中,DSSTNE 在 MovieLens 的稀疏数据上,在单 M40 GPU 上取得了比 TensorFlow 快 14.8 倍的性能提升(注意是和老版的 TensorFlow 比较),如图 2-10 所示。一方面是因为 DSSTNE 对稀疏数据的优化;另一方面是 TensorFlow 在数据传输到 GPU 上时花费了大量时间,而 DSSTNE 则优化了数据在GPU内的保留;同时 DSSTNE 还拥有自动模型并行功能,而 TensorFlow 中则需要手动优化,没有自动支持。

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

图2-10 DSSTNE 在稀疏数据上与 TensorFlow 的性能对比

雷锋网按:在下周二(2月28日)雷锋网(公众号:雷锋网)硬创公开课栏目将会推出黄文坚老师的公开课《深度学习之经典卷积神经网络的技术浅析》(点击了解详情),欢迎大家报名!

雷锋网版权文章,未经授权禁止转载。详情见。


TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/71645.html

(0)
上一篇 2021年8月11日 18:43
下一篇 2021年8月11日 18:43

相关推荐

发表回复

登录后才能评论