arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2210.04052cs.CRcs.LG
收藏

FedDef:防御基于联邦学习的网络入侵检测系统中的梯度泄露

FedDef: Defense Against Gradient Leakage in Federated Learning-based Network Intrusion Detection Systems

Tsinghua University(清华大学)

Jiahui Chen, Yi Zhao, Qi Li, Xuewei Feng, Ke Xu

AI总结:

本文针对联邦学习网络入侵检测系统中的梯度泄露风险,提出隐私评估指标和基于优化的输入扰动防御策略FedDef,在保持模型精度的同时显著提升隐私保护能力。

摘要:

深度学习(DL)方法已被广泛应用于基于异常的网络入侵检测系统(NIDS)以检测恶意流量。为拓展DL方法的使用场景,联邦学习(FL)允许多个用户在尊重个人数据隐私的基础上训练全局模型。然而,在现有防御措施下,基于FL的NIDS对现有隐私攻击的鲁棒性尚未得到系统评估。为解决此问题,我们提出两种专为基于FL的NIDS设计的隐私评估指标:(1)隐私分数,利用重构攻击评估原始与恢复的流量特征之间的相似性;(2)规避率,利用恢复的流量进行对抗攻击以评估其对抗NIDS的能力。我们通过实验表明,现有防御措施几乎无法提供保护,且相应的对抗流量甚至能规避SOTA NIDS Kitsune。为防御此类攻击并构建更鲁棒的基于FL的NIDS,我们进一步提出FedDef,一种具有理论保证的新型基于优化的输入扰动防御策略。它通过最小化梯度距离实现高实用性,并通过最大化输入距离实现强隐私保护。我们在四个数据集上评估了四种现有防御措施,结果表明我们的防御在隐私保护方面优于所有基线方法,隐私分数最高提高7倍,同时在最优参数组合下模型精度损失保持在3%以内。

英文摘要:

Deep learning (DL) methods have been widely applied to anomaly-based network intrusion detection system (NIDS) to detect malicious traffic. To expand the usage scenarios of DL-based methods, federated learning (FL) allows multiple users to train a global model on the basis of respecting individual data privacy. However, it has not yet been systematically evaluated how robust FL-based NIDSs are against existing privacy attacks under existing defenses. To address this issue, we propose two privacy evaluation metrics designed for FL-based NIDSs, including (1) privacy score that evaluates the similarity between the original and recovered traffic features using reconstruction attacks, and (2) evasion rate against NIDSs using adversarial attack with the recovered traffic. We conduct experiments to illustrate that existing defenses provide little protection and the corresponding adversarial traffic can even evade the SOTA NIDS Kitsune. To defend against such attacks and build a more robust FL-based NIDS, we further propose FedDef, a novel optimization-based input perturbation defense strategy with theoretical guarantee. It achieves both high utility by minimizing the gradient distance and strong privacy protection by maximizing the input distance. We experimentally evaluate four existing defenses on four datasets and show that our defense outperforms all the baselines in terms of privacy protection with up to 7 times higher privacy score, while maintaining model accuracy loss within 3% under optimal parameter combination.

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

↑