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

一,剪枝分类

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

1.1,非结构化剪枝

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

1.2,结构化剪枝

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

1.3,本地与全局修剪

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

相关内容

热门资讯

卡塔尔代表团抵达伊朗,讨论双边... 据伊朗方面消息,当地时间6月10日,一个卡塔尔代表团抵达伊朗德黑兰,讨论双边关系和地区局势。据了解,...
【哈尔滨新闻】聚焦“东北超” ... (来源:哈尔滨新闻眼)      今天,征战2026 东北地区城市足球联赛的呼和浩特队一行顺利抵达哈...
2026年浙江高考顺利落幕 成... 中新网杭州6月10日电(曹丹)6月10日下午,随着地理选考结束铃声响起,2026年浙江高考顺利收官。...
广东高院发布12项“粤法护企”... 中新网广州6月10日电 (方伟彬 马卓尔 徐华磊)广东省高级人民法院10日发布司法服务保障民营经济高...
深圳液冷展直击 | 乔锋智能:... (来源:乔锋智能)展会直击乔锋智能硬核亮相液冷赋算力 智铣启未来  聚焦液冷赛道,聚力智造新程!6月...