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

AESPA:保精度低次多项式激活的快速私人推理

AESPA: Accuracy Preserving Low-degree Polynomial Activation for Fast Private Inference

Seoul National University(首尔国立大学)

Jaiyoung Park, Michael Jaemin Kim, Wonkyung Jung, Jung Ho Ahn

摘要:

混合私有推理(PI)协议是一种重要的PI技术,它综合利用了多方计算(MPC)和同态加密。然而,即使是最先进的PI协议也会受到非线性层,特别是激活函数的影响。虽然标准的非线性激活函数可以产生更高的模型精度,它必须通过一个昂贵的混淆电路MPC原语处理。多项式激活可以通过Beaver的乘法三倍MPC原语来处理,但到目前为止已经导致严重的准确性下降。在本文中,我们提出了一个保持精度的低次多项式激活函数(AESPA),它利用了ReLU的Hermite扩展和基规范化。我们将AESPA应用于流行的ML模型,如VGGNet,ResNet和预激活ResNet,以显示与具有ReLU激活的标准模型相当的推理准确性,实现了优于先前低次多项式研究的准确性。当应用于最先进的Delphi PI协议上的全USB基线时,AESPA显示出高达42.1倍和28.3倍的在线延迟和通信成本降低。

英文摘要:

Hybrid private inference (PI) protocol, which synergistically utilizes both multi-party computation (MPC) and homomorphic encryption, is one of the most prominent techniques for PI. However, even the state-of-the-art PI protocols are bottlenecked by the non-linear layers, especially the activation functions. Although a standard non-linear activation function can generate higher model accuracy, it must be processed via a costly garbled-circuit MPC primitive. A polynomial activation can be processed via Beaver's multiplication triples MPC primitive but has been incurring severe accuracy drops so far. In this paper, we propose an accuracy preserving low-degree polynomial activation function (AESPA) that exploits the Hermite expansion of the ReLU and basis-wise normalization. We apply AESPA to popular ML models, such as VGGNet, ResNet, and pre-activation ResNet, to show an inference accuracy comparable to those of the standard models with ReLU activation, achieving superior accuracy over prior low-degree polynomial studies. When applied to the all-RELU baseline on the state-of-the-art Delphi PI protocol, AESPA shows up to 42.1x and 28.3x lower online latency and communication cost.

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

↑