TensorX
返回文献探索

Paper · arXiv 2501.01830

Auto-RT: Automatic Jailbreak Strategy Exploration for Red-Teaming Large Language Models

Yanjiang Liu, Shuhen Zhou, Yaojie Lu, Huijia Zhu, Weiqiang Wang, Hongyu Lin, Ben He, Xianpei Han, Le Sun

17 upvotesJanuary 3, 2025arXiv 预印本
AI 摘要

Auto-RT, a reinforcement learning framework, enhances the detection of complex vulnerabilities in large language models by optimizing attack strategies through early exploration and progressive reward tracking.

reinforcement learningAuto-RTEarly-terminated ExplorationProgressive Reward Trackingintermediate downgrade modelsvulnerability exploitationlarge language modelsattack strategiesexploration efficiencysuccess rates

Abstract

Automated red-teaming has become a crucial approach for uncovering vulnerabilities in large language models (LLMs). However, most existing methods focus on isolated safety flaws, limiting their ability to adapt to dynamic defenses and uncover complex vulnerabilities efficiently. To address this challenge, we propose Auto-RT, a reinforcement learning framework that automatically explores and optimizes complex attack strategies to effectively uncover security vulnerabilities through malicious queries. Specifically, we introduce two key mechanisms to reduce exploration complexity and improve strategy optimization: 1) Early-terminated Exploration, which accelerate exploration by focusing on high-potential attack strategies; and 2) Progressive Reward Tracking algorithm with intermediate downgrade models, which dynamically refine the search trajectory toward successful vulnerability exploitation. Extensive experiments across diverse LLMs demonstrate that, by significantly improving exploration efficiency and automatically optimizing attack strategies, Auto-RT detects a boarder range of vulnerabilities, achieving a faster detection speed and 16.63\% higher success rates compared to existing methods.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Auto-RT: Automatic Jailbreak Strategy Exploration for Red-Teaming Large Language Models | TensorX