北大、蚂蚁三个维度解构高效隐私保护机器学习:前沿进展+发展方向

本项目的领导者为李萌,于 2022 年加入北京大学人工智能研究院和集成电路学院创建高效安全计算实验室。他曾任职于美国 Facebook 公司的 Reality Lab,作为技术主管主导虚拟现实和增强现实设备中的高效 AI 算法和『芯片』研究。他的研究兴趣集中于高效、安全人工智能加速算法和『芯片』,旨在通过算法到『芯片』的跨层次协同设计和优化,为人工智能构建高能效、高可靠、高安全的算力基础,曾获 DAC 生成式人工智能系统设计竞赛第一名、AICAS 大模型系统设计竞赛第一名、CCF 集成电路 Early Career Award、欧洲设计自动化协会最佳博士论文等一系列奖项。

在数据隐私日益重要的 AI 时代,如何在保护用户数据的同时高效运行机器学习模型,成为了学术界和工业界共同关注的难题。

北大团队最新完成的综述《Towards Efficient Privacy-Preserving Machine Learning: A Systematic Review from Protocol, Model, and System Perspectives》系统性地梳理了当前隐私保护机器学习(PPML)领域的三大优化维度,首次提出跨协议、模型和系统三个层级的统一视角,为学术界和工业界提供了更加清晰的知识脉络与方向指引。

本文由北京大学助理教授李萌课题组和蚂蚁集团机构的多位研究者共同完成。论文题目及完整作者列表如下:

  • 论文标题:Towards Efficient Privacy-Preserving Machine Learning: A Systematic Review from Protocol, Model, and System Perspectives
  • 论文地址:https://arxiv.org/pdf/2507.14519

团队还建立了一个长期维护的 GitHub 项目,持续收录高质量 PPML 文献,欢迎大家 star,并提出宝贵的意见和补充:

  • 文献库:https://github.com/PKU-SEC-Lab/Awesome-PPML-Papers

文章的总体结构如下:

层级一:协议层级优化

尽管密码学协议为数据隐私保护提供了严格的安全保证,但其应用于人工智能计算,仍面临巨大开销。本综述指出当前协议设计主要存在以下核心痛点:1)基于不经意传输(OT)的协议有极高的通信开销和基于同态加密(HE)的协议面临严重计算瓶颈;2)现有协议忽视模型固有的结构特性(如稀疏性、量化鲁棒性),因此缺乏 “模型感知” 的协议设计。

本综述分别从人工智能模型的线性算子和非线性算子切入,主要讨论了基于 OT 和 HE 的协议设计和发展脉络。综述中重点回答了在不同场景中,应该使用何种协议以及 HE 编码方式。综述还分析了在交互式和非交互式协议框架下的图级协议,比如秘密分享和 HE 之间的转换、全同态中的自举方案。以下是关于编码方案的总结:

层级二:模型层级优化

本综述强调在传统明文机器学习模型中的设计(如 ReLU 剪枝、模型量化)在 PPML 中往往会导致高昂代价。综述系统地归纳了当前 PPML 领域的四类模型层优化策略:1)线性层优化:比如高效卷积设计、低秩分解、线性层融合;2)非线性层 ReLU 和 GeLU 优化:比如多项式近似、剪枝和 GeLU 的替换;3)非线性层 Softmax 优化:比如昂贵算子的替换、KV cache 剪枝、注意力头融合;4)低精度量化,包括 OT 和 HE 友好的量化算法。下表概括了线性层和非线性层的优化方案:

层级三:系统层级优化

本综述指出,即便协议和模型层级已经得到优化,系统层级若无法 “感知协议特性”,将难以释放真正性能。综述中梳理了两个方向的优化路径:1)编译器设计:从协议特性感知、灵活编码、Bootstrapping 支持等方面展开了讨论;2)GPU 设计:分别讨论了操作层面加速与 PPML 系统层面的优化,通过对比现有 GPU 加速实现中典型 PPML 工作负载的执行时间,对相关技术进行了总结。下图是 HE 编译器的梳理:

下表对比了 GPU 加速的 HE 框架:

总结与讨论

