【深度学习】常见的一些深度学习框架

╰半夏微凉° 2024-04-19 09:52 81阅读 0赞

### 文章目录 ###

*  前言
 *  一、背景介绍
 *  二、Theano
 *  三、TensorFlow
 *  四、Keras
 *  五、Caffe/Caffe2
 *  六、MXNet
 *  七、CNTK
 *  八、其他框架

# 前言 #

本篇博文旨在主要介绍一些常见的深度学习框架

# 一、背景介绍 #

目前常见的深度学习框架不尽相同，主要有`TensorFlow、Caffe、Theano、Keras`等，常见的深度学习框架如借图如下所示。这些深度学习框架被应用于计算机视觉、语音识别、自然语言处理与生物信息学等领域，并获取了极好的效果。本片博文就主要介绍一下当前深度学习领域影响力比较大的几个框架，只作为学习，还望小伙伴们对于不足之处多多包涵。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0ppYWppa2FuZ19qams_size_16_color_FFFFFF_t_70]

# 二、Theano #

Theano最初诞生于蒙特利尔大学LIS实验室，于2008年开始开发，是第一个有较大影响力的 Python深度学习框架。  
        Theano是Python是一个库，可用于定义、优化和计算数学表达式，特别是多维(numpy.ndaray)。在解决包含大量数据的问题时，使用Theano编程可实现比手写C语言更快的速度，而通过GPU加速， Theano甚至可以比基于CPU计算的C语言快上好几个数量级。 Theano结合了计算机代数系统（Computer Algebra System,cas）和优化编译器，还可以为多种数学运算生成定制的C语言代码。对于包含重复计算的复杂数学表达式的任务而言，计算速度很重要，因此这种CAS和优化编译器的组合是很有用的。对需要将每一种不同的数学表达式都计算一遍的情况， Theano可以最小化编译解析的计算量，但仍然会给出如自动微分那样的符号特征。  
        Theano诞生于研究机构，服务于研究人员，其设计具有较浓厚的学术气息，但在工程设计上有较大的缺陷。一直以来， Theano因难调试、构建图慢等缺点为人所诟病。为了加速深度学习研究，人们在它的基础之上，开发了 `Lasagne、 Blocks、 PyLearn22和Keras`等第三方框架，这些框架以 Theano为基础，提供了更好的封装接口以方便用户使用。

# 三、TensorFlow #

1、2015年11月10日， Google宣布推出全新的机器学习开源工具`TensorFlow`。 `TensorFlow`最初是由 Google机器智能研究部门的 Google Brain团队开发，基于 Google2011年开发的深度学习基础架构`DistBelief`构建起来的。 TensorFlow主要用于进行机器学习和深度神经网络研究，但它是一个非常基础的系统，因此也可以应用于众多领域。  
       2、TensorFlow在很大程度上可以看作`Theano`的后继者，不仅因为它们有很很多共同的开发者，而且它们还拥有相近的设计理念，都是基于计算图实现自动微分系统。TensorFlow使用数据流图进行数值计算，图中的节点代表数学运算，而图中的边则表在这些节点之间传递的多维数组（张量）  
       3、TensorFlow编程接口支持Python和C++。随着1.0版本的公布，`java、Go、R和Haskell API的 alpha版本也被支持`。此外， TensorFlow还可在 Google Cloud和AWS中运 行。 TensorFlow还支持 Windows7、 Windows10和 Windows Server2016.由于 TensorFlow使用C++ Eigen库，所以库可在ARM架构上编译和优化。这也就意味着用户可以在各种服务器和移动设备上部署自己的训练模型，无须执行单独的模型解码器或者加载Python解释器。  
作为当前最流行的深度学习框架， TensorFlow获得了极大的成功，对它的批评也不绝于耳，总结起来主要有以下四点：

> ·1、过于复杂的系统设计， TensorFlow在 GitHub代码仓库的总代码量超过100万行。这么大的代码仓库，对于项目维护者来说维护成为了一个难以完成的任务，而对读者来说，学习 TensorFlow底层运行机制更是一个极其痛苦的过程，并且大多数时候这种尝试以放弃告终。  
> ·2、频繁变动的接口。 TensorFlow的接口一直处于快速迭代之中，并且没有很好地考虑向后兼容性，这导致现在许多开源代码已经无法在新版的 TensorFlow上运行，同时也间接导致了许多基于 TensorFlow的第三方框架出现BUG。  
> 3、接口设计过于晦涩难懂。在设计 TensorFlow时，创造了图、会话、命名空间、 Place-Holder等诸多抽象概念，对普通用户来说难以理解。同一个功能， TensorFlow提供了多种实现，这些实现良莠不齐，使用中还有细微的区别，很容易将用户带入坑中。  
> 4、·文档混乱脱节。 TensorFlow作为一个复杂的系统，文档和教程众多，但缺乏明显的条理和层次，虽然查找很方便，但用户却很难找到一个真正循序渐进的入门教程。

