Facebook 開源全新深度學習框架 Caffe2，讓 AI 從雲端走向終端

雷鋒網消息。在昨晚開幕的 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 平台，詳情參見：

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 中。