CVPR 2019 | Re pr : 卷积滤波器的训练改进

RePr展示一种周期性移除与取回卷积滤波器的训练策略,通过减少冗余的学习特征,改进模型的泛化能力。对常规卷积网络和更复杂的现代深度学习网络架构都很有效,在多种任务上的准确率都有所提升,甚至小模型也有不错的效果。

论文地址:

https://arxiv.org/pdf/1811.07275.pdf

正交特征

卷积特征是多个独立卷积核激活的线性组合,如果一个卷积特征能改善模型的泛化能力,可以认为这个卷积特征是有效的。卷积特征的有效方向越多,对特征空间的表达能力就越强,泛化能力就越好。在过去,我们都尝试使用正则化减少激活特征的相关性,但是效果的改进很有限。

最近的研究成果表明,权重的正交性可以改善模型训练收敛的稳定程度,并有效加速模型的训练速度。本文在卷积网络(ConvNet)引入正交特征,并初步形成特征正交性排名的准则。实验表明,通过正则化代价函数项,对权重正交性改善有限,卷积特征不足以表达整个输入数据的流形空间。提出临时移除重复滤波器策略作为一种类似正则化的方法,在不影响模型收敛的情况下,带来很好的特征正交性。

模型大小与特征相关性

过去我们认为小模型精度不足的原因是权值参数过少,导致卷积特征少无法表达特征空间。因此VGG-16网络在 CIFAR-10数据集上训练形成的相关冗余特征应该远多于ConvNet等小模型。RePr利用典型相关分析(CCA)研究ConvNet卷积层的特征相关性,发现任何滤波器间的相关是训练方法效率不够,而不是过参数化导致的。通过移去特定滤波器,验证此滤波器对模型效果的贡献,在测试集上考察该情况模型的精度来评估。

CCA分析ConvNet的卷积滤波器相关性  

训练策略

RePr的训练流程比较简单,训练整个网络后根据内部滤波器正交性(Inter-Filter Orthogonality)暂时屏蔽30%的滤波器,继续调优网络参数。把屏蔽掉的滤波器取回再次训练剩下的网络,与现有滤波器组合正交方式实现重新初始化,迭代重复该过程多次。

RePr 训练策略

Inter-Filter Orthogonality

Greedy Oracle不一定是寻找最差的滤波器度量最好的标准,而且在大型网络中计算存在困难。因此RePr设计了内部滤波器正交性,作为滤波器相关性排名的度量。公式如下:

上式是一个层的多个卷积核的组合,用矩阵表示为W_l。对W_l归一化后,与其转置相乘得到P_l(大小为{J_l}
/{timesJ_l})。第i行是其他滤波器对第i个滤波器的投影,即相关性,正交性越大值就越小,此行的总和亦越小,因此可以用于滤波器的排名。

实验结果

标准训练方案和RePr的性能对比

上图是标准训练方案和RePr方法在三层ConvNet网络上,CIFAR-10数据集中的性能对比。图中A点进行第一次重复滤波器移除。 C点表示此时模型在测试集的准确率,测试集的准确率下降幅度,比在训练集的下降幅度要小,毕竟训练集具有过拟合现象。D点上测试集准确率已经基本和C点相近,但此时的滤波器只有原来的70%,这与最近剪枝工作成果结论一致。取回30%的滤波器进行正交重初始化并训练到E点时,测试准确率已经远超标准训练方案的效果。

RePr在目标检测的效果

RePr在VQA任务的效果

作者在其他任务上对 RePr进行对比实验,发现效果都有所提高。

结论

RePr训练策略周期性地移除和取回表达力过低的滤波器,移除低质量的滤波器,有利于RePr在剩余滤波器中重新发现新的特征。当重新取回低质量滤波器,并进行正交重初始化和继续训练后,RePr能使模型容量的分配更加稳定有效。

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

推荐阅读更多精彩内容