提出可证明鲁棒的联邦强化学习框架,防御新型角度攻击
Provably Robust Federated Reinforcement Learning
- 用多策略聚合与多数投票/几何中位数决策增强抗攻击能力
- 实验显示新攻击可严重破坏传统方法,而本方案显著提升鲁棒性
- 适合关注联邦学习安全、尤其对抗恶意节点的系统设计者
联邦强化学习(FRL)使多个智能体在中心服务器指导下联合学习全局决策策略。尽管具有优势,其去中心化架构易受投毒攻击。现有针对FRL设计的拜占庭鲁棒聚合方法在我们工作中被发现对新型‘归一化攻击’不免疫。该攻击不同于以往放大策略更新差异距离的方式,而是聚焦于最大化更新前后方向的夹角偏差。为应对这一威胁,我们提出一种可证明安全的集成式FRL方法:训练多个由不同智能体组基于基础聚合规则学习的全局策略,测试时对特定状态分别预测动作,离散动作系统采用多数投票,连续动作系统采用几何中位数决定最终动作。实验表明,归一化攻击能显著破坏非集成型拜占庭鲁棒方法,而本方案展现出强抗攻击性能。
原文摘要 · Abstract (English)
Federated reinforcement learning (FRL) allows agents to jointly learn a global decision-making policy under the guidance of a central server. While FRL has advantages, its decentralized design makes it prone to poisoning attacks. To mitigate this, Byzantine-robust aggregation techniques tailored for FRL have been introduced. Yet, in our work, we reveal that these current Byzantine-robust techniques are not immune to our newly introduced Normalized attack. Distinct from previous attacks that targeted enlarging the distance of policy updates before and after an attack, our Normalized attack emphasizes on maximizing the angle of deviation between these updates. To counter these threats, we develop an ensemble FRL approach that is provably secure against both known and our newly proposed attacks. Our ensemble method involves training multiple global policies, where each is learnt by a group of agents using any foundational aggregation rule. These well-trained global policies then individually predict the action for a specific test state. The ultimate action is chosen based on a majority vote for discrete action systems or the geometric median for continuous ones. Our experimental results across different settings show that the Normalized attack can greatly disrupt non-ensemble Byzantine-robust methods, and our ensemble approach offers substantial resistance against poisoning attacks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。