近日,微軟和Facebook宣布了一個新的合作開源項目,旨在為不同編程框架之間的神經網絡創建共享模型。這個被稱為開發式神經網絡交換(ONNX)的新項目將會在Cognitive Toolkit、PyTorch和Caffe2之間實現模型共享。
ONNX會幫助解決目前機器學習生態系統中的一個關鍵問題。雖然現在有眾多的神經網絡和機器學習系統,但是它們之間是不互通的,無法跨系統操作。通過使用ONNX,Facebook可以導出PyTorch創建的訓練模型,然後使用Caffe2進行推斷。這一功能的實現是非常重要的。微軟表示正在開發支持ONNX的Cognitive Toolkit版本。
該系統會跟蹤使用框架執行生成神經網絡的方式,然後使用該信息去創建一個可以移植的通用算圖。鑒於每種框架在計算時會產生非常類似的最終結果,這種方法是可行的,即使存在著較高級別的差異。
目前,ONNX最大的問題是與其他一些流行的機器學習框架不兼容,包括來自於谷歌的TensorFlow和亞馬遜首選的機器學習框架Apache MXNet。
不過,這對項目的實現並沒有多大的影響。Facebook表示,為了支持該項目,它必須對PyTorch和Caffe2進行更改。微軟和Facebook希望開源社區能夠幫助他們演化ONNX,並在未來實現更多框架的支持。
此外,ONNX暫不支持的那些更復雜的網絡,比如PyTorch中動態流量控制所創建的網絡,也是Facebook計劃在未來添加的功能。
已有 322 位網友參與,快來吐槽:
發表評論