计算机科学 > 机器学习
[提交于 2025年8月3日
]
标题: 隐私保护的量化BERT模型推理
标题: Privacy-Preserving Inference for Quantized BERT Models
摘要: 随着生成式机器学习模型在医疗保健和个人化服务等隐私敏感领域中的日益部署,确保安全推理已成为一个关键挑战。安全多方计算(MPC)能够实现隐私保护的模型推理,但其存在较高的通信和计算开销。主要瓶颈在于浮点运算的昂贵安全评估。量化通过将浮点运算转换为低精度整数计算,提供了一个有前景的解决方案,显著减少了开销。然而,现有的基于MPC的量化推理方法要么依赖于公开的量化参数——带来隐私风险,要么在处理非线性函数如激活函数和softmax时效率低下。在本工作中,我们提出了一种细粒度、逐层的量化方案,并在安全环境中支持1比特权重的全连接层。我们设计了一个多输入查找表协议,以高效且安全地评估softmax。此外,我们使用双秘密共享方案,并通过查找表进行精度转换,完全消除了截断开销。在BERT-base模型上的实验评估表明,我们的方法相比Lu\emph{等}的速度提升了$8\times$,相比Gupta\emph{等}的速度提升了$9\times$,相比Knott\emph{等}的速度提升了$22 \times$。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.