4 个月前

深度炸制卷积神经网络

深度炸制卷积神经网络

摘要

深度卷积神经网络的全连接层通常包含超过90%的网络参数,并且消耗了存储网络参数所需内存的大部分。在内存受限的环境中(如GPU或嵌入式设备)运行深度神经网络时,减少参数数量同时基本保持相同的预测性能至关重要。本文展示了如何利用核方法,特别是单个Fastfood层,来替代深度卷积神经网络中的所有全连接层。这种新颖的Fastfood层可以与卷积层一起进行端到端训练,使我们能够将其结合成一种新的架构,称为“深炸卷积网络”(Deep Fried Convolutional Networks),该架构在MNIST和ImageNet数据集上训练时,显著减少了卷积网络的内存占用,而不会降低预测性能。

代码仓库

v0lta/wavelet-network-compression
pytorch
GitHub 中提及

基准测试

基准方法指标
image-classification-on-mnistDeep Fried Convnets
Percentage error: 0.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
深度炸制卷积神经网络 | 论文 | HyperAI超神经