热门

最新

红包

立Flag

投票

同城

我的

发布
ai_conf
AMiner:AI科研助手
5 年前
trueai_conf

IJCAI 2020 论文推荐:
TRP: Trained Rank Pruning for Efficient Deep Neural Networks
为了在移动电话等边缘设备上实现DNNs,低秩逼近(low-rank approximation)由于其理论基础坚固和实现高效而被广泛采用。之前的一些研究工作试图通过低秩分解直接逼近预先训练好的模型;然而,参数中微小的近似误差会引起较大的预测损失。因此性能通常会显著下降,并且需要进行复杂的微调来恢复精度。显然,从训练中分离低秩逼近不是最优的。
与以往的工作不同,本文将低秩逼近和正则化融入到训练过程中。作者提出了Trained Rank Pruning  (TRP),它在低秩逼近和训练之间交替。在训练过程中,TRP在施加低秩约束的同时保持了原始网络的容量。利用随机次梯度下降优化的核正则化进一步提高了TRP的低秩。经过TRP训练的网络本质上具有低秩结构,近似于可忽略的性能损失,从而消除了低秩分解后的微调过程。
该方法在CIFAR-10和ImageNet上进行了综合评价,性能优于以往使用低秩逼近的压缩方法。
论文链接:
https://www.aminer.cn/pub/5eabf34391e011664ffd2878?conf=ijcai2020?f=cs

CSDN App 扫码分享
分享
评论
1
打赏
  • 复制链接
  • 举报
下一条:
刚刚登陆的群众,开始他的吃瓜旅程……
立即登录