基于pytorch实现模型剪枝
创始人
2024-05-26 18:09:05
  • 一,剪枝分类
  • 二,PyTorch 的剪枝
  • 三,总结
  • 参考资料

一,剪枝分类

所谓模型剪枝,其实是一种从神经网络中移除"不必要"权重或偏差(weigths/bias)的模型压缩技术。关于什么参数才是“不必要的”,这是一个目前依然在研究的领域。

1.1,非结构化剪枝

非结构化剪枝(Unstructured Puning)是指修剪参数的单个元素,比如全连接层中的单个权重、卷积层中的单个卷积核参数元素或者自定义层中的浮点数(scaling floats)。其重点在于,剪枝权重对象是随机的,没有特定结构,因此被称为非结构化剪枝

1.2,结构化剪枝

与非结构化剪枝相反,结构化剪枝会剪枝整个参数结构。比如,丢弃整行或整列的权重,或者在卷积层中丢弃整个过滤器(Filter)。

1.3,本地与全局修剪

剪枝可以在每层(局部)或多层/所有层(全局)上进

相关内容

热门资讯

“我忘记1月1日就可以拘留了”... 1月3日,湛江3名15岁少年因持械追打他人涉嫌寻衅滋事,被依法执行行政拘留13天。这是2026年1月...
国金证券完成“23国金01”公... 中访网数据  国金证券股份有限公司于2026年1月12日顺利完成其“23国金01”公司债券的本息兑付...
【连续报道·报效祖国 建功西部...   【连续报道·报效祖国 建功西部】光明日报编辑部:  我是一名矿产普查与勘探专业的博士生。一年多来...
循环经济发展“十五五”规划今年... (来源:经济参考报) 记者从1月13日国新办举行的国务院政策例行吹风会获悉,2026年国家发展改革委...
【评论】破解华北农村取暖困局,... 界面新闻记者 | 翟瑞民界面新闻编辑 | 王强随着小寒节气到来,北方地区气温已降至零下,天寒地冻,但...