Pruning by Explaining: A Novel Criterion for Deep Neural Network Pruning

原文鏈接:https://arxiv.org/pdf/1912.08881v1.pdf? ?發(fā)表:

code:

編輯:Daniel

本文提出了一種新的剪枝標(biāo)準(zhǔn)豪嗽,將一種解釋圖像分類的方法LRP用于剪枝。


首先介紹一下LRP,通過定義每一層對上一層結(jié)果的貢獻和相關(guān)性册着,最后推導(dǎo)至原圖中的像素層面。

上圖左側(cè)為一般的圖片分類過程螃宙,右側(cè)為解釋過程雁社,從預(yù)測結(jié)果向前傳播,并將預(yù)測結(jié)果關(guān)聯(lián)到每一層的特征久脯,最終在輸入圖像的像素層面進行可視化纳胧,得出每個像素對于最終分類結(jié)果的貢獻。

對于LRP來說帘撰,每層的相關(guān)性是守恒的跑慕,即

對于一般的多層網(wǎng)絡(luò),示意圖如下:

左側(cè)為一般的前向傳播骡和,右側(cè)為LRP分配示意圖相赁,為了滿足各層相關(guān)性守恒,有以下兩式:

左邊層等于右邊輸入之和


右邊層等于向左邊輸出之和

再將權(quán)重考慮進來慰于,可得:

這個地方公式比較多钮科,看一下原文會更清楚。(原文鏈接:https://www.scienceopen.com/document?vid=09930944-7198-4cbd-840d-a8e5a7dc053b)

對于一般的多層網(wǎng)絡(luò)婆赠,前向傳播中有如下公式:

其中w為權(quán)重绵脯,b為偏置佳励,g為激活函數(shù)。

在經(jīng)過最后一個全連接層后蛆挫,f(x)被定義為:

因此兩個節(jié)點之間的關(guān)聯(lián)值被定義為:

將一層的節(jié)點相加可得:

為了避免由于z太小導(dǎo)致R取到無界值赃承,文中加入ε≥0:

然后文中又將正數(shù)和負(fù)數(shù)分開處理,得到

最終得到:(說實話這一步?jīng)]太懂)



介紹完了LRP悴侵,這篇剪枝論文和LRP這篇是同一群作者瞧剖,所以就把LRP直接用在了剪枝上。LRP通過類似后向傳播的方式可免,將預(yù)測結(jié)果分配到輸入圖像的每個像素點上抓于,以此來觀察每個像素點對于最終分類結(jié)果的貢獻。而在LRP傳播過程中浇借,網(wǎng)絡(luò)中每層的每個filter也都積累了相應(yīng)的貢獻值捉撮,作者將每層中貢獻值較小的filter刪除,來達到剪枝的目的妇垢,示意圖如下:

上圖左側(cè)為一般的前向傳播巾遭,中間為LRP傳播過程,右側(cè)為剪枝過程闯估。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末灼舍,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子睬愤,更是在濱河造成了極大的恐慌片仿,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,718評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件尤辱,死亡現(xiàn)場離奇詭異砂豌,居然都是意外死亡,警方通過查閱死者的電腦和手機光督,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,683評論 3 385
  • 文/潘曉璐 我一進店門阳距,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人结借,你說我怎么就攤上這事筐摘。” “怎么了船老?”我有些...
    開封第一講書人閱讀 158,207評論 0 348
  • 文/不壞的土叔 我叫張陵咖熟,是天一觀的道長。 經(jīng)常有香客問我柳畔,道長馍管,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,755評論 1 284
  • 正文 為了忘掉前任薪韩,我火速辦了婚禮确沸,結(jié)果婚禮上捌锭,老公的妹妹穿的比我還像新娘。我一直安慰自己罗捎,他們只是感情好观谦,可當(dāng)我...
    茶點故事閱讀 65,862評論 6 386
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著桨菜,像睡著了一般豁状。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上倒得,一...
    開封第一講書人閱讀 50,050評論 1 291
  • 那天替蔬,我揣著相機與錄音,去河邊找鬼屎暇。 笑死,一個胖子當(dāng)著我的面吹牛驻粟,可吹牛的內(nèi)容都是我干的根悼。 我是一名探鬼主播,決...
    沈念sama閱讀 39,136評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼蜀撑,長吁一口氣:“原來是場噩夢啊……” “哼挤巡!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起酷麦,我...
    開封第一講書人閱讀 37,882評論 0 268
  • 序言:老撾萬榮一對情侶失蹤矿卑,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后沃饶,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體母廷,經(jīng)...
    沈念sama閱讀 44,330評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,651評論 2 327
  • 正文 我和宋清朗相戀三年糊肤,在試婚紗的時候發(fā)現(xiàn)自己被綠了琴昆。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,789評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡馆揉,死狀恐怖业舍,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情升酣,我是刑警寧澤舷暮,帶...
    沈念sama閱讀 34,477評論 4 333
  • 正文 年R本政府宣布,位于F島的核電站噩茄,受9級特大地震影響下面,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜巢墅,卻給世界環(huán)境...
    茶點故事閱讀 40,135評論 3 317
  • 文/蒙蒙 一诸狭、第九天 我趴在偏房一處隱蔽的房頂上張望券膀。 院中可真熱鬧,春花似錦驯遇、人聲如沸芹彬。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,864評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽舒帮。三九已至,卻和暖如春陡叠,著一層夾襖步出監(jiān)牢的瞬間玩郊,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,099評論 1 267
  • 我被黑心中介騙來泰國打工枉阵, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留译红,地道東北人。 一個月前我還...
    沈念sama閱讀 46,598評論 2 362
  • 正文 我出身青樓兴溜,卻偏偏與公主長得像侦厚,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子拙徽,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,697評論 2 351

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