本综述强调,仅仅在某一层级优化已难以满足大模型时代对隐私与效率的双重要求。综述提出必须从 “跨层级协同优化” 的角度重新设计 PPML 的方案,未来的研究方向包括:1)协议 - 模型 - 系统协同优化和设计;2)构建面向大模型隐私推理的隐私计算方案;3)面向边缘设备部署的轻量化隐私计算方案。

值得一提的是,李萌老师课题组近年来围绕上述三个层面,也开展了一系列相关研究工作,欢迎各位相关领域老师、同学多多交流。下图总结了课题组已经发表的相关工作:

本综述详细讨论了跨层级优化带来的挑战与机遇,分别阐述了模型和协议的系统优化、协议和系统的系统优化。例如模型量化难以直接给 PPML 带来期望的收益,非线性层优化难以带来系统级的效率提升,现代 GPU 加速了明文机器学习,但其有限的精度支持给 HE 所需的高精度模块化算术带来了挑战。

综述还进一步从线性层和非线性层角度讨论了大模型对 PPML 的独特挑战,并提出除了无需训练的优化方式,还可以考虑用参数高效微调(比如 LoRA)等技术去构建 PPML 友好的大模型结构。

特别声明:[北大、蚂蚁三个维度解构高效隐私保护机器学习:前沿进展+发展方向] 该文观点仅代表作者本人,今日霍州系信息发布平台,霍州网仅提供信息存储空间服务。

猜你喜欢

iPhone 17 Pro系列机型直降300元,会是提前在给iPhone 18系列铺路吗(iPhone 17 Pro系列有京东方屏幕吗)

另外在核心性能方面,iPhone 17 Pro系列还搭载有全新的A19 Pro『芯片』,这款『芯片』相较之前的iPhone Air搭载的A19 Pro,多出了一个GPU核心,在图片处理方面,性能会更突出。有相关博主称…

iPhone 17 Pro系列机型直降300元,会是提前在给iPhone 18系列铺路吗(iPhone 17 Pro系列有京东方屏幕吗)

事业爆红,却难逃焦虑?31岁『张晚意』,撕开了多少人的“婚育”伪装(事业火爆)

但『张晚意』让我看到了一个事实:每个人头上都有一把无形的尺子,而这把尺子就是所谓的“社会时钟”。我们害怕,当我们终于有能力、也有底气去建立一个家庭时,却发现最好的年华已经溜走,身体和精力都不再允许我们去追求那些曾…

事业爆红,却难逃焦虑?31岁『张晚意』,撕开了多少人的“婚育”伪装(事业火爆)

杨志刚:面瘫还结巴,却被哥哥捧了15年,现在兄弟俩成内娱清流(杨志刚帅)

有些人认为它继承了前两部的精彩,再现了大唐盛世下的奇异故事,但也有不少人对剧中的服装、道具乃至演员表现提出了质疑,尤其是杨志刚饰演的男主角🎭️苏无名,争议声不断。但也有观众认为他的演技不够精致,过于做作,无法让角…

杨志刚:面瘫还结巴,却被哥哥捧了15年,现在兄弟俩成内娱清流(杨志刚帅)

吕良伟广西祭祖引爆全场!劳斯莱斯开道儿子撑伞累到换手,身份争议却被一句话打脸(吕良伟承认过自己是广西人吗)

谁知道他转手就把伞递给了旁边的长辈,嘴里还说着“您年纪大,可别淋着”。他在台上讲话的时候直接敞开了说:“现在祭祖不是争谁的根在哪,是让宗亲们多走动。” 但那天在陆川,看着族人们围着吕良伟有说有笑,他给长辈递…

吕良伟广西祭祖引爆全场!劳斯莱斯开道儿子撑伞累到换手,身份争议却被一句话打脸(吕良伟承认过自己是广西人吗)

藏在消费好物里的 “宝藏材料”!氧化锆陶瓷结构件的 4 大优势太能打(消费购物平台)

随着智能穿戴设备普及,材质的亲肤性成为核心诉求 —— 手表、手环、耳机等产品长期接触皮肤,一旦致敏就会影响使用体验。氧化锆陶瓷在这方面堪称 “满分材质”,不含镍、铬等易致敏金属元素,也没有塑料中的刺激性添加…

藏在消费好物里的 “宝藏材料”!氧化锆陶瓷结构件的 4 大优势太能打(消费购物平台)