计算机视觉中的深度模型压缩

来源 :浙江大学 | 被引量 : 0次 | 上传用户:cyc198810
下载到本地 , 更方便阅读
声明 : 本文档内容版权归属内容提供方 , 如果您对本文有版权争议 , 可与客服联系进行内容授权或下架
论文部分内容阅读
近些年来,卷积神经网络已然成为计算机视觉领域的默认支柱性方法。无论在高级视觉任务(如分类、检测),还是在低级视觉任务(如神经风格迁移、超分辨率),都取得了令人瞩目的成果。然而,卷积神经网络往往参数量巨大,导致所需要的存储量、计算量、能耗均比较大,这严重限制了它们的实际应用。模型压缩是一个剔除深度神经网络冗余度从而降低存储量、加快速度的新兴研究领域。在本论文中,作者提出了两种结构化剪枝算法(SPP和IncReg)用于在分类任务上对现代深度神经网络进行加速。同时,在低级视觉领域中的风格迁移问题上,作者基于知识蒸馏提出了 StyleDistill模型压缩方法对VGG19网络进行剪枝。与本领域其他最优方法相比,本文中充足的实验验证了这三种方法的有效性。本文具体的创新点和贡献总结如下:1.参数剪枝是一类很有潜力的模型压缩方法,其目标是剔除掉网络中无用的参数同时保持精度不大幅下降。在众多的剪枝方法中,结构化剪枝是为了产生规整的稀疏性从而可以实现加速而非存储量上的压缩。对剪枝而言,一个至关重要但目前仍未解决的问题是参数的相对重要性标准,它对于剪枝性能有着很重要的影响。由于神经网络的严重非凸性与过参,提出一个理论上完备、实际中可行的重要性标准是极其困难的。所以,本论文中,作者提出换一个角度提升剪枝性能,即设计一个更好的剪枝过程。基于这样的思想,SPP算法被提出。它对网络中每一个权重分配一个剪枝概率,从而使得原来离散的剪枝变得连续化,这样有利于网络在剪枝过程中自我调整、恢复。同时,通过调整这些剪枝概率,我们还可以纠正由于参数重要性标准(如L1-norm)不够完善而导致的误判。SPP算法配合列剪枝在大型卷积网络(AlexNet,VGG16,ResNet50)在ImageNet数据集上得到了良好的结果。但是其缺点是训练中不够稳定。为了解决这个问题,作者进一步提出IncReg算法。与SPP算法不同,IncReg是基于正则化的剪枝算法。IncReg以一种增量的方式来调节L2约束项前的系数。和SPP一样,IncReg也具有纠正参数重要性误判的情况,且比SPP的剪枝过程更加平滑。这对于剪掉大量参数和剪枝紧凑网络(如ResNet)的场景特别有用。IncReg最终获得了比SPP和其他众多本领域最优算法更好的结果。2.除了高级视觉任务上的模型压缩,本文还关注低级视觉,在这些领域CNN的实际应用也很大程度上被其复杂度所限制。具体来说,本文关注的是风格迁移任务。作者基于知识蒸馏提出了 StyleDistill算法来对VGG19网络的滤波器进行剪枝。配合风格化算法lNWCT,本文压缩完的模型尽管只有原模型的1/15.5大小,但仍具有同样甚至更优的风格迁移效果。更重要的是,基于该小模型,我们首次可以在12GB GPU上实现超高分辨率(超过4千万像素)的任意风格迁移。为了验证所提出的方法的通用性,本文还在照片风格迁移(基于PhotoWCT算法)和基于优化的风格迁移(基于Gatys算法)场景下评估了压缩后的模型,依然取得令人鼓舞的效果。
其他文献
本文以隋唐为主要朝代,以胡乐人为主要研究对象,分别从国家、部落、家族、个体胡乐人入手,以声乐、器乐、理论、作曲、歌舞等体裁为依托,来论述他们在历史中的来源与构成、在
在全媒体时代,传统媒体与新媒体以其自身独特的优势,影响着新闻事件的传播与发展态势。而在传统媒体与新媒体之间产生的博弈,也深刻影响着两者各自的发展,以及整个新闻业未来
包括地外事件在内的地球中阵发性的意义重新引起了18世纪均变论与灾变论之间的论争。看来有些前寒武纪构造热事件可能关系到和代表着大直径(Dp10km)抛射物大型撞击的最终效果。特别是
随着素质教育的全面推进,高校教育教学以及人才培养理念、目标等都有了全新的变化,大学生体育素质发展是一大关键点,体育课程教学的重要性不断呈现。同时,在西方球类运动热潮
随着高中新课改的深入开展,传统的文言文课堂教学问题日益突出,课堂教学方式更是饱受诟病。在这样的背景下,作为一名高中语文教师,要了解文言文教学的误区,并适应新课改,具备
目的:观察针刀疗法对第三腰椎横突综合征模型大鼠组织中成纤维细胞增殖的影响。方法:68只SD大鼠按照随机数字表法分为4组,每组17只。针刀干预组:于造模后15 d,在麻醉下,用针
目的 :探讨基于前列腺磁共振扩散加权图像(Diffusion-weighted imaging,DWI)的诺模图鉴别前列腺癌(Prostate cancer,PCa)与前列腺增生(Benign prostatic hyperplasia,BPH)的
<正>过去,园林作为建筑周边的装饰工程大行其道;今天,我们推崇作为城市公共基础设施的新景观。传统的园林作为城市装饰割裂了城市与自然的交融;现代的景观立足城市公共基础设
对100所国家示范性高职院校教务处长的职业经历、教育背景、科研等方面的调查,发现我国高职院校教务处长具有以下特征:学历和职称偏低;职业经历简单;比较注重自身原专业的科
二战时,有人问一位著名的将军:“什么样的人适合当头?”将军回答:“聪明而懒惰的人”。这个答案同样适用于管理界。管理者的主要工作是什么?“Findtherightway,findtherightp