arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.04809cs.LGcs.AI
收藏

超越奖励的强化学习在网络安全防御中的应用

Beyond Rewards in Reinforcement Learning for Cyber Defence

University of Cambridge(剑桥大学)

Elizabeth Bates, Chris Hicks, Vasilios Mavroudis

AI总结:

本文研究了在网络安全防御中使用强化学习时,奖励函数结构对学习和策略行为的影响,通过比较稀疏和密集奖励函数,揭示了奖励、动作空间和子最优策略风险之间的复杂关系。

摘要:

近年来,自主网络安全防御代理在使用深度强化学习保护计算机网络方面引起了广泛关注。这些代理通常在网络安全 gym 环境中训练,使用密集的、高度工程化的奖励函数,结合多种惩罚和激励,以应对各种(不) desirable 状态和昂贵的操作。密集奖励有助于缓解探索复杂环境的挑战,但会偏向于次优且可能风险更大的解决方案,这对复杂的网络安全环境至关重要。我们通过多种稀疏和密集奖励函数、两种已确立的网络安全 gym、不同网络规模以及策略梯度和基于价值的 RL 算法,全面评估了奖励函数结构对学习和策略行为特征的影响。我们的评估得益于一种新的真实评估方法,使可以直接比较不同的奖励函数,揭示了奖励、动作空间和网络安全环境中子最优策略风险之间的微妙关系。我们的结果表明,稀疏奖励,如果目标一致且可以频繁遇到,能够提供增强的训练可靠性和更有效的网络安全防御代理,具有较低风险的策略。令人惊讶的是,稀疏奖励还能产生与网络安全守护者目标更一致的策略,并在不使用显式奖励基于数值惩罚的情况下,节省昂贵的防御操作。

英文摘要:

Recent years have seen an explosion of interest in autonomous cyber defence agents trained to defend computer networks using deep reinforcement learning. These agents are typically trained in cyber gym environments using dense, highly engineered reward functions which combine many penalties and incentives for a range of (un)desirable states and costly actions. Dense rewards help alleviate the challenge of exploring complex environments but risk biasing agents towards suboptimal and potentially riskier solutions, a critical issue in complex cyber environments. We thoroughly evaluate the impact of reward function structure on learning and policy behavioural characteristics using a variety of sparse and dense reward functions, two well-established cyber gyms, a range of network sizes, and both policy gradient and value-based RL algorithms. Our evaluation is enabled by a novel ground truth evaluation approach which allows directly comparing between different reward functions, illuminating the nuanced inter-relationships between rewards, action space and the risks of suboptimal policies in cyber environments. Our results show that sparse rewards, provided they are goal aligned and can be encountered frequently, uniquely offer both enhanced training reliability and more effective cyber defence agents with lower-risk policies. Surprisingly, sparse rewards can also yield policies that are better aligned with cyber defender goals and make sparing use of costly defensive actions without explicit reward-based numerical penalties.

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

↑