1. 首页
  2. 教育
  3. 深度学习

为什么这三家公司最终都不约而同地放弃TensorFlow

为什么这三家公司最终都不约而同地放弃TensorFlow,而转向了PyTorch?深度学习框架PyTorch凭借其相对易用性,已经渗透到了企业中。本文提到的三家公司告诉我们,为什么他们选择PyTorch而不是Google著名的TensorFlow框架。

为什么这三家公司最终都不约而同地放弃TensorFlow,而转向了PyTorch?

深度学习框架PyTorch凭借其相对易用性,已经渗透到了企业中。本文提到的三家公司告诉我们,为什么他们选择PyTorch而不是Google著名的TensorFlow框架。

深度学习是机器学习的一个子类别,它使用多层神经网络将历史上很难完成的机器任务——比如图像识别、自然语言处理和机器翻译——实现大规模自动化。

自2015年以来,TensorFlow就从Google中脱颖而出,在研究和商业领域成为最受欢迎的开源深度学习框架。但是2016年从Facebook诞生的PyTorch由于社区推动的易用性改进和越来越广泛的用例部署,而迅速赶上TensorFlow。

在汽车行业,PyTorch的应用尤为广泛——它可以应用于Tesla和LyftLevel5等公司的实验性自动驾驶系统。该框架也被媒体公司用来对内容进行分类和推荐,并在工业应用中支持机器人

FacebookAI的人工智能产品主管JoeSpisak告诉InfoWorld,虽然他对企业采用PyTorch的人数增加感到高兴,但要扩大行业应用范围,还有很多工作要做。

“MLOps和Kubeflow管道以及这些社区的支持将会带来新一轮的采用浪潮,”他说。“对于那些刚起步的人来说,这些工具都相当不错,它们使用了托管服务和一些开源的东西,比如AWS的SageMaker或AzureML。”

Disney:识别电影中的动画面孔

自2012年以来,媒体巨头Disney的工程师和数据科学家一直在构建该公司所谓的“内容基因组”(ContentGenome)的知识图谱,将内容元数据聚合在一起,从而推动Disney庞大的内容库中基于机器学习的搜索和个性化应用。

“这个元数据改进了Disney故事讲述者用于制作内容的工具;激发故事中的迭代创新;通过推荐引擎,数字导航和内容发现提高用户体验;并实现了商业智能。”Disney的开发人员Miquel±ngelFarré、AnthonyAccardo、MarcJunyent、MonicaAlfaro和CescGuitart在今年7月的一篇博客文章中写道。

在这之前,Disney不得不投资一个庞大的内容注释项目,求助于其数据科学家,利用深度学习的图像识别模型,训练一个自动化标记管道,以识别海量的人物,地点和地点图像。

Disney的工程师们最初尝试了包括TensorFlow在内的各种框架,但在2019年,他们决定将PyTorch与之结合起来。工程师们从传统的方向梯度直方图(histogramoforientedgradients,HOG)特征描述器和流行的支持向量机(supportvectormachines,SVM)模型,转向了被称为基于区域卷积神经网络(regionswithconvolutionalneuralnetworks,R-CNN)的目标检测架构版本。在处理Disney内容中常见的真人动作、动画和视觉效果的组合时,后者更为有利。

“在动画片中很难定义什么是人脸,所以我们转向深度学习方法,使用目标检测器,并使用迁移学习。”Disney研究工程师MonicaAlfaro向InfoWorld解释道。在仅仅处理了几千张人脸之后,新模型已经能够大致识别所有三种用例中的人脸。它于2020年1月投入生产。

她表示:“我们现在只使用一种模型来识别三种类型的人脸,这对于像《复仇者联盟》这样的漫威电影来说是很棒的。因为在这类电影中,钢铁侠和TonyStark或者任何一个戴面具的角色都需要被识别出。”

由于工程师要处理如此大量的视频数据以并行地训练和运行模型,所以他们也希望在投入生产阶段时,能够运行在昂贵的高性能GPU上。

CPU的转变让工程师可以更快地重新训练和更新模型。它还加快了向Disney各个小组分发测试结果的速度,将处理时间从一部长篇电影的大约一个小时缩短到今天的5到10分钟。

“TensorFlow目标检测器在生产环境中会存在内存问题,并且很难更新;而PyTorch有同样的目标检测器和Faster-RCNN,因此我们开始使用PyTorch来解决所有的问题。”Alfaro说。

对工程团队来说,这种从一个框架到另一个框架的转换也出奇地简单。“(改用PyTorch)很容易,因为它都是内置的,你只需要插入一些功能,就可以快速上手,所以它的学习曲线并不陡峭。”Alfaro说。

当他们遇到任何问题或瓶颈时,充满活力的PyTorch社区随时提供帮助。

BlueRiverTechnology:除草机器人

BlueRiverTechnology设计了一款机器人,这种机器人结合了数字寻路、集成摄像头计算机视觉技术,可以在农作物不受影响的情况下,在近乎实时的环境给杂草喷洒除草剂,帮助农民更有效地节约昂贵且可能对环境造成破坏的除草剂。

