该论文关注大模型私有推理中的性能瓶颈问题。生成式 LLM 在代码生成、问答等任务上表现优异,但其自回归解码(autoregressive decoding)逐 token 顺序生成,部署时会将用户输入与提示暴露给服务方,引发严重隐私顾虑,因此出现了基于全同态加密(FHE)的私有推理框架。现有 FHE 框架的核心缺陷在于非线性算子(如 Softmax、激活函数等)的同态评估效率极低,其开销在解码阶段占据主导,导致端到端延迟不可接受。作者观察到解码阶段的非线性运算负载模式并不一致,存在异质性,因而适合用混合方案分别处理。基于此提出 ROSETTA 混合 CKKS/TFHE 框架,包含两项关键贡献:一是基于 TFHE 的自适应分段查找表(segmented lookup-table)协议,用于高效且精确地评估非线性运算,缓解 TFHE 在精度与电路规模上的取舍;二是方案感知的算子选择框架,能够自动判定每个非线性算子交由 CKKS 还是 TFHE 执行,以最小化端到端解码延迟,实现两种同态方案在精度、吞吐与延迟上的互补。实验中,ROSETTA 相较当前最优框架 CacheMir 在 Softmax 上取得最高 4.8 倍加速,端到端解码取得 1.5 至 2.1 倍加速。该工作属于隐私保护机器学习与同态加密系统工程交叉方向,适合从事机密计算、隐私推理加速、密码工程以及需要在合规敏感场景中部署 LLM 的研究者与工程师阅读。
💡 推荐理由: 同态加密私有推理是 LLM 落地于医疗、金融等高敏场景的关键技术路径,但非线性算子开销长期制约可用性。该工作通过 CKKS/TFHE 混合调度显著降低解码延迟,为可实用的密态推理系统提供了工程化思路。
🎯 建议动作: 研究跟进