论文标题

FLAD:DDOS攻击检测的自适应联合学习

FLAD: Adaptive Federated Learning for DDoS Attack Detection

论文作者

Doriguzzi-Corin, Roberto, Siracusa, Domenico

论文摘要

联邦学习(FL)最近从网络安全社区获得了越来越多的考虑,这是一种与网络威胁的分布式培训培训深度学习模型的一种方式,而没有披露培训数据。然而,网络安全中FL的采用仍处于起步阶段,并且尚未正确解决一系列实际方面。实际上,FL概念核心的联合平均算法需要测试数据的可用性来控制FL过程。尽管这在某些域中可能是可行的,但是如果不披露敏感信息,新发现的攻击的测试网络流量不能总是共享的。在本文中,我们解决了在动态网络安全方案中FL过程的融合,在该方案中,必须经常使用训练有素的模型以新的近期攻击配置文件进行更新,以使联邦所有成员都具有最新的检测功能。为此,我们提出了FLAD(自适应联合学习方法DDOS攻击检测),这是基于自适应机制的网络安全应用的FL解决方案,该机制通过向那些更难学习的攻击概况分配更多计算来协调FL过程,而无需分享任何测试数据以监视训练有素的模型的性能。使用最新的DDOS攻击数据集,我们证明,在一系列不平衡的异质DDOS攻击数据集中,FLAD在收敛时间和准确性方面都超过了最先进的FL算法。我们还在现实的场景中展示了方法的鲁棒性,在该场景中,我们多次对深度学习模型进行了重新训练,以在预训练的模型上引入新攻击的概况。

Federated Learning (FL) has been recently receiving increasing consideration from the cybersecurity community as a way to collaboratively train deep learning models with distributed profiles of cyber threats, with no disclosure of training data. Nevertheless, the adoption of FL in cybersecurity is still in its infancy, and a range of practical aspects have not been properly addressed yet. Indeed, the Federated Averaging algorithm at the core of the FL concept requires the availability of test data to control the FL process. Although this might be feasible in some domains, test network traffic of newly discovered attacks cannot be always shared without disclosing sensitive information. In this paper, we address the convergence of the FL process in dynamic cybersecurity scenarios, where the trained model must be frequently updated with new recent attack profiles to empower all members of the federation with the latest detection features. To this aim, we propose FLAD (adaptive Federated Learning Approach to DDoS attack detection), an FL solution for cybersecurity applications based on an adaptive mechanism that orchestrates the FL process by dynamically assigning more computation to those members whose attacks profiles are harder to learn, without the need of sharing any test data to monitor the performance of the trained model. Using a recent dataset of DDoS attacks, we demonstrate that FLAD outperforms state-of-the-art FL algorithms in terms of convergence time and accuracy across a range of unbalanced datasets of heterogeneous DDoS attacks. We also show the robustness of our approach in a realistic scenario, where we retrain the deep learning model multiple times to introduce the profiles of new attacks on a pre-trained model.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源