DARC:可微分架构压缩
机器学习
2019-05-21 v1 计算机视觉与模式识别
机器学习
摘要
在许多学习场景中,推理时的资源比训练时的资源受限得多。本文研究了一种称为可微分架构压缩(DARC)的通用范式,它结合模型压缩与架构搜索来学习推理时资源高效的模型。给定一个资源密集型基础架构,DARC 利用训练数据学习哪些子组件可被更廉价的替代方案替换。该高层技术可应用于任意神经架构,我们报告了在用于图像分类的最先进卷积神经网络上的实验。对于在 CIFAR-10 上准确率为 的 WideResNet,我们将单样本推理速度提升 、内存占用降低 ,且无精度损失。对于在 ImageNet 上 Top1 准确率为 的 ResNet,我们将批量推理速度提升 、内存占用降低 ,精度损失 。我们还在简化情形下给出了理论 Rademacher 复杂度界,说明 DARC 如何避免过拟合尽管存在过参数化。
引用
@article{arxiv.1905.08170,
title = {DARC: Differentiable ARchitecture Compression},
author = {Shashank Singh and Ashish Khetan and Zohar Karnin},
journal= {arXiv preprint arXiv:1905.08170},
year = {2019}
}