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

移动设备神经语言模型的自适应剪枝

Adaptive Pruning of Neural Language Models for Mobile Devices

University of Waterloo(滑铁卢大学)

Raphael Tang, Jimmy Lin

摘要:

神经语言模型(NLM)存在于一个准确性-效率的权衡空间中,在这个空间中,更好的困惑通常以更大的计算复杂度为代价。在移动设备上的软件键盘应用程序中,这意味着更高的功耗和更短的电池寿命。本文是第一次尝试,据我们所知,在探索NLM的准确性和效率的权衡。基于准递归神经网络(QRNNs),我们应用修剪技术来提供一个“旋钮”来选择不同的操作点。此外,我们提出了一个简单的技术来恢复一些困惑使用一个可以忽略不计的内存量。我们的实证评估考虑了Raspberry Pi上的困惑度和能耗,我们展示了哪些方法提供了最佳的困惑功耗操作点。在一个操作点处,其中一种技术能够提供超过现有技术的40%的能量节省,而复杂度仅相对增加17%。

英文摘要:

Neural language models (NLMs) exist in an accuracy-efficiency tradeoff space where better perplexity typically comes at the cost of greater computation complexity. In a software keyboard application on mobile devices, this translates into higher power consumption and shorter battery life. This paper represents the first attempt, to our knowledge, in exploring accuracy-efficiency tradeoffs for NLMs. Building on quasi-recurrent neural networks (QRNNs), we apply pruning techniques to provide a "knob" to select different operating points. In addition, we propose a simple technique to recover some perplexity using a negligible amount of memory. Our empirical evaluations consider both perplexity as well as energy consumption on a Raspberry Pi, where we demonstrate which methods provide the best perplexity-power consumption operating point. At one operating point, one of the techniques is able to provide energy savings of 40% over the state of the art with only a 17% relative increase in perplexity.

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

↑