arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.17267cs.LGcs.AIcs.CV
收藏

Bit-Mixer:具有运行时位宽选择的混合精度网络

Bit-Mixer: Mixed-precision networks with runtime bit-width selection

Samsung AI Cambridge(三星AI剑桥);Queen Mary University of London(伦敦玛丽女王大学)

Adrian Bulat, Georgios Tzimiropoulos

摘要:

混合精度网络允许对网络中的每一层进行可变位宽量化。现有工作的一个主要限制是每个层的位宽必须在训练时间内预定义。如果在其上部署网络的设备的特性在运行时期间改变,则这几乎不允许灵活性。在这项工作中,我们提出了比特混合器,第一个方法来训练一个元量化的网络,在测试期间,任何层都可以改变其投标宽度,而不会影响整个网络的能力,高度准确的推理。为此,我们做出了两个关键贡献:(a)过渡批规范,以及(b)一个3阶段优化过程,该过程被证明能够训练这样的网络。我们表明,我们的方法可以产生混合精度网络,表现出理想的灵活性,在设备上部署,而不影响准确性。代码将提供。

英文摘要:

Mixed-precision networks allow for a variable bit-width quantization for every layer in the network. A major limitation of existing work is that the bit-width for each layer must be predefined during training time. This allows little flexibility if the characteristics of the device on which the network is deployed change during runtime. In this work, we propose Bit-Mixer, the very first method to train a meta-quantized network where during test time any layer can change its bid-width without affecting at all the overall network's ability for highly accurate inference. To this end, we make 2 key contributions: (a) Transitional Batch-Norms, and (b) a 3-stage optimization process which is shown capable of training such a network. We show that our method can result in mixed precision networks that exhibit the desirable flexibility properties for on-device deployment without compromising accuracy. Code will be made available.

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

↑