卷積網(wǎng)絡(luò)的可視化解釋-類激活映射

wk

引言

在整篇文章論述開始之前鸣哀,我們先做一些概念性的講解鋪墊架忌。卷積神經(jīng)網(wǎng)絡(luò)的各層卷積單元在模型網(wǎng)絡(luò)中實際上有充當(dāng)了目標(biāo)檢測器的作用吞彤,盡管沒有提供對目標(biāo)位置的監(jiān)督我衬。雖然其擁有在卷積層中定位對象的非凡能力,但當(dāng)使用全連接層進(jìn)行分類時饰恕,這種能力就會喪失挠羔。基于此埋嵌,提出了CAM(類激活映射)的概念破加,采用全局平均池化,以熱力圖的形式告訴我們雹嗦,模型通過哪些像素點得知圖片屬于某個類別范舀,使模型透明化和具有可解釋性,如圖1所示:

(圖1) 具有可解釋性的熱力圖

1. Global Average Pooling的工作機制

(圖2) global average pooling

這里我們假設(shè)最后的類別數(shù)為n了罪,最后一層含有n個特征圖锭环,求每張?zhí)卣鲌D所有像素的平均值,后接入一個有n個神經(jīng)元的全連接層泊藕。要有n個特征圖的原因在于辅辩,每個特征圖主要提取了某一類別相關(guān)的某些特征。

2. 什么是CAM娃圆?

CNN最后一層特征圖富含有最為豐富類別語意信息(可以理解為高度抽象的類別特征)玫锋,因此CAM基于最后一層特征圖進(jìn)行可視化。CAM能讓我們對CNN網(wǎng)絡(luò)有很好的解釋作用讼呢,利用特征圖權(quán)重疊加的原理獲得熱圖撩鹿,詳細(xì)工作原理如圖3所示。

(圖3) CAM原理圖

設(shè)最后一層有n張?zhí)卣鲌D悦屏,記為A^1,A^2,...A^n三痰,分類層中一個神經(jīng)元有n個權(quán)重吧寺,一個神經(jīng)元對應(yīng)一類,設(shè)第i個神經(jīng)元的權(quán)重為w^1散劫,w^2,...w^n稚机,則第c類的CAM的生成方式為:

L_{CAM}^c = \sum_{i=1}^n w_i^cA^i

生成的CAM大小與最后一層特征圖的大小一致,接著進(jìn)行上采樣即可得到與原圖大小一致的CAM获搏。

2.1. 為什么如此計算可以得到類別相關(guān)區(qū)域

用GAP表示全局平均池化函數(shù)赖条,沿用上述符號,第c類的分類得分為S_c常熙,GAP的權(quán)重為w_i^c纬乍,特征圖大小為c_1*c_2,第i個特征圖第k行第j列的像素值為A_{kj}^i裸卫,則有:

S_c = \sum_{i=1}^n w_i^c GAP(A_i)

= \sum_{i=1}^n w_i^c \frac 1Z \sum_{k=1}^{c_1} \sum_{j=1}^{c_2}A_{kj}^i

= \frac 1Z \sum_{i=1}^n \sum_{k=1}^{c_1} \sum_{j=1}^{c_2} w_i^c A_{kj}^i

特征圖中的一個像素對應(yīng)原圖中的一個區(qū)域仿贬,而像素值表示該區(qū)域提取到的特征,由上式可知S_c的大小由特征圖中像素值與權(quán)重決定墓贿,特征圖中像素值與權(quán)重的乘積大于0茧泪,有利于將樣本分到該類,即CNN認(rèn)為原圖中的該區(qū)域具有類別相關(guān)特征聋袋。式1就是計算特征圖中的每個像素值是否具有類別相關(guān)特征队伟,如果有,我們可以通過上采樣幽勒,看看這個像素對應(yīng)的是原圖中的哪一部分嗜侮。GAP的出發(fā)點也是如此,即在訓(xùn)練過程中讓網(wǎng)絡(luò)學(xué)會判斷原圖中哪個區(qū)域具有類別相關(guān)特征啥容,由于GAP去除了多余的全連接層锈颗,并且沒有引入?yún)?shù),因此GAP可以降低過擬合的風(fēng)險咪惠』髦ǎ可視化的結(jié)果也表明,CNN正確分類的確是因為注意到了原圖中正確的類別相關(guān)特征硝逢。

2.2. CAM缺陷

需要修改網(wǎng)絡(luò)結(jié)構(gòu)并重新訓(xùn)練模型姨拥,導(dǎo)致在實際應(yīng)用中并不方便。

3. Grad-CAM

3.1. Grad-CAM結(jié)構(gòu)

(圖4) Grad-CAM結(jié)構(gòu)圖

