arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.06513cs.LGcs.AIcs.CVcs.NEstat.ML

A Privacy-Preserving-Oriented DNN Pruning and Mobile Acceleration Framework

Northeastern University(东北大学);The College of William and Mary(威廉与玛丽学院);University of Connecticut(康涅狄格大学);University of Illinois at Chicago(伊利诺伊大学芝加哥分校)

Yifan Gong, Zheng Zhan, Zhengang Li, Wei Niu, Xiaolong Ma, Wenhao Wang, Bin Ren, Caiwen Ding, Xue Lin, Xiaolin Xu, Yanzhi Wang

英文摘要:

Weight pruning of deep neural networks (DNNs) has been proposed to satisfy the limited storage and computing capability of mobile edge devices. However, previous pruning methods mainly focus on reducing the model size and/or improving performance without considering the privacy of user data. To mitigate this concern, we propose a privacy-preserving-oriented pruning and mobile acceleration framework that does not require the private training dataset. At the algorithm level of the proposed framework, a systematic weight pruning technique based on the alternating direction method of multipliers (ADMM) is designed to iteratively solve the pattern-based pruning problem for each layer with randomly generated synthetic data. In addition, corresponding optimizations at the compiler level are leveraged for inference accelerations on devices. With the proposed framework, users could avoid the time-consuming pruning process for non-experts and directly benefit from compressed models. Experimental results show that the proposed framework outperforms three state-of-art end-to-end DNN frameworks, i.e., TensorFlow-Lite, TVM, and MNN, with speedup up to 4.2X, 2.5X, and 2.0X, respectively, with almost no accuracy loss, while preserving data privacy.

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

↑