学习笔记——神经网络压缩

简介: 学习笔记——神经网络压缩

当我们需要将模型部署在一些移动设备上时,但是现在的移动设备的资源是有限的,而存储的空间也是有限的,计算量也是有限的,所以就不能存太大的网络结构,否则就不能实时进行。


一、Network Pruning——网络剪枝


大概的意思就是将一个比较大的network,他的一些weight或者是一些neuron剪掉,让他变得比较小。


  • 问题:为什么能够剪枝处理?

因为现在一般会相信,网络通常被过度参数化(存在大量冗余权重或神经元)


那些现在,将这些没有用到的权重或者是神经元将其剪掉,这样处理之后就会得到一个比较小的网络结构。


如图所示,现在一般来说有一些简单的步骤:

image.png

1)将训练好的网络去评估weight与neuron的重要性,进行一个排序,然后剔除比较靠后的值。


  • 如何衡量一个weight或者是一个neuron的重要性呢?

如果一个weight的值接近于0,那可以相信这个weight是没有那么重要的,而当这个weight的值正很大或者是负很大,那么有理由相信其是比较重要的。


同样的,对于一个神经元来说,如果一个神经元的输出几乎为0,那么就有理由相信这个神经元是没有什么作用的。


2)剪枝之后,将网络进行微调,如果模型已经够小了,那么就可以结束剪枝操作了,如果还比较大,那么可以重复这个过程直到满足要求为止。


但是,既然大的网络需要剪枝处理,那么为什么一开始就不训练一个小的网络呢?一个可能的感受是,小的网络比较难以去训练,然后大的网络比较容易去优化,去优化。一般来说,训练过程中存在鞍点或者局部最优解的问题。而如果网络够大,那么这种情况就不会太严重。现在有足够多的文献可以证明,只要网络够大够深,就可以用gradient descent直接找到全局最优解。所以训练一个大的网络,再剪枝处理是比较好的


一个比较好玩的解释是,在一个大的网络结构中,其实可以看成是有很多个小的网络组成的,每一个小的网络就有可能是一种初始化的参数,而这些小的网络有些可以train起来,而有些会train不起来。所以大的网络结构中容易训练的原因可能是,其中这么多个小网络,只有有一个可以train起来了,那么大的网络就可以train起来了。paper链接:https://arxiv.org/abs/1803.03635

image.png


  • 问题:prun weight与prun neuron哪个会比较好?

在实际的操作过程中,如果是prun weight其实是不太好实现的,因为网络架构变得不规则。有些神经元有2个输入,而有些有4个。

image.png

就算是有办法实现,由于是不规则矩阵,所以就没有办法使用GPU加速的。使用一般来说,将weight设为0,就是假设其被prun掉了。但是这样做的话其实是没有减小模型的大小的。


所以,prun neuron是比较容易操作的,因为网络架构是规则的。

image.png


二、Knowledge Distillation——知识蒸馏


如图所以,知识蒸馏的过程是相当于用一个大的模型去带一个小的模型。现在训练好一个比较大的模型,然后将这个大模型的输出让小模型去训练,这时候可能会得到一些比较好的结果。有时还会出现一些神奇的地方就是,小模型从大模型中知道了数字”7“长的比较像数字”1“与“9”,而当其没有看见过数字“7”的时候还可能正确的识别到。(为了获得更好的软标签)

image.png

另外一个比较有用的地方是,其可以将集成的模型全部变成一个模型,也就是将一堆模型的得到的结果去平均起来,然后让这个学生模型去学习这个Emseble Network的output,现在希望这个student Network的结果与这个ensemble的结果一样。所以现在只需要一个model就可以得到类似ensemble的效果。

image.png

在实际处理中,为了避免出现one-hot的label值,所以会希望这个不同的label间的分数拉近一点,这样就可以使得其与真正的data学习到的内容不一样。

image.png

不过好像实际效果没有很好。


三、Parameter Quantization——参数量化


一些方法:


1)使用较少的位来表示一个值

2)权重聚类

image.png

3)用较少的比特表示频繁的簇,用较多的比特表示稀有的簇 eg:Huffman encoding(略)


四、Architecture Design——架构设计


调整network的架构设计让其变得只需要比较少的参数


1. Full connect

插入一层线性的全连接层使得参数量减少,但这样的话会一定程度上限制了network训练出来的参数,也会限制network原来可以做到的事情。

image.png


2. Depthwise Separable Convolution