Grad-CAM 和 CAM 基本思路一樣渠鸽,區(qū)別就在于如何獲取每個特征圖的權(quán)重叫乌,采用了梯度的全局平均來計算權(quán)重。定義了Grad-CAM中第\kappa個特征圖對應(yīng)類別c的權(quán)重:
\alpha_\kappa^c = \frac 1Z \sum_i \sum_j \frac {y^c}{\alpha A_{ij}^k}

其中徽缚,Z表示特征圖像素個數(shù)憨奸,y^c表示第c類得分梯度,A_{ij}^k表示第k個特征圖中凿试,(i,j)位置處的像素值排宰。然后再求得所有的特征圖對應(yīng)的類別的權(quán)重后進(jìn)行加權(quán)求和似芝,這樣便可以得到最后的熱力圖,求和公式如下:
L_{Grad-CAM}^c = ReLU(\sum_k \alpha_k^c A^k)

3.2. Grad-CAM效果

(圖5) Grad-CAM效果演示

4. 結(jié)論

可視化可以進(jìn)一步區(qū)別分類板甘,準(zhǔn)確地更好地揭示分類器的可信賴性党瓮,并幫助識別數(shù)據(jù)集中的偏差。真正的AI應(yīng)用盐类,也更應(yīng)該讓人們信任和使用它的行為寞奸。

5. 代碼實現(xiàn)

https://github.com/jacobgil/keras-cam

參考文獻(xiàn)

  • B. Zhou, A. Khosla, A. Lapedriza, A. Oliva, and A. Torralba. Object detectors emerge in deep scene cnns. International Conference on Learning Representations, 2015.
  • Computers - Computer Graphics; Investigators from Georgia Institute of Technology Have Reported New Data on Computer Graphics (Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization). 2020, :355-.
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市在跳,隨后出現(xiàn)的幾起案子枪萄,更是在濱河造成了極大的恐慌,老刑警劉巖猫妙,帶你破解...
    沈念sama閱讀 217,509評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件瓷翻,死亡現(xiàn)場離奇詭異,居然都是意外死亡割坠,警方通過查閱死者的電腦和手機齐帚,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,806評論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來韭脊,“玉大人童谒,你說我怎么就攤上這事单旁』Ω幔” “怎么了?”我有些...
    開封第一講書人閱讀 163,875評論 0 354
  • 文/不壞的土叔 我叫張陵象浑,是天一觀的道長蔫饰。 經(jīng)常有香客問我,道長愉豺,這世上最難降的妖魔是什么篓吁? 我笑而不...
    開封第一講書人閱讀 58,441評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮蚪拦,結(jié)果婚禮上杖剪,老公的妹妹穿的比我還像新娘。我一直安慰自己驰贷,他們只是感情好盛嘿,可當(dāng)我...
    茶點故事閱讀 67,488評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著括袒,像睡著了一般次兆。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上锹锰,一...
    開封第一講書人閱讀 51,365評論 1 302
  • 那天芥炭,我揣著相機與錄音漓库,去河邊找鬼。 笑死园蝠,一個胖子當(dāng)著我的面吹牛渺蒿,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播彪薛,決...
    沈念sama閱讀 40,190評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼蘸嘶,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了陪汽?” 一聲冷哼從身側(cè)響起训唱,我...
    開封第一講書人閱讀 39,062評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎挚冤,沒想到半個月后况增,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,500評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡训挡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,706評論 3 335
  • 正文 我和宋清朗相戀三年澳骤,在試婚紗的時候發(fā)現(xiàn)自己被綠了埂伦。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片镰绎。...
    茶點故事閱讀 39,834評論 1 347
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖罢缸,靈堂內(nèi)的尸體忽然破棺而出肤京,到底是詐尸還是另有隱情颊艳,我是刑警寧澤,帶...
    沈念sama閱讀 35,559評論 5 345
  • 正文 年R本政府宣布忘分,位于F島的核電站棋枕,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏妒峦。R本人自食惡果不足惜重斑,卻給世界環(huán)境...
    茶點故事閱讀 41,167評論 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望肯骇。 院中可真熱鬧窥浪,春花似錦、人聲如沸笛丙。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,779評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽若债。三九已至符相,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背啊终。 一陣腳步聲響...
    開封第一講書人閱讀 32,912評論 1 269
  • 我被黑心中介騙來泰國打工镜豹, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人蓝牲。 一個月前我還...
    沈念sama閱讀 47,958評論 2 370
  • 正文 我出身青樓趟脂,卻偏偏與公主長得像,于是被迫代替她去往敵國和親例衍。 傳聞我的和親對象是個殘疾皇子昔期,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,779評論 2 354

推薦閱讀更多精彩內(nèi)容