一张图表明:研究人员喜爱PyTorch和TensorFlow
研究人员对PyTorch的兴趣正在快速增加
编者注:人工智能伦敦大会(2019年10月14日至17日)上有多个关于PyTorch、TensorFlow和其他重要题目非常好的培训课程辅导课

在近期我们进行了一项调查——企业里的人工智能,收到了超过1300份的回复。我们发现几个机器学习的库和框架被广泛地使用。超过半数的受访者说他们已经使用了TensorFlow和scikit-learn,第三位被使用的是PyTorch和Keras。

我最近参加了RISELabCaroline Lemieux做的一个有趣的演讲。她介绍了AutoPandas和依赖于程序合成的自动化工具的最新工作。在演讲过程中,Lemieux回顾了他们收集的对于不同深度学习框架和数据科学库的使用统计数据。她很大度地分享给我了一些数据。我用这些数据绘制出下面这张图:

image1-e18a246da831903928949a96696ffb38

图1 发布在arXiv.org上的提到每个框架的论文数量。来源:数据来自RISELab,图由Ben Lorica提供

图1里的数字是对预发表平台arXiv.org上的论文进行全文搜索获得的,即在全文搜索里,提到每个框架的论文数量。基于这个指标我们发现,在研究人员里最流行的两个深度学习框架是TensorFlow和PyTorch。从2019年1月到6月底,有1800多篇论文提到了TensorFlow,同时也有差不多的论文提到了PyTorch。最值得注意的是,研究人员里对于PyTorch的兴趣正在快速增加:年度增长率为194%(2019年1月到6月相比于2018年1月到6月)。

在某种程度上研究人员和老师都是最爱尝鲜的用户,并且强烈影响着未来从业人员可能会使用的东西。看起来PyTorch也在从数据科学家、开发人员和机器学习工程师和公司社区中吸引用户。在最近的一篇文章中我们概述了用于管理企业中机器学习的工具套件。我们在其中强调的许多工具和公司都支持流行的机器学习库和深度学习框架(特别是TensorFlow和PyTorch),因此我们预测这两个框架对于企业用户来说都是同样可行的选择。

相关内容:

Ben Lorica

Ben Lorica是O’Reilly Media公司的首席数据科学家,同时也是Strata数据会议和O’Reilly人工智能会议的内容日程主管。他曾在多种场景下应用商业智能、数据挖掘、机器学习和统计分析技术,这些场景包括直销、消费者与市场研究、定向广告、文本挖掘和金融工程。他的背景包括在投资管理公司、互联网初创企业和金融服务公司就职。

Research lab (source: SMU Libraries Digital Collections on Flickr)