跳到正文
原文
arXiv· arXiv:2609.05770· Duc Dm, Khai Le-Duc, Nguyen Do, Minh Son Hoang, Florent Draye, Thai Hoang, Hoang Phuong Dam, Jiarui Liu, Chris Ngo, Terry Jingchen Zhang, Anh Le Duc Tran, Nhat Do Minh, Minh Ngoc Le, My T. Thai, Ran Xu, Silvio Savarese, Mona Diab, Bernhard Schölkopf, Zhijing Jin, Huy L. Nguyen, Daeyoung Kim·本站收录· 原文发表

RAPTOR:面向混合专家模型的角色感知隐私训练

RAPTOR: Role-Aware Private Training for Mixture-of-Experts

AI 导读

研究者提出 RAPTOR,一个面向稀疏 MoE 模型的角色感知差分隐私微调框架,并证明其满足 (ε,δ)-DP。论文指出,现有方法把稀疏 MoE 当作单个稠密块处理,导致全局裁剪压制专家梯度、批级归一化稀释稀疏专家更新、固定隐私噪声降低低负载专家信噪比三类失败模式。RAPTOR 交替优化共享层与专家层,采用专家专属裁剪与噪声、公开期望所有者分母及与专家计数无关的更新计划;由于每条记录只归属一个专家,一层内更新全部 E 个专家的隐私开销不高于更新单个专家。在 Switch Transformer 与 OLMoE 的 GLUE 任务微调及 DeepSeek-VL2-Tiny 上,该方法在多个隐私预算下稳定优于标准 DP 基线,预算最紧时优势通常最大。代码与模型已在 GitHub 开源。

阅读原文arxiv.org