基于深度强化学习的对抗性攻击防御算法

AN ADVERSARIAL ATTACK DEFENSE ALGORITHM BASED ON DEEP REINFORCEMENT LEARNING

  • 摘要: 基于深度学习的恶意流量检测技术被广泛应用于现实场景下的入侵检测系统,为保障对抗性攻击场景下入侵检测系统的有效性,从模型优化角度,提出一种基于深度强化学习的对抗性训练算法。采用梯度投影下降算法,在阈值范围能生成非线性扰动,采用构造深度强化学习模型,通过与目标模型的交互搜索优化对抗样本,用最优对抗样本对模型进行训练,提升其鲁棒性。

     

    Abstract: Malicious traffic detection technology based on deep learning is widely used in intrusion detection systems in realistic scenarios. In order to ensure the effectiveness of intrusion detection systems in adversarial attack scenarios, this paper proposes an adversarial training algorithm based on deep reinforcement learning from the perspective of model optimization. The gradient projection descent algorithm was used to generate nonlinear disturbances in the threshold range. A deep reinforcement learning model was used to optimize the adversarial samples through interactive search with the target model. The model was trained with the optimal adversarial samples to improve its robustness.

     

/

返回文章
返回