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