雷鋒網消息。在昨晚開幕的 F8 年度開發者大會上,Facebook 發布了一款全新的開源深度學習框架 Caffe2。按照官網介紹,它最大的特點就是輕量、模塊化和擴展性,即一次編碼,到處運行(和 Java 的宣傳語類似)。說得更直白一點,就是 Caffe2 可以方便地為手機等移動終端設備帶來 AI 加持,讓 AI 從雲端走向終端。
Caffe2 相比 Caffe 更新了什麼
Caffe2 官方博客表示:長期以來,提起 AI 模型的訓練和部署,人們通常都會聯想到大數據中心或超級計算機。之所以會出現這種現象,是因為小型計算設備在大規模圖像、視頻、文本和語音處理上具有明顯的短板,無論在速度還是可靠性上都遠遠不及大型數據中心。因此在行動裝置上部署 AI 模型,使其能夠快速準確地處理相關分析任務,一直是一項有待解決的難題。要解決這個難題,讓 AI 終端化,除了有賴於移動硬體平臺的發展之外,在軟體框架層面也需要一個靈活輕便的框架支持。
而 Caffe2 正是這樣一個框架。它是在此前流行的開源框架 Caffe 基礎上的重構和升級,一方面集成了諸多新出現的算法和模型,另一方面在保證運算性能和可擴展性的基礎上重點加強了框架在輕量級硬體平臺的部署能力。按照官方介紹,Caffe2 除了支持現在已經成為標配的雲端 GPU 加速之外,還可以部署在包括 iOS,Android,英偉達 Tegra X1 和樹莓派(Raspberry Pi)等在內的各種移動平臺上。用戶只需要加載 Caffe2 框架,然後通過幾行簡單的 API 接口調用(Python 或 C++),就能在手機 App 上實現包括圖像識別、自然語言處理和計算機視覺等在內的各種 AI 功能。
Caffe 和 Caffe2 背後的主要開發者,加州大學伯克利分校博士賈揚清在接受外媒採訪時表示:
「幾乎所有的深度學習框架都或多或少地提到了擴展性,但我們可以相當自信地說,Caffe2 在這方面要優於其他框架。」
在 Caffe 平臺的另一項核心競爭力:Model Zoo 社區方面,Caffe2 也提供了完整的支持。此前,為數眾多的 Caffe 開發者和研究人員都會經常在 Model Zoo 社區分享一些已經實現好的組件或模型,其他開發者可以直接下載和修改,這大大加快了大家的開發進度,避免了重複勞動。在更新了 Caffe2 之後,官方表示用戶依然可以在 Model Zoo 社區上傳相關組件,而且官方提供了腳本工具可以將老版 Caffe 模塊一鍵轉換到 Caffe2 平臺,詳情參見:
http://caffe2.ai/docs/zoo.html
Caffe2 得到了哪些支持值得注意的是,除了 Caffe2 框架本身之外,Facebook 還在發布會上宣布了一系列和 Caffe2 相關的雲平臺和硬體平臺支持。
雲平臺方面,亞馬遜 AWS 表示目前在旗下的 Deep Learning AMI 上已經加入了 Caffe2 支持,用戶可以直接在 AWS 運行相關樣例。微軟 Azure 也表示已經在旗下 Data Science Virtual Machine (DSVM)上兼容 Caffe2。
硬體平臺方面,Nvidia 加速計算總經理兼副總裁 Ian Buck 在博客中表示,旗下的 GPU 可以接入 Facebook 伺服器進行深度學習計算,並將幫助更多人學習使用 Caffe2。高通表示,公司正在與 Facebook 合作,進一步優化 Caffe2 框架,並將深度適配旗下 Snapdragon 神經處理引擎(NPE)。英特爾表示,未來將集成旗下數學內核庫(MKL)到 Caffe2 中。
目前,Caffe2 框架已經被 Facebook 內部採用,開發者和研究人員們正在使用該框架提供的各種工具訓練大型的機器學習模型,並為 Facebook 旗下的移動應用提供 AI 智能體驗。
Caffe2 官網:http://caffe2.ai/
GitHub 開源地址:https://github.com/caffe2/caffe2
雷鋒網(公眾號:雷鋒網(公眾號:雷鋒網))相關閱讀:
F8 2017 | AR、VR、AI一個不落,Facebook 帶你通往未來世界
盤點四大民間機器學習開源框架:Theano、Caffe、Torch 和 SciKit-learn
雷鋒網版權文章,未經授權禁止轉載。詳情見轉載須知。