Karpathy更新深度學習開源框架排名：TensorFlow第一，PyTorch第二

新智元報道

來源：Twitter

作者：文強

【新智元導讀】上周Keras作者Fran?ois Chollet才曬了使用谷歌搜索Index得到的開源深度學習框架排名，前天好久沒出山的大神Andrej Karpathy拋出更新，使用更精確的數據，得到過去6年、4300篇ArXiv機器學習論文里提到深度學習框架：PyTorch漲勢驚人。

上周，Keras作者、谷歌研究科學家Fran?ois Chollet曬出一張圖，他使用Google Search Index，展示了過去三個月，ArXiv上提到的深度學習框架排行，新智元也做了報道：

TensorFlow排名第一，Keras排名第二，之後依次是Caffe、PyTorch、Theano，MXNet、Chainer，以及CNTK。

不少人評論，咦，PyTorch這麼靠後？CNTK更是不科學……

昨天，手握ArXiv Sanity大數據、現在主管特斯拉人工智慧部門的Andrej Karpathy，給出了更加精確的結果。

過去一個月，各個框架在論文中被提到（單次計算）的比例分別是：TensorFlow 14.3%，PyTorch 4.7％，Keras 4.0％，Caffe 3.8％，Theano 2.3％，Torch 1.5％，MXNet、Chainer和CNTK均小於1％：

上圖中，綠色是Theano，紫紅色是TensorFlow（上升最高的那條），藍色是Keras，棕黃色是Caffe，紫色是Torch，紫紅色是PyTorch，淺綠是MXNet，藍色是Chainer，紅色是CNTK。

沒錯，上面的數據確實是過去一個月，Karpathy也在後面給出了更正說明。

Karpathy表示，綜合過去6年發表在ArXiv的4300篇機器學習論文（數據來源：cs.[CV|CL|LG|AI|NE]/stat.ML），根據其中各框架被提及的次數得到的總排名是：TensorFlow 5.9%，Caffe 5.4%，Theano 3.2%，Keras 2.3%，Torch 1.6％，PyTorch 1%，其他0.5%

Theano餘威仍在，PyTorch漲勢驚人。

要知道，在2017年3月，Karpathy第一次做這個全面統計的時候，各框架的排名是：

那時候PyTorch才剛剛發布不久，Theano也沒有說退休。

一年過去，如果有什麼沒有變，那就是王者TensorFlow——2017年3月提交的所有論文中，有10％提到了TensorFlow。 當然，並非每篇論文都會聲明所使用的框架，但如果假設論文以某種固定的隨機概率獨立於框架來聲明框架，那麼機器學習社區里看起來大約有40％的人都在使用TensorFlow（或者更多，如果把Keras也計算進去）。

2017年3月的數據：紫色是Caffe，綠色是TensorFlow，紅色是Keras，黃色是Torch，淺藍色是PyTorch，藍色是Theano

我們期待最新一度的統計結果。

P.S. 真的沒有找到PaddlePaddle。

了解更多：Andrej Karpathy從Arxiv論文看機器學習趨勢：

https://medium.com/@karpathy/a-peek-at-trends-in-machine-learning-ab8a1085a106

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！