深度学习框架的流行趋势正在快速变化，其中最受瞩目的莫过于TensorFlow和PyTorch。前段时间，TensorFlow发布了2.0（Alpha）版本，不仅推出了许多新功能，也通过引入Keras高级API和EagerExecution模式让入门门槛进一步降低。但是这些改变并没有减慢PyTorch追赶的步伐。在今天这篇文章中，本文作者通过GitHub、Medium文章、arXiv论文和LinkedIn等多个维度评估了这两款深度学习框架的最新发展趋势。

2018年9月，我在这篇文章中从市场需求、使用和受欢迎程度三方面比较了所有主要的深度学习框架。其中，TensorFlow是深度学习框架中无可争议的重量级冠军，PyTorch则是引发了热烈讨论的年轻新秀。

在过去六个月中，这些领先的深度学习框架又有什么变化呢？

图

为了回答这个问题，我查看了Indeed、Monster、LinkedIn和SimplyHired上的职位列表数量。我还评估了Google搜索量、GitHub活动、 Medium 文章、ArXiv文章和Quora主题关注者的变化。总的来说，这些资料来源描绘了市场需求、使用和兴趣增长的全面情况。

集成和更新

我们最近在TensorFlow和PyTorch框架中看到了几个重要的进展。

PyTorch v1.0于2018年10月发布，同时FastAI v1.0发布。这两个版本的发布都是重要的里程碑，标志着深度学习框架趋于成熟。

TensorFlow 2.0 alpha于2019年3月4日发布，它增加了新的功能并改善了用户体验，并且更加紧密地集成了Keras作为其高级API。

方法论

在本文中，我将Keras和FastAI包含在比较中，因为它们与TensorFlow和PyTorch紧密集成，并且还提供了评估TensorFlow和PyTorch的尺度。

我不会在本文中探索其他深度学习框架。我希望我能够收到Caffe、Theano、MXNET、CNTK、DeepLearning4J或Chainer等值得讨论的反馈。虽然这些框架各有其优点，但似乎没有一个框架显示出其可以媲美TensorFlow或PyTorch，并且它们也没有与这两个框架紧密结合。

搜索在2019年3月20日至21日进行。源数据在此Google表格中。

我使用了plotly数据可视化库来探索流行度，对于交互式plotly图表，请在此处查看我的Kaggle Kernel。

让我们看看每个类别的结果。

在线职位列表的变化

为了确定在当今的就业市场中需要哪些深度学习库，我在Indeed、LinkedIn、Monster和SimplyHired上搜索了工作列表。

我同时搜索了关键词”机器学习“和深度学习框架名，比如，TensorFlow通过”机器学习 TensorFlow“搜索关键词进行评估。用此方法是出于与历史比较的原因。不加关键词”机器学习“的搜索没有产生明显不同的结果。搜索区域是美国。

我从2019年3月的职位列表数量中减去了六个月前的职位列表数量，下面就是我发现的内容：

TensorFlow的职位列表增幅略大于PyTorch， Keras也显示出了职位列表数的增长 —— 大约是TensorFlow的一半，FastAI仍然没有出现在任何工作列表中。

请注意，除了LinkedIn之外，PyTorch在其他所有求职网站上都有更多的额外岗位列表。另外，从绝对数量上讲，TensorFlow的工作列表数量几乎是PyTorch或Keras的三倍。

Google搜索活动的平均变化

在大型搜索引擎上进行网络搜索的行为是衡量人气的指标。我查看了过去一年Google趋势中的搜索记录。我在世界范围内搜索了对机器学习和人工智能类别的兴趣。 Google不提供绝对搜索数字，但确实提供了相对数据。

我统计了过去六个月的平均兴趣分数，并将其与更早的六个月的平均兴趣分数进行了比较。

在过去的六个月中，TensorFlow的相对搜索量有所下降，而PyTorch的相对搜索量却在增长。

下面这个来自谷歌的图表显示了过去一年各个框架的搜索兴趣。

（蓝色表示TensorFlow；黄色表示Keras；红色表示PyTorch；绿色表示FastAI）

Medium新文章

Medium是数据科学文章和教程的热门聚集地。我希望你会喜欢它！

在过去的六个月里，我统计了在谷歌网站搜索出的Medium.com文章数量，发现与TensorFlow和Keras相关的文章数量差不多，而PyTorch则相对较少。

作为高级API，Keras和FastAI受到新的深度学习从业者的欢迎， Medium有许多教程展示了如何使用这些框架。

arXiv新文章

arXiv是一个在线存储库，大多数深度学习学术文章都发布在此。我用Google搜索了在过去六个月里在 arXiv上提及各个框架的新文章。

TensorFlow的新文章出现率最高。

