arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.15564cs.LG
收藏

基于原型的个性化剪枝

Prototype-based Personalized Pruning

Qualcomm AI Research(高通AI研究院);Qualcomm Technologies, Inc.(高通技术公司)

Jangho Kim, Simyung Chang, Sungrack Yun, Nojun Kwak

摘要:

如今,随着智能手机等边缘设备的普及,对个性化服务的需求越来越大。然而,传统的个性化方法不适合边缘设备,因为需要对有限的个人数据进行重新训练或微调。此外,对于资源有限的边缘设备来说,完整模型可能过于沉重。不幸的是,可以处理模型复杂性问题的模型压缩方法也需要再训练阶段。这些多个训练阶段通常在设备上学习期间需要巨大的计算成本,这可能是边缘设备的负担。在这项工作中,我们提出了一个动态的个性化方法,称为基于原型的个性化修剪(PPP)。PPP考虑了个性化和模型效率的两端。在训练网络之后,PPP可以很容易地用代表个人数据特征的原型来修剪网络,并且它在没有重新训练或微调的情况下表现良好。我们在计算机视觉和关键词定位的几个任务上验证了PPP的有用性。

英文摘要:

Nowadays, as edge devices such as smartphones become prevalent, there are increasing demands for personalized services. However, traditional personalization methods are not suitable for edge devices because retraining or finetuning is needed with limited personal data. Also, a full model might be too heavy for edge devices with limited resources. Unfortunately, model compression methods which can handle the model complexity issue also require the retraining phase. These multiple training phases generally need huge computational cost during on-device learning which can be a burden to edge devices. In this work, we propose a dynamic personalization method called prototype-based personalized pruning (PPP). PPP considers both ends of personalization and model efficiency. After training a network, PPP can easily prune the network with a prototype representing the characteristics of personal data and it performs well without retraining or finetuning. We verify the usefulness of PPP on a couple of tasks in computer vision and Keyword spotting.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