论文笔记——N2N Learning: Network to Network Compression via Policy Gradient Reinforcement Learning...

论文地址：https://arxiv.org/abs/1709.06030

1. 论文思想

利用强化学习，对网络进行裁剪，从Layer Removal和Layer Shrinkage两个维度进行裁剪。 一个是对层判断是否进行裁剪，一个是判断一层中的参数的裁剪。

2. 原理图

3. 实现细节

将层信息进行编码表示，然后送入双端的LSTM中，最后通过Softmax学出多个行为的概率。然后来决定层的裁剪信息。

4. 结果

ResNet-34上实现了10倍的压缩。

转载于:https://www.cnblogs.com/zhonghuasong/p/7861524.html

总结

以上是生活随笔为你收集整理的论文笔记——N2N Learning: Network to Network Compression via Policy Gradient Reinforcement Learning...的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。