1. 首页
  2. 人工智能
  3. 论文/代码
  4. 用于模型压缩的结构化多哈希

用于模型压缩的结构化多哈希

上传者: 2021-01-22 16:06:08上传 .PDF文件 508.67 KB 热度 9次

尽管深层神经网络(DNN)取得了成功,但现有的大型模型仍然太大,无法部署在资源少的设备或将多个模型保存在内存中的普通服务器配置中。模型压缩方法通过减少模型的内存占用量,等待时间或能源消耗来解决此限制,同时对准确性的影响最小。..

Structured Multi-Hashing for Model Compression

Despite the success of deep neural networks (DNNs), state-of-the-art models are too large to deploy on low-resource devices or common server configurations in which multiple models are held in memory. Model compression methods address this limitation by reducing the memory footprint, latency, or energy consumption of a model with minimal impact on accuracy.We focus on the task of reducing the number of learnable variables in the model. In this work we combine ideas from weight hashing and dimensionality reductions resulting in a simple and powerful structured multi-hashing method based on matrix products that allows direct control of model size of any deep network and is trained end-to-end. We demonstrate the strength of our approach by compressing models from the ResNet, EfficientNet, and MobileNet architecture families. Our method allows us to drastically decrease the number of variables while maintaining high accuracy. For instance, by applying our approach to EfficentNet-B4 (16M parameters) we reduce it to to the size of B0 (5M parameters), while gaining over 3% in accuracy over B0 baseline. On the commonly used benchmark CIFAR10 we reduce the ResNet32 model by 75% with no loss in quality, and are able to do a 10x compression while still achieving above 90% accuracy.

用户评论