Pruning 系列 (六)remove剪枝

本文对比了填0剪枝和Remove剪枝的差异,重点介绍了Remove剪枝的目的、操作以及在模型优化中的效果。通过示例展示了Remove剪枝如何减少模型参数,但可能影响网络表示能力。修剪后的模型参数减少了2,448个,对模型尺寸影响较小。" 85004282,7242371,多模态数据嵌入:知识库补全新方法,"['知识图谱', '自然语言处理', '深度学习', '多模态', '知识库']

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境

  • python 3.9
  • numpy 1.24.1
  • pytorch 2.0.0+cu117

一、填0与remove剪枝的差异

直接填0的剪枝:

优点:

  • 保留了原始网络结构,便于实现和微调。
  • 部分减少模型的计算量。

缺点:

  • 零权重仍然需要存储,因此不会减少内存使用。
  • 一些硬件和软件无法利用稀疏计算,从而无法提高计算效率。

直接remove的剪枝:

优点:

  • 可以减少模型的计算量和内存使用。
  • 可以通过减少网络容量来防止过拟合。

缺点:

  • 可能会降低网络的表示能力,导致性能下降。
  • 需要对网络结构进行改变,这可能会增加实现和微调的复杂性。

二、remove剪枝目的

ONNX 线上预览:地址

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

_yuki_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值