神經(jīng)網(wǎng)絡(luò)（NN）架構(gòu)圖制作起來往往費(fèi)時(shí)耗力，很多時(shí)候機(jī)器學(xué)習(xí)研究人員需要從頭開始構(gòu)建相關(guān)圖。

近日，機(jī)器之心在 GitHub 上發(fā)現(xiàn)這樣一個(gè)項(xiàng)目 ——NN-SVG，可以為研究人員解決這一煩心事。項(xiàng)目作者 Alexander Lenail 為 MIT 計(jì)算系統(tǒng)生物學(xué)二年級(jí)博士生，研究興趣主要是創(chuàng)建用于生物分析、理解和設(shè)計(jì)的計(jì)算工具。

項(xiàng)目地址：https://github.com/alexlenail/NN-SVG

據(jù)作者介紹，NN-SVG 是一個(gè)參數(shù)化而非手動(dòng)構(gòu)建神經(jīng)網(wǎng)絡(luò)架構(gòu)圖的交互工具。該工具還能夠?qū)?gòu)建的圖導(dǎo)出至可縮放矢量圖形（Scalable Vector Graphics, SVG）文件，適合用在學(xué)術(shù)論文或網(wǎng)頁(yè)中。

具體地，NN-SVG 工具可以構(gòu)建以下三種風(fēng)格的神經(jīng)網(wǎng)絡(luò)圖：

• 全連接神經(jīng)網(wǎng)絡(luò)（FCNN）
• LeNet 論文中提出的一類卷積神經(jīng)網(wǎng)絡(luò)（CNN）
• 風(fēng)格類似于 AlexNet 論文中提出的深度神經(jīng)網(wǎng)絡(luò)（DNN）

其中，前兩類神經(jīng)網(wǎng)絡(luò)圖形使用 JavaScript 庫(kù) D3.js 完成的，后一類由 Three.js 完成。NN-SVG 的亮點(diǎn)在于可以根據(jù)使用者的喜好來改變神經(jīng)網(wǎng)絡(luò)圖的大小、顏色和布局參數(shù)。

我們僅以 FCNN 風(fēng)格的生成圖為例進(jìn)行講解。下圖中都是可以改變的組件，如邊緣寬度、邊緣不透明度、邊緣權(quán)重對(duì)應(yīng)的顏色、節(jié)點(diǎn)直徑、層間距、隨機(jī)權(quán)重等。

比如，改變「邊緣權(quán)重對(duì)應(yīng)的顏色」前后：

再比如，添加「18 和 20 個(gè)權(quán)重」前后：

LeNet 風(fēng)格和 AlexNet 風(fēng)格的神經(jīng)網(wǎng)絡(luò)圖構(gòu)建過程與之類似。總之，在這個(gè)工具中，一切都是可以交互的。

項(xiàng)目作者希望 NN-SVG 工具可以節(jié)省機(jī)器學(xué)習(xí)研究人員的時(shí)間，也希望這個(gè)軟件在某些情境下可以作為教學(xué)工具。

感興趣的小伙伴可以去試試。

NN-SVG 地址：https://alexlenail.me/NN-SVG/LeNet.html