提出双模型方法,在不牺牲效率的前提下提升MoE的抗攻击能力。
Optimizing Robustness and Accuracy in Mixture of Experts: A Dual-Model Approach
- 针对专家网络易受攻击的问题,仅加固一个专家并设计新损失函数。
- 双模型线性融合可灵活调节鲁棒性与准确率的平衡,显著提升性能。
- 理论证明了模型鲁棒性边界,适合注重安全性的实际部署场景。
Mixture of Experts(MoE)在复杂任务中表现出色,但对对抗攻击敏感,制约其在高可靠性场景的应用。本文分析发现,专家网络比路由器更易受攻击。基于此,提出一种针对性鲁棒训练方法,仅需加固一个专家,通过新损失函数提升对抗鲁棒性,且不影响训练与推理效率。进一步引入双模型策略,将标准MoE与鲁棒化MoE线性融合,通过平滑参数灵活控制鲁棒性-准确率权衡。理论层面,推导出单模型与双模型的认证鲁棒性边界。为突破性能上限,提出联合训练策略JTDMoE,使双模型在鲁棒性和准确率上均优于独立训练。在CIFAR-10和TinyImageNet数据集上,使用ResNet18和ViT架构的实验验证了方法有效性。代码已开源:https://github.com/TIML-Group/Robust-MoE-Dual-Model。
原文摘要 · Abstract (English)
Mixture of Experts (MoE) have shown remarkable success in leveraging specialized expert networks for complex machine learning tasks. However, their susceptibility to adversarial attacks presents a critical challenge for deployment in robust applications. This paper addresses the critical question of how to incorporate robustness into MoEs while maintaining high natural accuracy. We begin by analyzing the vulnerability of MoE components, finding that expert networks are notably more susceptible to adversarial attacks than the router. Based on this insight, we propose a targeted robust training technique that integrates a novel loss function to enhance the adversarial robustness of MoE, requiring only the robustification of one additional expert without compromising training or inference efficiency. Building on this, we introduce a dual-model strategy that linearly combines a standard MoE model with our robustified MoE model using a smoothing parameter. This approach allows for flexible control over the robustness-accuracy trade-off. We further provide theoretical foundations by deriving certified robustness bounds for both the single MoE and the dual-model. To push the boundaries of robustness and accuracy, we propose a novel joint training strategy JTDMoE for the dual-model. This joint training enhances both robustness and accuracy beyond what is achievable with separate models. Experimental results on CIFAR-10 and TinyImageNet datasets using ResNet18 and Vision Transformer (ViT) architectures demonstrate the effectiveness of our proposed methods. The code is publicly available at https://github.com/TIML-Group/Robust-MoE-Dual-Model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。