Our increasingly connected world continues to face an ever-growing amount of network-based attacks. Intrusion detection systems (IDS) are an essential security technology for detecting these attacks. Although numerous machine learning-based IDS have been proposed for the detection of malicious network traffic, the majority have difficulty properly detecting and classifying the more uncommon attack types. In this paper, we implement a novel hybrid technique using synthetic data produced by a Generative Adversarial Network (GAN) to use as input for training a Deep Reinforcement Learning (DRL) model. Our GAN model is trained with the NSL-KDD dataset for four attack categories as well as normal network flow. Ultimately, our findings demonstrate that training the DRL on specific synthetic datasets can result in better performance in correctly classifying minority classes over training on the true imbalanced dataset.
翻译:入侵探测系统(IDS)是检测这些袭击的基本安全技术。 虽然已提出许多基于机器学习的IDS用于检测恶意网络交通,但大多数人难以正确检测和分类更罕见的攻击类型。 在本文中,我们采用了一种新型混合技术,使用由基因反转网络(GAN)制作的合成数据来培训深强化学习模型。我们的GAN模型是用NSL-KDD数据集对四种攻击类别和正常网络流动进行的培训。最终,我们的研究结果表明,对DRL进行具体合成数据集培训,可以导致更好地将少数群体课程与真正失衡数据集培训进行正确分类。