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