【CN110059740A】一种针对嵌入式移动端的深度学习语义分割模型压缩方法【专利】

  • 格式:pdf
  • 大小:473.85 KB
  • 文档页数:9

(19)中华人民共和国国家知识产权局
(12)发明专利申请
(10)申请公布号 (43)申请公布日 (21)申请号 201910294185.4
(22)申请日 2019.04.12
(71)申请人 杭州电子科技大学
地址 310018 浙江省杭州市下沙高教园区2
号大街
(72)发明人 戴国骏 严嘉浩 张桦 吴以凡 
史建凯 
(74)专利代理机构 杭州君度专利代理事务所
(特殊普通合伙) 33240
代理人 朱月芬
(51)Int.Cl.
G06K 9/62(2006.01)
G06N 3/04(2006.01)
(54)发明名称
一种针对嵌入式移动端的深度学习语义分
割模型压缩方法
(57)摘要
本发明公开了一种针对嵌入式移动端的深
度学习语义分割模型压缩方法。

本发明中固定训
练得到的教师网络参数权值,对鉴别网络与学生
网络进行不断训练与学习,在三个不同层次上进
行蒸馏(成对蒸馏、像素蒸馏、整体蒸馏),以使总
体优化目标(交叉熵损失、像素蒸馏损失、成对蒸
馏损失、整体蒸馏损失)不断优化,最终使经过蒸
馏得到的学生网络在满足IoU(Intersection
over Union)下降极少的情况下参数数量大大减
少以及网络前向推算时间大量减少。

本发明解决
了嵌入式移动端因为自身GPU能力以及供电受限
的情况下无法搭载大型深度学习网络的问题,并
大量减少任务计算时间,使嵌入式移动端平台搭
载复杂的深度网络模型成为可能。

权利要求书2页 说明书4页 附图2页CN 110059740 A 2019.07.26
C N 110059740
A
1.一种针对嵌入式移动端的深度学习语义分割模型压缩方法,其特征在于包括教师网络、学生网络、鉴别网络、三种不同级别的蒸馏结构、网络内部的三种损失函数以及深度学习训练平台和嵌入式移动端平台;其中蒸馏结构包括成对蒸馏结构、像素蒸馏结构、整体蒸馏结构;损失函数包括区别损失函数、分割损失函数、蒸馏损失函数;
所述教师网络原型基于深度学习ESPNet分割网络模型,接收输入的图像,输出在特征图、像素级别上的分类取值,送入损失函数进行下一步处理;
所述学生网络为需要得到的小网络,实现过程中需要其接受输入图像,输出在特征图、像素级别的分类取值,送入损失函数进行下一步处理;
所述鉴别网络相当于生成式对抗网络中的判别器,其接受教师网络中对输入原图的像素分割作为真实图,接受学生网络中对输入原图的像素分割作为分割图,使用Wasserstein 距离计算两种分割结果表示教师网络分割结果与当前学生网络的分割结果是否匹配;
所述成对蒸馏结构,为使用已经训练好且分割结果良好的教师网络,计算教师网络特征图中两两像素之间的相关性,以及学生网络特征图中两两像素之间的相关性,并使用这两者的相关性计算平方差计算成对蒸馏损失l pa (S),使学生网络特征图尽可能逼近教师网络特征图;
所述像素蒸馏结构,为教师网络的单像素输出概率与学生网络的单像素输出概率作为输入,使用Kullback -Leibler距离计算对应像素点上的教师网络与学生网络的每类概率距离作为输出,即为像素蒸馏损失l pi (S);
所述整体蒸馏结构,为接受教师网络与学生网络对同一张原始图像的分割图,使用鉴别网络作为判别器计算两张分割图的匹配得分作为整体蒸馏损失l ho (S,D);
所述区别损失函数,即为Wasserstein距离;
所述分割损失函数,即为交叉熵损失;
所述蒸馏损失函数,即为成对蒸馏损失、像素蒸馏损失以及Holistic损失;
所述深度学习训练平台为一台具有较强计算能力的服务器,其使用X99E -WS主板,搭载1颗intel i7-7700k CPU和两张3840个CUDA核心,12GB GDDR5显存的Pascal架构显卡Titan Xp;其拥有的足够的显存能够加载并支持较大Caffe模型的参数估计,拥有足够多的CUDA和极高的1582Mhz加速频率能加速得到模型参数;
所述嵌入式移动端平台为具有一定计算能力的Nvidia Jetson TX1型号计算机,其GPU 采用maxwell架构,拥有256个CUDA核心;CPU采用Quad ARM A57/2 MB L2;支持4k*2k 30Hz HEVC编码或4k*2k 60Hz解码;4GB 64bit LPDDR4内存;16G eMMC闪存,能搭载目前常用的深度学习框架,具有一定的深度学习前向推测能力。

2.根据权利要求1所述的一种针对嵌入式移动端的深度学习语义分割模型压缩方法,其特征在于:
针对移动端平台的深度学习语义分割模型压缩方法中的优化目标有像素蒸馏损失:
其中表示从学生S网络结构中产生的第i个像素的类概率,表示从教师T网络结构中产生的第i个像素的类概率,KL()函数为计算以上两者的Kullback -Leibler距离,R=
权 利 要 求 书1/2页2CN 110059740 A。