Runtime Network Routing for Efficient Image Classification

这篇是清华鲁继文组PAMI 2018年的文章,是NIPS 2017年《Runtime neural pruning》的拓展版本
两个版本有稍微的出入,主要是加了“Multi-Path Network”的部分。不过这一部分看起来更像是ensemble多个模型,故此这部分不会细讲,主要讨论“Runtime Neural Pruning”的实现细节。

这篇文章讲了什么

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

贡献

提供了剪枝的新思路,不再探索静态的剪枝策略,保留大网络的全部表达能力
但这里会有另一个槽点:个人理解,剪枝是为了可以用更少的参数,追求在廉价设备上保持性能,所以文章提出的保留全部参数,动态选择(剪枝)路径,是否违背了剪枝这个task的motivation

设计了怎样的模型

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

建立在怎样的假设之上

剪枝策略十分粗糙,默认选择每层feature map前面的channel,更难的sample才会用到后面的channel


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

实现细节

主要是Policy Network部分

state

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

将上一层的Fi(feature map) [N, n1, H, W]通过global average pooling得到一个[N, n1],通过encoderi得到固定长度的特征作为state[N, C]输入到RNN中,RNN输出hidden state[N, C],通过decoderi得到这个decision的Q-value
值得注意的是,对每一层都有独立的encoderi跟decoderi。槽点:用大量(冗余)的policy network参数去对backbone剪枝(减少参数)

action

ai表示选择保留前i组的channel

reward

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

architecture

算法

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

十分粗略的解读,主要是方便自己以后查看。
水平有限,希望没有误人子弟,有任何问题欢迎留言或者私信讨论(不定时查看……)

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 156,757评论 4 359
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,478评论 1 289
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 106,540评论 0 237
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,593评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 51,903评论 3 285
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,329评论 1 210
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,659评论 2 309
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,383评论 0 195
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,055评论 1 238
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,337评论 2 241
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,864评论 1 256
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,227评论 2 251
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,820评论 3 231
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 25,999评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,750评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,365评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,260评论 2 258

推荐阅读更多精彩内容