TensorFlow官方发布剪枝优化工具：参数减少80%，精度几乎不变

晓查 编译自 Medium量子位 报道 | 公众号 QbitAI去年TensorFlow官方推出了模型优化工具，最多能将模型尺寸减小4倍，运行速度提高3倍。最近现又有一款新工具加入模型优化“豪华套餐”，这就是基于Keras的剪枝优化工具。训练AI模型有时需要大量硬件资源，但不是每个人都有4个GPU的豪华配置，剪枝优化可以帮你缩小模型尺寸，以较小的代价进行推理。什么是权重剪枝？权重剪枝（Weight Pruning）优化，就是消除权重张量中不必要的值，减少神经网络层之间的连接数量，减少计算中涉及的参数，从而降低操作次数。这样做的好处是压缩了网络的存储空间，尤其是稀疏张量特别适合压缩。例如，经过处理可以将MNIST的90％稀疏度模型从12MB压缩到2MB。此外，权重剪枝与量化（quantizati ………………………………