由于直接使用 TensorFlow的生产力过于低下，包括 Google官方等众多开发者尝试基于 TensorFlow构建一个更易用的接口，包括`Keras、 Sonnet、 TFLearn、 TensorLayerSlim、Fold、 PrettyLayer`等数不胜数的第三方框架每隔几个月就会在新闻中出现一次，但是又大多归于沉寂，至今 TensorFlow仍没有一个统一易用的接口。

# 四、Keras #

Keras是一个高层神经网络API，由纯Python编写而成并使用`TensorFlow、Theano及CNTK`作为后端。 Keras为支持快速实验而生，能够把想法迅速转换为结果。Keras应该是深度学习框架之中最容易上手的一个，它提供了一致而简洁的API，能够极大地减少一般应用下用户的工作量，避免用户重复造轮子。  
       严格意义上讲， Keras并不能称为一个深度学习框架，它更像一个深度学习接口，它构建于第三方框架之上。 Keras的缺点很明显：过度封装导致丧失灵活性。 Keras最初作为Theano的高级API而诞生，后来增加了`TensorFlow和CTK`作为后端。为了屏蔽后端的差异性，提供一致的用户接口，Keras做了层层封装，导致用户在新增操作或是获取底层的数据信息时过于困难。同时，过度封装也使得 Keras的程序过于缓慢，许多BUG都隐藏于封装之中，在绝大多数场景下， Keras大部分框架中最慢的一个。  
       学习 Keras十分容易，但是很快就会遇到瓶颈，因为它缺少灵活性。另外，在使用Keras的大多数时间里，用户主要是在调用接口，很难真正学习到深度学习的内容。总结下来：入门最简单，但是不够灵活，使用受限。

# 五、Caffe/Caffe2 #

KerasCaffe的全称是Convolutional Architecture for Fast Feature Embedding，它是一个清晰、高效的深度学习框架，核心语言是C++，它支持命令行、 Python和 MATLAB接口，既可以在CPU上运行，也可以在GPU上运行。  
       Caffe的优点是：简洁快速，缺点是缺少灵活性。不同于 Keras因为太多的封装导致灵活性丧失， Caffe灵活性的缺失主要是因为它的设计。在 Caffe中最主要的抽象对象是层，每实现一个新的层，必须要利用C++实现它的前向传播和反向传播代码，而如果想要新层运行在GPU上，还需要同时利用CUDA实现这一层的前向传播和反向传播。这种限制使得不熟悉C++和CUDA的用户扩展 Caffe十分困难。  
       Caffe凭借其易用性、简洁明了的源码、出众的性能和快速的原型设计获取了众多用户，曾经占据深度学习领域的半壁江山。但是在深度学习新时代到来之时， Caffe已经表现出明显的力不从心，诸多问题逐渐显现（包括灵活性缺失、扩展难、依赖众多环境难以配置、应用局限等）。尽管现在在 GitHub上还能找到许多基于 Caffe的项目，但是新的项目已经越来越少。  
       Caffe的作者从加州大学伯克利分校毕业后加入了 Google，参与过 TensorFlow的开发，后来离开 Google加入FAIR，担任工程主管，并开发了 Caffe2。Caffe2是一个兼具表现力、速度和模块性的开源深度学习框架。它沿袭了大量的 Caffe设计，可解决多年来在 Caffe的使用和部署中发现的瓶颈问题。 Caffe2的设计追求轻量级，在保有扩展性和高性能的同时， Caffe2也强调了便携性。 Caffe2从一开始就以性能、扩展、移动端部署作为主要设计目标。 Caffe2的核心C++库能提供速度和便携性，而其 Python和C++API使用户可以轻松地在`Linux、 Windows、is、 Android，甚至 Raspberry Pi和NVIDIA Tegra上`进行原型设计、训练和部署。  
       Caffe2继承了 Caffe的优点，在速度上令人印象深刻。 Facebook人工智能实验室与应用机器学习团队合作，利用Caffe2大幅加速机器视觉任务的模型训练过程，仅需1小时就训练完 ImageNet这样超大规模的数据集。然而尽管已经发布半年多，开发一年多， Caffe2仍然是一个不太成熟的框架，官网至今没提供完整的文档，安装也比较麻烦，编译过程时常出现异常，在 GitHub上也很少找到相应的代码。  
       极盛的时候，Caffe占据了计算机视觉研究领域的半壁江山，虽然如今 Caffe已经很少用于学术界，但是仍有不少计算机视觉相关的论文使用Caffe由于其稳定、出众的性能，不少公司还在使用 Caffe部署模型。Caffe2尽管做了许多改进，但是还远没有达到替代Caffe的地步，其次，文档不够完善，但性能优异，几乎全平台支持（ Caffe2）适合生产环境。

