arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.04432cs.CRcs.LG

FreqFed:一种基于频率分析的联邦学习投毒攻击缓解方法

FreqFed: A Frequency Analysis-Based Approach for Mitigating Poisoning Attacks in Federated Learning

Technical University of Darmstadt(达姆施塔特工业大学);University of Würzburg(维尔茨堡大学)

Hossein Fereidooni, Alessandro Pegoraro, Phillip Rieger, Alexandra Dmitrienko, Ahmad-Reza Sadeghi

AI总结:

本文提出FreqFed,一种基于频域分析的联邦学习聚合机制,通过将模型更新变换到频域并识别核心频率分量来过滤恶意更新,从而在不依赖攻击类型和数据分布假设的情况下有效缓解定向与非定向投毒攻击,且对模型效用影响极小。

摘要:

联邦学习(FL)是一种协作学习范式,允许多个客户端在不共享其训练数据的情况下联合训练模型。然而,FL容易受到投毒攻击的影响,攻击者将操纵后的模型更新注入联邦模型聚合过程,以破坏或摧毁预测(非定向投毒)或植入隐藏功能(定向投毒或后门)。现有的FL投毒攻击防御存在若干局限,例如依赖关于攻击类型、策略或数据分布的特定假设,或者在面对先进注入技术与策略时不够鲁棒,同时难以维持聚合模型的效用。为解决现有防御的不足,我们采用一种通用且完全不同的方法来检测投毒(定向和非定向)攻击。我们提出FreqFed,一种新颖的聚合机制,将模型更新(即权重)变换到频域,在频域中我们可以识别出继承了足够权重信息的核心频率分量。这使我们能够在客户端本地训练期间有效过滤恶意更新,无论攻击类型、策略和客户端数据分布如何。我们在不同应用领域广泛评估了FreqFed的效率和有效性,包括图像分类、单词预测、物联网入侵检测和语音识别。我们证明FreqFed能够有效缓解投毒攻击,同时对聚合模型效用的影响可以忽略不计。

英文摘要:

Federated learning (FL) is a collaborative learning paradigm allowing multiple clients to jointly train a model without sharing their training data. However, FL is susceptible to poisoning attacks, in which the adversary injects manipulated model updates into the federated model aggregation process to corrupt or destroy predictions (untargeted poisoning) or implant hidden functionalities (targeted poisoning or backdoors). Existing defenses against poisoning attacks in FL have several limitations, such as relying on specific assumptions about attack types and strategies or data distributions or not sufficiently robust against advanced injection techniques and strategies and simultaneously maintaining the utility of the aggregated model. To address the deficiencies of existing defenses, we take a generic and completely different approach to detect poisoning (targeted and untargeted) attacks. We present FreqFed, a novel aggregation mechanism that transforms the model updates (i.e., weights) into the frequency domain, where we can identify the core frequency components that inherit sufficient information about weights. This allows us to effectively filter out malicious updates during local training on the clients, regardless of attack types, strategies, and clients' data distributions. We extensively evaluate the efficiency and effectiveness of FreqFed in different application domains, including image classification, word prediction, IoT intrusion detection, and speech recognition. We demonstrate that FreqFed can mitigate poisoning attacks effectively with a negligible impact on the utility of the aggregated model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