arXiv:2607.29221cs.CRcs.AI2026-07被引 2

让弱客户端安全外包大模型计算,用噪声掩码实现高效保密推理。

MOSAIC: Masked Outsourcing of Secure AI Computations

  • 通过矩阵乘法掩码协议,将大模型推理安全外包给不可信服务器。
  • 700亿参数模型上困惑度接近量化方案,媲美全精度推理。
  • 适合需要隐私保护的大规模数据中心部署,如云AI服务。

我们解决可信但计算能力弱的客户端向不可信但算力强大的服务器安全高效外包AI计算的问题,客户端同时持有输入和模型,服务器不得获取任何信息。MOSAIC的核心是一种新型矩阵乘法掩码协议,可扩展至远超以往的大矩阵,支持现代工作负载(如大型Transformer推理)的安全外包。通过在乘法结果中引入少量噪声以放宽正确性要求,MOSAIC实现了最优渐近客户端开销,并使实际运行时间比之前方法快数个数量级。其安全性基于决策LWE和LPN假设。由于噪声在Transformer多层中累积,关键挑战是控制误差增长;MOSAIC采用基于随机哈达玛旋转的误差缩放机制应对。在700亿参数的大型Transformer模型上,MOSAIC的困惑度与主流量化方法相当,甚至在HumanEval上达到全精度BF16推理水平。最后,我们展示了端到端实现,表明如MOSAIC等技术可为现代数据中心提供大规模机密AI的可行路径。非机密推理已通过分阶段(预填充/解码)、分层和时序调度,利用类似RDMA的网络在异构硬件间移动激活值、缓存的键值对和权重以最大化利用率。MOSAIC通过保持可信计算基(TCB)小,将大部分AI计算外包给不可信加速器,实现机密计算的规模化。

原文摘要 · Abstract (English)

We address the challenge of securely and efficiently outsourcing AI computations from a trusted but computationally weak client to an untrusted but powerful server, in the setting where the client holds both the input and the model, and the server must learn neither. We present MOSAIC, whose core is a novel matrix-multiplication masking protocol that scales to far larger matrices than prior work, enabling the safe outsourcing of modern workloads such as large transformer inference. By introducing small amounts of noise to the multiplication result and thereby relaxing correctness, MOSAIC achieves optimal asymptotic client overhead and concrete runtimes orders of magnitude faster than prior work. Its security reduces to the decisional LWE and LPN assumptions. Because this noise accumulates across the many layers of a transformer, a key technical challenge is bounding error growth; MOSAIC addresses this with an error-scaling mechanism based on random Hadamard rotations. On large 70B transformer models, MOSAIC's perplexity is comparable to popular quantization approaches and even matches full-precision BF16 inference on HumanEval. Finally, we present an end-to-end implementation showing how ideas like MOSAIC can promise a path towards large-scale confidential AI in modern data centers. Non-confidential inference is already distributed across phase (prefill/decode), layer, and time to maximize utilization of heterogeneous hardware, using RDMA-like networking to move activations, cached KV values, and weights across nodes. MOSAIC enables scaling of confidential compute by keeping the trusted computing base (TCB) small and outsourcing the bulk of the AI computation to untrusted accelerators.

机密计算大模型推理安全外包隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。