具体细节在另外的一篇blog有详细介绍:https://blog.csdn.net/weixin_44751294/article/details/116883445


DW卷积 MobileNet中提出的一种卷积结构,其实是一种特殊的组卷积,可以减少卷积的计算量。每一个fulter只需要处理一个channels就可以了。

image.png

PW 卷积 也是MobileNet中提出,需要和DW卷积一起使用,其实也就是1x1的卷积,这个是GoogleNet中提出

image.png

DW卷积 + PW卷积 = DWP深度可分离卷积,使用与轻量级网络中,可以减少Flops的计算量。


对于这样的结构被广泛在各种的轻量级网络中:



五、Dynamic Computation——动态计算


1.Train multiple classifiers


但是一般来说移动设备上的内存不足以存储多个网络模型


2.Classifiers at the intermedia layer


动态的调整中间层的分类结果,如果资源还充足就训练比较好的网络;而如果资源不充足,就使用比较前面的一些网络的输出结果,不过这也会破坏CNN的结构,因为前面的一些中间层提前的至少一些低维特征,不容易分别。如图所示。

image.png

目录
相关文章
|
6月前
|
网络协议 Linux C++
Linux C/C++ 开发(学习笔记十二 ):TCP服务器(并发网络编程io多路复用epoll)
Linux C/C++ 开发(学习笔记十二 ):TCP服务器(并发网络编程io多路复用epoll)
112 0
|
6月前
|
网络协议 Linux C++
Linux C/C++ 开发(学习笔记十一 ):TCP服务器(并发网络网络编程 一请求一线程)
Linux C/C++ 开发(学习笔记十一 ):TCP服务器(并发网络网络编程 一请求一线程)
126 0
|
27天前
|
Ubuntu 网络安全 图形学
Ubuntu学习笔记(二):ubuntu20.04解决右上角网络图标激活失败或者消失,无法连接有线问题。
在Ubuntu 20.04系统中解决网络图标消失和无法连接有线网络问题的方法,其中第三种方法通过检查并确保Windows防火墙中相关服务开启后成功恢复了网络连接。
303 0
Ubuntu学习笔记(二):ubuntu20.04解决右上角网络图标激活失败或者消失,无法连接有线问题。
|
5月前
|
存储 算法 网络虚拟化
【计算机网络】学习笔记,第三篇:数据链路层
现在的光纤宽带接入 FTTx 都要使用 PPPoE 的方式进行接入。在 PPPoE 弹出的窗口中键入在网络运营商购买的用户名和密码,就可以进行宽带上网了 利用 ADSL 进行宽带上网时,从用户个人电脑到家中的 ADSL 调制解调器之间,也是使用 RJ-45 和 5 类线(即以太网使用的网线)进行连接的,并且也是使用 PPPoE 弹出的窗口进行拨号连接的
79 5
|
26天前
|
机器学习/深度学习 数据可视化 Linux
Seaborn可视化学习笔记(一):可视化神经网络权重分布情况
这篇文章是关于如何使用Seaborn库来可视化神经网络权重分布的教程,包括函数信息、测试代码和实际应用示例。
33 0
|
3月前
|
机器学习/深度学习 自然语言处理 并行计算
【深度学习+面经】Transformer 网络学习笔记
Transformer模型的核心概念、优缺点以及在多个领域的应用,并提供了针对Transformer架构的面试问题及答案。
146 2
|
4月前
|
编解码
Sora信息问题之Sora的视频压缩网络如何解决
Sora信息问题之Sora的视频压缩网络如何解决
56 0
|
6月前
|
监控 Java 数据库连接
【后台开发】TinyWebser学习笔记(1)网络编程基础知识
【后台开发】TinyWebser学习笔记(1)网络编程基础知识
54 3
|
6月前
|
网络协议 Docker 容器
Ubantu docker学习笔记(七)容器网络
Ubantu docker学习笔记(七)容器网络
|
6月前
|
网络协议 网络架构 数据格式
计算机网络学习笔记(一)
OSI七层网络模型旨在支持异构网络互联,从下到上分别为物理层(传输比特流)、数据链路层(帧传输,如交换机)、网络层(IP数据包路由,如路由器)、传输层(TCP/UDP,端到端通信)、会话层(管理会话)、表示层(数据格式转换)和应用层(用户接口,如FTP, SMTP)。每一层负责不同的通信功能,并通过协议如PPP, IP, TCP等协同工作。
36 0