这家位于加利福尼亚州森尼韦尔的公司在2017年吸引了重型设备制造商JohnDeere的关注,当时该公司以3.05亿美元的价格被收购,目的是将该技术整合到其农业设备中。

BlueRiverTechnology的研究人员利用各种深度学习框架进行了试验,试图训练计算机视觉模型来识别杂草和农作物之间的差异,这对处理棉花植物来说是一个巨大的挑战,因为棉花植物不幸地与杂草相似。

BlueRiverTechnology的计算机视觉和机器学习主管ChrisPadwick在他8月份的一篇博客文章中写道,训练有素的农艺师被征召来执行人工图像标注任务,并使用PyTorch训练卷积神经网络(CNN),以“分析每一帧图像,并生成像素级精度的地图,显示农作物和杂草所在的位置”。

“像其他公司一样,我们也试过Caffe、TensorFlow和PyTorch,”Padwick告诉InfoWorld。“对我们来说,这几乎是一种开箱即用的方法,我们既没有收到过任何bug报告,也没有出现过阻塞bug。在分布式计算中,它真的很亮眼,它的性能确实比TensorFlow要好,TensorFlow在数据并行方面相当复杂。”

Padwick表示,PyTorch框架的普及和简单性使得他在快速招聘新员工方面具有优势。话虽如此,Padwick仍然梦想着这样一个世界:“人们可以在自己喜欢的环境中进行开发。”有些人喜欢用ApacheMXNet、Darknet或Caffe进行研究,但是在生产环境中,它们只能使用单一语言,而PyTorch拥有我们成功所需的一切。”

Datarock:基于云端的采矿业图像分析

Datarock是由一群地球科学家创立的澳大利亚初创公司,正在将计算机视觉技术应用于采矿业。更具体地说,它的深度学习模型正在帮助地质学家比以前更快地分析钻探岩心样本图像。

通常,地质学家会一厘米一厘米地仔细研究这些样品,以评估矿物学和结构;而工程师则会寻找诸如断层、裂缝和岩石质量等物理特征。这一过程是缓慢的,而且很容易出现人为错误。

Datarock的首席运营官BrentonCrawford告诉InfoWorld:“计算机可以像工程师一样看到岩石。如果你能在图像中看到,我们就可以训练一个模型去分析它,就像训练一个人类一样。”

与BlueRiver类似,Datarock在生产环境中使用了RCNN模型的一种变体,研究人员转向数据增强技术,以便在早期阶段收集足够的训练数据。

“在最初的发现期之后,团队开始将各种技术结合起来,为钻探岩心图像创建一个图像处理工作流程。这项工作包括开发一系列深度学习模型,能够将原始图像处理为结构化格式,并分割重要的地质信息。”研究人员在一篇博客文章中写道。

通过DataRock的技术,客户只需半小时就能得到结果,而手工记录结果需要五、六小时。Crawford说,这样可以使地质学家不再从事更繁重的工作。但是,“当我们将更困难的事情实现自动化时,我们确实会受到一些阻力,不得不解释为什么要把它们作为训练模型和使反馈循环转动的系统的一部分。”

和许多公司训练深度学习计算机视觉模型一样,Datarock也是从TensorFlow开始的,但很快就转向了PyTorch。

Datarock的机器学习主管DuyTinTruong告诉InfoWorld:“一开始我们使用的是TensorFlow,但是由于某些神秘的原因它会崩溃。”他说,“当时PyTorch和Detecton2刚刚发布,非常符合我们的需求,所以经过一些测试之后,我们发现它更容易调试和工作,而且占用的内存也更少,所以我们进行了转换。”

Datarock还报告说,当在GPU上运行模型时,从TensorFlow到PyTorch和Detectron2的推理性能提高了4倍,而在CPU上则提高了3倍。

Truong列举了PyTorch不断增长的社区、设计良好的界面、易于使用和更好的调试作为切换的理由,并指出,尽管“从界面的角度来看,它们是完全不同的,但如果你懂TensorFlow,那么切换起来是相当容易的,特别是如果你了解Python的时候。”

作者介绍:

ScottCarey,经验丰富的多媒体记者,专注于科技、旅游、体育、文化和社交媒体。目前是记者和编辑,管理着一支由B2B记者组成的优秀团队,并且具有丰富的数字技能,在为网络和纸媒撰稿、报道事件、电视记者和电台记者方面的经验。

责任编辑:lq

免责声明:本站旨在传递信息,不代表有传资讯的观点和立场。本站遵循行业规范,如转载您的文章未标注版权,请联系我们(QQ:78799268)改正。本站的原创文章,如若转载,请注明出处:http://www.ainoline.cn/jiaoyu/shenduxuexi/16371.html

发表评论

电子邮件地址不会被公开。 必填项已用*标注

联系我们

在线咨询:点击这里给我发消息

邮件:78799268@qq.com

工作时间:周一至周五,9:30-18:30,节假日休息

QR code