跳到正文
arXiv:cs.AI· Roman Belaire, Arunesh Sinha, Pradeep Varakantham·· 10 小时前

用分层强化学习进行多轮对话红队测试

There are Levels to It: Red Teaming Dialogue With Hierarchical Reinforcement Learning

AI 导读

研究提出一种新的红队测试范式,将红队测试形式化为分层强化学习框架中的马尔可夫决策过程,通过 token 级 harm reward 训练生成式智能体学习多样化的多轮攻击,持续发现可绕过基线的漏洞。该方法在稀疏奖励与长时程规划挑战下取得新的 SOTA,将 LLM 红队测试重构为基于轨迹的原则性过程。

正文

View PDF HTML (experimental)

Abstract:Red teaming is essential for securing Large Language Models, yet current automated methods remain limited by templates and single-turn attacks. To simulate the complex, interactive nature of real-world adversarial attacks, we introduce a novel red teaming paradigm designed to maximize expected cumulative harm through strategic interaction. By formalizing red teaming as a Markov Decision Process in a hierarchical reinforcement learning framework, we navigate the challenges of sparse rewards and long-horizon planning. Our generative agent learns diverse, multi-turn attacks using a token-level harm reward, consistently uncovering vulnerabilities that bypass baselines. This approach achieves a new state of the art and reframes LLM red teaming as a principled, trajectory-based process.
Comments: NeurIPS 2026
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
Cite as: arXiv:2508.04451 [cs.LG]
  (or arXiv:2508.04451v2 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2508.04451

arXiv-issued DOI via DataCite

Submission history

From: Roman Belaire [view email]
[v1] Wed, 6 Aug 2025 13:52:00 UTC (2,043 KB)
[v2] Tue, 13 Oct 2026 01:53:44 UTC (1,893 KB)

来源:arXiv:cs.AI · arxiv.org