ONNX神經(jīng)網(wǎng)絡(luò)開發(fā)框架互通生態(tài)

ONNX，即 Open Neural Network Exchange ，是微軟和 Facebook 發(fā)布的一個(gè)深度學(xué)習(xí)開發(fā)工具生態(tài)系統(tǒng)，旨在讓 AI 開發(fā)人員能夠隨著項(xiàng)目發(fā)展而選擇正確的工具。

ONNX 所針對(duì)的是深度學(xué)習(xí)開發(fā)生態(tài)中最關(guān)鍵的問(wèn)題之一，在任意一個(gè)框架上訓(xùn)練的神經(jīng)網(wǎng)絡(luò)模型，無(wú)法直接在另一個(gè)框架上用。開發(fā)者需要耗費(fèi)大量時(shí)間精力把模型從一個(gè)開發(fā)平臺(tái)移植到另一個(gè)。因此，如何實(shí)現(xiàn)不同框架之間的互操作性，簡(jiǎn)化從研究到生產(chǎn)的路徑，成為了重中之重。

ONNX 的工作原理是：

實(shí)時(shí)跟蹤某個(gè)神經(jīng)網(wǎng)絡(luò)是如何在這些框架上生成的，接著，使用這些信息創(chuàng)建一個(gè)通用的計(jì)算圖，即符合ONNX 標(biāo)準(zhǔn)的計(jì)算圖。這樣做行得通，是因?yàn)樵谟?jì)算方面，這些框架產(chǎn)生的最終結(jié)果都非常接近，雖然更高級(jí)的表達(dá)不同。

ONNX 目前支持 Caffe2、PyTorch、Microsoft Cognitive Toolkit、Apache MXNet 等工具。