點(diǎn)擊上方“程序員大白”,選擇“星標(biāo)”公眾號(hào)
重磅干貨,第一時(shí)間送達(dá)

機(jī)器學(xué)習(xí)從業(yè)者,從此不再愁如何畫(huà)神經(jīng)網(wǎng)絡(luò)圖了。

論文、博客寫(xiě)好了,里面的圖可怎么畫(huà)?對(duì)于很多研究人員和開(kāi)發(fā)者來(lái)說(shuō),內(nèi)容的「可視化」是一個(gè)大問(wèn)題。如果從頭開(kāi)始畫(huà),配色、空間布局都很傷腦筋,而且畫(huà)丑了也拿不出手,要是有模板可以套就好了。這套模板名叫 ML Visuals,是專(zhuān)為解決神經(jīng)網(wǎng)絡(luò)畫(huà)圖問(wèn)題設(shè)計(jì)的。下邊這張圖是不是很面熟?這就是大名鼎鼎的 Transformer 架構(gòu),右邊是谷歌大腦論文《Attention Is All You Need》中的 Transformer 架構(gòu)原圖,左邊是用上述模板重新畫(huà)出來(lái)的,還原度是不是很高,而且看起來(lái)很漂亮?
ML Visuals 提供了更加專(zhuān)業(yè)、美觀、恰當(dāng)?shù)囊曈X(jué)效果和圖形,可以幫助機(jī)器學(xué)習(xí)社區(qū)改善科學(xué)溝通。用戶(hù)可以在 PPT 或博客中免費(fèi)使用其中的視覺(jué)效果和圖形。該項(xiàng)目由 dair.ai 社區(qū)創(chuàng)建和維護(hù)(dair.ai 社區(qū)致力于民主化人工智能研究、教育和技術(shù)),社區(qū)成員將繼續(xù)更新,添加更常見(jiàn)的圖形和基本要素。用戶(hù)可以免費(fèi)下載、復(fù)制、分發(fā)、重用這里的模板,還可以根據(jù)自己的需求進(jìn)行自定義。ML 專(zhuān)用畫(huà)圖模板,拯救手殘黨ML Visuals 項(xiàng)目中每個(gè)好看又復(fù)雜的圖都是由一個(gè)個(gè)小模塊組成的。在這套模板中,作者用到了多種模塊,如表示過(guò)程、操作或轉(zhuǎn)換的圓角矩形,表示神經(jīng)元或任意操作的小圓圈,表示向量的一排小方塊以及表示多維數(shù)組的網(wǎng)格等。

選擇其中一些模塊組合使用,就可以得到神經(jīng)網(wǎng)絡(luò)的一層:
它們還可以用來(lái)表示神經(jīng)網(wǎng)絡(luò)中的某個(gè)過(guò)程,如下圖中的分詞過(guò)程:
再?gòu)?fù)雜一點(diǎn),還可以畫(huà)出整個(gè)神經(jīng)網(wǎng)絡(luò):
此外,這套模板還可用來(lái)表示機(jī)器學(xué)習(xí)領(lǐng)域的一些概念,如線性回歸:

對(duì)于喜歡「暗黑模式」的同學(xué),這個(gè)項(xiàng)目也照顧到了:
除此之外,該項(xiàng)目還提供了背景模板,方便用戶(hù)進(jìn)行圖形自定義。
該項(xiàng)目使用 Google Slides 對(duì)所有視覺(jué)效果和圖形進(jìn)行維護(hù),用戶(hù)可以通過(guò) Google Slides 使用其中的組件,或執(zhí)行添加、自定義等操作。想要添加自定義圖,只需添加一張新的幻燈片,并重用任意基礎(chǔ)視覺(jué)組件。不過(guò)你需要先獲取編輯權(quán)限(點(diǎn)擊「只能查看」工具欄下的「請(qǐng)求編輯權(quán)限」或者發(fā)郵件至 [email protected] 獲取編輯權(quán)限)。用戶(hù)還可以復(fù)制這些幻燈片并按照自己的意愿進(jìn)行自定義。dair.ai 社區(qū)鼓勵(lì)用戶(hù)添加自己的圖形并允許他人重用。不過(guò),圖形作者最好在幻燈片備注中添加作者信息,方便其他人在重用這些圖形時(shí)進(jìn)行恰當(dāng)?shù)匾谩D形作者或設(shè)計(jì)者還可以添加自己的姓名和郵件地址,方便其他用戶(hù)咨詢(xún)相關(guān)問(wèn)題。此外,圖形作者可以為自己的視覺(jué)圖形提供簡(jiǎn)單的描述,以幫助其他用戶(hù)理解用途。用戶(hù)需要注意的是,在獲取編輯權(quán)限后,不要?jiǎng)h除別人添加的圖形。如果想要進(jìn)行改進(jìn),用戶(hù)可以新增一張幻燈片,并添加改進(jìn)版本。下載幻燈片的過(guò)程很簡(jiǎn)單,只需點(diǎn)擊「文件→下載」,然后選擇合適的格式即可。
除此以外,如果用戶(hù)在圖形自定義方面需要幫助,或者有一些對(duì)他人有益的想法,可以打開(kāi) issue 頁(yè)面(https://github.com/dair-ai/ml-visuals/issues/new)進(jìn)行描述。
推薦閱讀
13個(gè)你一定要知道的PyTorch特性
解讀:為什么要做特征歸一化/標(biāo)準(zhǔn)化?
一文搞懂 PyTorch 內(nèi)部機(jī)制
張一鳴:每個(gè)逆襲的年輕人,都具備的底層能力
關(guān)于程序員大白
程序員大白是一群哈工大,東北大學(xué),西湖大學(xué)和上海交通大學(xué)的碩士博士運(yùn)營(yíng)維護(hù)的號(hào),大家樂(lè)于分享高質(zhì)量文章,喜歡總結(jié)知識(shí),歡迎關(guān)注[程序員大白],大家一起學(xué)習(xí)進(jìn)步!