1. 首页
  2. 人工智能
  3. 深度学习
  4. 模型压缩之通道剪枝channel pruning master.zip

模型压缩之通道剪枝channel pruning master.zip

上传者: 2020-07-17 06:38:19上传 ZIP文件 495.49KB 热度 26次
对训练好的模型进行通道剪枝(channel pruning),分为两步进行:第一步是channel selection,采用LASSO regression来做,其实就是添加了一个L1范数来约束权重,因为L1范数可以使得权重更加稀疏,这样就可以把那些稀疏的channel剪掉;第二步是reconstruction,基于linear least squares来约束剪枝后输出的feature map要尽可能和减枝前的输出feature map相等,也就是最小二乘值越小越好。
用户评论