arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1904.02205cs.LGstat.ML
收藏

深度网络的渐进随机二值化

Progressive Stochastic Binarization of Deep Networks

Johannes Gutenberg-University of Mainz(美因茨约翰内斯·古腾堡大学)

David Hartmann, Michael Wand

摘要:

最近的大量研究都集中在通过降低(i)数值表示(例如,通过确定性或随机量化)和(ii)算术运算(例如,通过权重的二进制化)的复杂性来提高深度网络的内存占用和推理速度。我们提出了一种用于深度网络的随机二进制化方案,该方案通过将自身限制为小整数和固定移位的加法来实现对硬件的有效推理。与以前的方法不同,潜在的随机近似是渐进的,从而允许在运行时对每个操作的准确性进行自适应控制。在低精度设置中,我们匹配先前二值化方法的精度。我们的表示是无偏的-它接近连续计算,随着样本量的增加。在高精度的制度,计算成本是有竞争力的与以前的量化方案。渐进随机二值化还允许在单个网络内进行局部化的动态精度控制,从而为自适应地集中计算注意力提供了一种新的工具。我们在各种架构的网络上评估我们的方法,这些网络已经在ImageNet上进行了预训练。与代表性的成本与以前的计划相比,我们获得的精度接近原来的浮点实现。这包括修剪网络,除了某些类型的分离卷积的已知特殊情况。通过使用渐进式采样集中计算注意力,我们将ImageNet上的推理成本进一步降低了33%(在网络修剪之前)。

英文摘要:

A plethora of recent research has focused on improving the memory footprint and inference speed of deep networks by reducing the complexity of (i) numerical representations (for example, by deterministic or stochastic quantization) and (ii) arithmetic operations (for example, by binarization of weights). We propose a stochastic binarization scheme for deep networks that allows for efficient inference on hardware by restricting itself to additions of small integers and fixed shifts. Unlike previous approaches, the underlying randomized approximation is progressive, thus permitting an adaptive control of the accuracy of each operation at run-time. In a low-precision setting, we match the accuracy of previous binarized approaches. Our representation is unbiased - it approaches continuous computation with increasing sample size. In a high-precision regime, the computational costs are competitive with previous quantization schemes. Progressive stochastic binarization also permits localized, dynamic accuracy control within a single network, thereby providing a new tool for adaptively focusing computational attention. We evaluate our method on networks of various architectures, already pretrained on ImageNet. With representational costs comparable to previous schemes, we obtain accuracies close to the original floating point implementation. This includes pruned networks, except the known special case of certain types of separated convolutions. By focusing computational attention using progressive sampling, we reduce inference costs on ImageNet further by a factor of up to 33% (before network pruning).

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

↑