Runtime Network Routing for Efficient Image Classification

這篇是清華魯繼文組PAMI 2018年的文章响逢,是NIPS 2017年《Runtime neural pruning》的拓展版本
兩個(gè)版本有稍微的出入喊废,主要是加了“Multi-Path Network”的部分。不過(guò)這一部分看起來(lái)更像是ensemble多個(gè)模型追城,故此這部分不會(huì)細(xì)講汰现,主要討論“Runtime Neural Pruning”的實(shí)現(xiàn)細(xì)節(jié)。

這篇文章講了什么

dynamic channle pruning

Motivation

  1. selects an optimal path inside the network
  2. simple image doesn't need complicated network

怎么做

Condition computation:Runtime Nerual Pruning based on the input image and current feature map, that is to say, this is the instance pruning strategy

貢獻(xiàn)

提供了剪枝的新思路掏觉,不再探索靜態(tài)的剪枝策略,保留大網(wǎng)絡(luò)的全部表達(dá)能力
但這里會(huì)有另一個(gè)槽點(diǎn):個(gè)人理解值漫,剪枝是為了可以用更少的參數(shù)澳腹,追求在廉價(jià)設(shè)備上保持性能,所以文章提出的保留全部參數(shù)杨何,動(dòng)態(tài)選擇(剪枝)路徑酱塔,是否違背了剪枝這個(gè)task的motivation

設(shè)計(jì)了怎樣的模型

  1. policy network(RNN):決策保留的channel
  2. backbone

建立在怎樣的假設(shè)之上

剪枝策略十分粗糙,默認(rèn)選擇每層feature map前面的channel晚吞,更難的sample才會(huì)用到后面的channel


2019-07-09 10-12-41 的屏幕截圖.png

實(shí)現(xiàn)細(xì)節(jié)

主要是Policy Network部分

state

2019-07-09 11-46-59 的屏幕截圖.png

將上一層的Fi(feature map) [N, n1, H, W]通過(guò)global average pooling得到一個(gè)[N, n1]延旧,通過(guò)encoderi得到固定長(zhǎng)度的特征作為state[N, C]輸入到RNN中,RNN輸出hidden state[N, C]槽地,通過(guò)decoderi得到這個(gè)decision的Q-value
值得注意的是,對(duì)每一層都有獨(dú)立的encoderi跟decoderi。槽點(diǎn):用大量(冗余)的policy network參數(shù)去對(duì)backbone剪枝(減少參數(shù))

action

ai表示選擇保留前i組的channel

reward

2019-07-09 17-23-16 的屏幕截圖.png

architecture

算法

2019-07-09 10-18-44 的屏幕截圖.png

十分粗略的解讀捌蚊,主要是方便自己以后查看集畅。
水平有限,希望沒有誤人子弟缅糟,有任何問(wèn)題歡迎留言或者私信討論(不定時(shí)查看……)

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末挺智,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子窗宦,更是在濱河造成了極大的恐慌赦颇,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,817評(píng)論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件赴涵,死亡現(xiàn)場(chǎng)離奇詭異媒怯,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)髓窜,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,329評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門扇苞,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人寄纵,你說(shuō)我怎么就攤上這事鳖敷。” “怎么了程拭?”我有些...
    開封第一講書人閱讀 157,354評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵定踱,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我恃鞋,道長(zhǎng)屋吨,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,498評(píng)論 1 284
  • 正文 為了忘掉前任山宾,我火速辦了婚禮至扰,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘资锰。我一直安慰自己敢课,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,600評(píng)論 6 386
  • 文/花漫 我一把揭開白布绷杜。 她就那樣靜靜地躺著直秆,像睡著了一般。 火紅的嫁衣襯著肌膚如雪鞭盟。 梳的紋絲不亂的頭發(fā)上圾结,一...
    開封第一講書人閱讀 49,829評(píng)論 1 290
  • 那天,我揣著相機(jī)與錄音齿诉,去河邊找鬼筝野。 笑死晌姚,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的歇竟。 我是一名探鬼主播挥唠,決...
    沈念sama閱讀 38,979評(píng)論 3 408
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼焕议!你這毒婦竟也來(lái)了宝磨?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,722評(píng)論 0 266
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤盅安,失蹤者是張志新(化名)和其女友劉穎唤锉,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體别瞭,經(jīng)...
    沈念sama閱讀 44,189評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡窿祥,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,519評(píng)論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了畜隶。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片壁肋。...
    茶點(diǎn)故事閱讀 38,654評(píng)論 1 340
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖籽慢,靈堂內(nèi)的尸體忽然破棺而出浸遗,到底是詐尸還是另有隱情,我是刑警寧澤箱亿,帶...
    沈念sama閱讀 34,329評(píng)論 4 330
  • 正文 年R本政府宣布跛锌,位于F島的核電站,受9級(jí)特大地震影響届惋,放射性物質(zhì)發(fā)生泄漏髓帽。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,940評(píng)論 3 313
  • 文/蒙蒙 一脑豹、第九天 我趴在偏房一處隱蔽的房頂上張望郑藏。 院中可真熱鬧,春花似錦瘩欺、人聲如沸必盖。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,762評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)歌粥。三九已至,卻和暖如春拍埠,著一層夾襖步出監(jiān)牢的瞬間失驶,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,993評(píng)論 1 266
  • 我被黑心中介騙來(lái)泰國(guó)打工枣购, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留嬉探,地道東北人擦耀。 一個(gè)月前我還...
    沈念sama閱讀 46,382評(píng)論 2 360
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像甲馋,于是被迫代替她去往敵國(guó)和親埂奈。 傳聞我的和親對(duì)象是個(gè)殘疾皇子迄损,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,543評(píng)論 2 349

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