# 六、MXNet #

MXNet是一个深度学习库，支持`C++、 Python、r、 Scala Julia MATLAB JavaScript`等语言；支持命令和符号编程；可以运行在CPU、GPU、集群、服务器、台式机或者移动设备上。 MXNet是CXXNet是的下一代， CXXNet借鉴了 Caffe的思想，但是在实现上更干净。在2014年的NIPS上，同为上海交大校友的陈天奇与李沐碰头，讨论到各自在做深度学习 Toolkits的项目组，发现大家普遍在做很多重复性的工作，例如文件 loading等。于是他们决定组建DMLC\[ Distributied（Deep） Machine Learning Community\]，号召大家一起合作开发 MXNet，发挥各自的特长，避免重复造轮子。  
       MXNet以其超强的分布式支持，明显的内存、显存优化为人所称道。同样的模型，MXNet往往占用更小的内存和显存，并且在分布式环境下， MXNet展现出了明显优于其他框架的扩展性能。  
       由于 MXNet最初由一群学生开发，缺乏商业应用，极大地限制了 MXNet的使用。2016年11月， MXNet被AWS正式选择为其云计算的官方深度学习平台。2017年1月，MXNet项目进入 Apache基金会，成为 Apache的孵化器项目。  
       尽管 MXNet拥有最多的接口，也获得了不少人的支持，但其始终处于一种不温不火的状态。个人认为这在很大程度上归结于推广不给力及接口文档不够完善。MXNet长期处于快速迭代的过程，其文档却长时间未更新，导致新手户难以掌握 MXNet，老用户常常需要查阅源码才能真正理解 MXNet接口的用法。

# 七、CNTK #

2015年8月，微软公司在 CodePlex上宣布由微软研究院开发的计算网络工具集CNTK将开源。5个月后，2016年1月25日，微软公司在他们的 GitHub仓库上正式开源了CNTK。早在2014年，在微软公司内部，黄学东博士和他的团队正在对计算机能够理解语音的能力进行改进，但当时使用的工具显然拖慢了他们的进度。于是，一组由志愿者组成的开发团队构想设计了他们自己的解决方案，最终诞生了CNTK。  
        根据微软开发者的描述，CNTK的性能比 Caffe、Theano、TensoFlow等主流工具都要强。CNTK支持CPU和GPU模式，和 TensorFlow/Theano一样，它把神经网络描述成一个计算图的结构，叶子节点代表输入或者网络参数，其他节点代表计算步骤。  
CNTK是一个非常强大的命令行系统，可以创建神经网络预测系统。CNTK最初是出于在 Microsoft内部使用的目的而开发的，一开始甚至没有 Python接口，而是使用了一种几乎没什么人用的语言开发的，而且文档有些晦涩难懂，推广不是很给力，导致现在用户比较少。但就框架本身的质量而言，CNTK表现得比较均衡，没有明显的短板，并且在语音领域效果比较突出。此外：社区不够活跃，但是性能突出，擅长语音方面的相关研究。

# 八、其他框架 #

除了上述的几个框架，还有不少的框架，都有一定的影响力和用户。比如百度开源`PaddlePaddle,CMU开发的 DyNet`简洁无依赖符合C++11标准的tiny-dnn，使用Java开发并且文档极其优秀的`Deeplearning44j`，还有英特尔开源的 `Nervana， Amazon开源的 DSSTNE`。这些框架各有优缺点，但是大多流行度和关注度不够，或者局限于一定的领域，因此不做过多的介绍。此外，还有许多专门针对移动设备开发的框架，如CoreML、MDL，这些框架纯粹为部署而诞生，不具有通用性，也不适合作为研究工具，同样不做介绍。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0ppYWppa2FuZ19qams_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/60635674cf1f411e9c003d2fb6fb2946.png