新的GitHub活动

GitHub上的近期活动是衡量框架流行度的另一个指标。我在下面的图表中描绘了GitHub点赞、复制、关注和贡献者的数量。

TensorFlow在每个类别中拥有最多的GitHub活动。然而，PyTorch在关注和贡献者的增长方面非常接近。此外，FastAI也增加了许多新的贡献者。

毫无疑问，Keras的一些贡献者对TensorFlow库也进行了研究。值得注意的是，TensorFlow和Keras都是由Google员工负责的开源产品。

Quora新粉丝

我新增了Quora主题关注者数量 —— 一个我之前没有统计数据的新类别。

在过去六个月中，TensorFlow主题新增的粉丝数量最多，而PyTorch和Keras的粉丝增量就少了很多。

获得所有数据后，我将其合并为一个度量指标。

增长分数计算方法

以下是我计算增长分数的方法：

把所有特征值缩放至0到1之间。
聚合在线职位列表和GitHub活动子类别。
对各个类别根据以下百分比加权。

4. 为了便于理解，各加权分数乘以100。
5. 将每个框架的类别分数汇总为单个增长分数。

其中，工作列表的权重占总分的三分之一多一点，原因也很俗气，有钱能使鬼推磨。这种权重系数的划分似乎是在各种类别之间取得了适当的平衡。与我在2018年对各框架能力系数的分析不同，这次我没有包括KDNuggets的使用情况调查（没有新数据），也没有包括书籍（六个月内发布的数量不多）。

结果

下面的表格中列出了四种主流框架在各个类目上的变化情况：

下面是各个类别的分数，以及最后汇总的分数：

下面是最终的增长分数：

TensorFlow是市场需求最多，也是增长最快的框架，它的领先地位不会在短期内被颠覆。 PyTorch也在迅速发展，它在工作列表中的大量增加证明了其使用和需求的增加。在过去的六个月里，Keras也有了很大的发展。最后，值得注意的是，FastAI是从较小的基数开始发展的，它是最年轻的深度学习框架。

TensorFlow和PyTorch都是很好的值得学习的框架。

学习建议

如果你想学习TensorFlow，我建议你从Keras开始。我推荐Chollet的Python深度学习和Dan Becker关于Keras的DataCamp课程。 Tensorflow 2.0通过tf.keras使用Keras作为其高级API。这里有Chollet对TensorFlow 2.0的快速入门介绍。

如果你想学习PyTorch，我建议你从FastAI的MOOC 实践深度学习编码(v3)开始，学习深度学习基础知识、FastAI和PyTorch的基础知识。

TensorFlow和PyTorch的未来发展方向是什么？

未来发展方向

我一直听说，与TensorFlow相比，人们更喜欢使用PyTorch。 PyTorch更具pythonic，并且具有更一致的API，它还具有原生的ONNX模型导出，可用于加速推理。此外，PyTorch与numpy共享许多命令，这减少了学习它的障碍。

然而，TensorFlow 2.0完全是为了改进用户体验，正如谷歌首席决策情报工程师Cassie Kozyrkov在此解释的那样。 TensorFlow现在将拥有更直接的API、简化的Keras集成和eager execution选项。这些变化以及TensorFlow的广泛采用应该有助于该框架在未来几年保持流行。

TensorFlow最近宣布了另一个激动人心的计划：Swift for TensorFlow的开发。 Swift是一种最初由Apple构建的编程语言，在执行和开发速度方面，Swift比Python有许多优势。 FastAI将在部分高级MOOC中使用Swift for TensorFlow ——请参阅FastAI联合创始人Jeremy Howard关于此主题的帖子。这种语言可能不会在一年或两年内进入黄金时间，但它可能是对当前深度学习框架非常有用的改进。

语言和框架之间的协作和交叉授粉肯定正在发生。

影响深度学习框架的另一个进步是量子计算。一台可用的量子计算机可能还需要几年的时间才会出现，但谷歌、IBM、微软和其他公司正在考虑如何将量子计算与深度学习相结合。各类框架需要适应这项新技术。

总结

目前来看，TensorFlow和PyTorch都在快速增长，两者现在都有很好的高级API —— tf.keras和FastAI——它们降低了深度学习入门的门槛。本文还带你了解了一点最近的发展和未来方向。

如果你想要以交互方式使用本文中的图表或复制Jupyter笔记本，可以访问这里：https://www.kaggle.com/discdiver/2019-deep-learning-framework-growth-scores

希望本文的深度学习框架比较能对你有所帮助。

原文链接：
https://towardsdatascience.com/which-deep-learning-framework-is-growing-fastest-3f77f14aa318

创作场景

PyTorch 杠上 TensorFlow：谁才是增长最快的深度学习框架？