一個(gè)像素的旅程:CNN可視化項(xiàng)目
點(diǎn)擊上方“小白學(xué)視覺(jué)”,選擇加"星標(biāo)"或“置頂”
重磅干貨,第一時(shí)間送達(dá)
郭一璞 發(fā)自 云凹非寺
量子位 報(bào)道 | 公眾號(hào) QbitAI
CNN是什么?美國(guó)有線電視新聞網(wǎng)嗎?
每一個(gè)對(duì)AI抱有憧憬的小白,在開(kāi)始的時(shí)候都會(huì)遇到CNN(卷積神經(jīng)網(wǎng)絡(luò))這個(gè)詞。
但每次,當(dāng)小白們想了解CNN到底是怎么回事,為什么就能聰明的識(shí)別人臉、聽(tīng)辨聲音的時(shí)候,就懵了,只好理解為玄學(xué):

好吧,維基百科解決不了的問(wèn)題,有人給解決了。

這個(gè)名叫CNN解釋器在線交互可視化工具,把CNN拆開(kāi)了揉碎了,告訴小白們CNN究竟是怎么一回事,為什么可以辨識(shí)物品。
它用TensorFlow.js加載了一個(gè)10層的預(yù)訓(xùn)練模型,相當(dāng)于在你的瀏覽器上跑一個(gè)CNN模型,只需要打開(kāi)電腦,就能了解CNN究竟是怎么回事。

而且,這個(gè)網(wǎng)頁(yè)工具還可以實(shí)現(xiàn)交互,只要點(diǎn)擊其中任何一個(gè)格子——就是CNN中的“神經(jīng)元”,就能顯示它的輸入是哪些、經(jīng)過(guò)了怎樣細(xì)微的變化。
甚至,連每一次卷積運(yùn)算都能看得清。
看清卷積
這個(gè)CNN解釋器的使用方式也非常簡(jiǎn)單:鼠標(biāo)戳戳戳就好了。
單擊神經(jīng)元,進(jìn)入彈性解釋視圖,就可以看到卷積核滑動(dòng)的過(guò)程的動(dòng)畫(huà)模擬:

點(diǎn)擊一個(gè)正在卷積的過(guò)程圖,就可以看到更具體的過(guò)程:

可以看到底層的卷積運(yùn)算過(guò)程,3×3的卷積核是如何經(jīng)過(guò)運(yùn)算被變成1個(gè)數(shù)字的。
看清ReLU和最大池化層
點(diǎn)擊一個(gè)ReLU層的神經(jīng)元,可以看具體過(guò)程,ReLU函數(shù)是這樣工作的:

點(diǎn)擊一個(gè)池化神經(jīng)元,也可以看具體最大池化層是怎樣工作的:

看清CNN是怎么輸出預(yù)測(cè)的
點(diǎn)擊最右側(cè)的輸出神經(jīng)元,進(jìn)入彈性解釋視圖:

可以查看Softmax函數(shù)的詳情:

識(shí)別「貍克」試試?
CNN解釋器默認(rèn)有10張圖片,你也可以增加自己的自定義圖片。
比如這個(gè):

芝士焗燈籠椒?燈籠椒披薩?還是什么鬼?
復(fù)制圖片鏈接或者上傳圖片之后,它經(jīng)過(guò)了10層處理,得出結(jié)論:

是燈籠椒,但也可能是臭蟲(chóng)。
但是,只能歸類(lèi)于右側(cè)原有的10個(gè)品類(lèi)里,比如放一只貍克:

就會(huì)被識(shí)別成espresso。

佐治亞理工中國(guó)博士出品
最后,這個(gè)CNN解釋器的作者是一位中國(guó)小哥,佐治亞理工的Zijie Wang,去年剛開(kāi)始讀機(jī)器學(xué)習(xí)博士,本科畢業(yè)于威斯康星大學(xué)麥迪遜分校,是一位GPA 3.95/4.00的大學(xué)霸。

他也曾經(jīng)做過(guò)一些其他有趣的數(shù)據(jù)可視化項(xiàng)目,比如威斯康星大學(xué)麥迪遜分校的中國(guó)本科生都來(lái)自哪里:

傳送門(mén)
CNN解釋器
https://poloclub.github.io/cnn-explainer/
GitHub
https://github.com/poloclub/cnn-explainer
論文
https://arxiv.org/abs/2004.15004
作者系網(wǎng)易新聞·網(wǎng)易號(hào)“各有態(tài)度”簽約作者
交流群
歡迎加入公眾號(hào)讀者群一起和同行交流,目前有SLAM、三維視覺(jué)、傳感器、自動(dòng)駕駛、計(jì)算攝影、檢測(cè)、分割、識(shí)別、醫(yī)學(xué)影像、GAN、算法競(jìng)賽等微信群(以后會(huì)逐漸細(xì)分),請(qǐng)掃描下面微信號(hào)加群,備注:”昵稱(chēng)+學(xué)校/公司+研究方向“,例如:”張三?+?上海交大?+?視覺(jué)SLAM“。請(qǐng)按照格式備注,否則不予通過(guò)。添加成功后會(huì)根據(jù)研究方向邀請(qǐng)進(jìn)入相關(guān)微信群。請(qǐng)勿在群內(nèi)發(fā)送廣告,否則會(huì)請(qǐng)出群,謝謝理解~
