微軟攜手FB推出開源項目 打造共享神經(jīng)網(wǎng)絡(luò)模型
9 月 8 日消息，據(jù) VentureBeat 報道，軟件巨頭微軟和社交網(wǎng)絡(luò) Facebook 今天宣布推出新的開源項目，旨在為不同編程框架的神經(jīng)網(wǎng)絡(luò)創(chuàng)建共享模型。這個新項目名為“開放神經(jīng)網(wǎng)絡(luò)交換”(ONNX)，它將允許人們在 Cognitive Toolkit、PyTorch 以及 Caffe2 之間共享模型。
ONNX 應(yīng)該可以幫助解決目前機器學(xué)習(xí)生態(tài)系統(tǒng)中的一個關(guān)鍵問題。開發(fā)和執(zhí)行神經(jīng)網(wǎng)絡(luò)和其他機器學(xué)習(xí)系統(tǒng)有很多不同的框架，但是它們都是截然不同的，而且不支持互操作。通過使用 ONNX，F(xiàn)acebook 可以導(dǎo)出使用 PyTorch 創(chuàng)建的訓(xùn)練模型，并使用 Caffe2 進行推理。這對于在研究中創(chuàng)建模型，并將其應(yīng)用于 Caffe2 中非常重要。微軟表示，它正致力于發(fā)布一款 Cognitive Toolkit 以支持 ONNX 項目。
這套系統(tǒng)的工作原理是，通過跟蹤這些框架在運行時如何生成神經(jīng)網(wǎng)絡(luò)，然后使用該信息創(chuàng)建可移動的通用計算圖。這是可能的，因為每個框架在計算時都會產(chǎn)生非常相似的結(jié)果，即使更高層次的表示存在差異。目前，ONNX ***的問題是它與其他流行的機器學(xué)習(xí)框架不兼容，包括起源于谷歌的 TensorFlow 和亞馬遜偏愛的機器學(xué)習(xí)框架 Apache MXNet。
然而，加大對該項目的支持非常重要。Facebook 表示，為了支持這個項目，它必須對 PyTorch 和 Caffe2 進行修改。微軟和 Facebook 都表示，他們希望開源社區(qū)能夠幫助他們改進 ONNX，以便將來支持更多框架。此外，ONNX 不支持哪些更為復(fù)雜的網(wǎng)絡(luò)，比如在 PyTorch 中創(chuàng)建動態(tài)流控制。但 Facebook 計劃在未來加入這樣的計劃。
