Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SelfGrader: LLM Jailbreak Detection via Anchored Token-Level Logits

About

Large Language Models (LLMs) are powerful tools for answering user queries, yet they remain highly vulnerable to jailbreak attacks. Existing guardrail methods typically rely on internal features or textual responses to detect malicious queries, which either introduce substantial latency or suffer from randomness in text generation. To overcome these limitations, we propose SelfGrader, a lightweight guardrail method that formulates jailbreak detection as a numerical grading problem using anchored token-level logits. Specifically, SelfGrader evaluates the safety of a user query within a compact set of numerical tokens (NTs) (e.g., 0-9) and interprets their logit distribution as an internal safety signal. To align these signals with the target safety rubric, SelfGrader constructs Probably Approximately Correct-guided ICL anchor examples and introduces a dual-perspective scoring rule that considers both the maliciousness and benignness of the query, yielding a stable and interpretable score that reflects harmfulness and reduces the false positive rate simultaneously. Extensive experiments across diverse jailbreak benchmarks, adaptive attacks, benign prompt benchmarks, multiple LLMs, and state-of-the-art guardrail baselines demonstrate that SelfGrader achieves strong robustness with low false positive rates, memory overhead, and latency.

Zikai Zhang, Rui Hu, Olivera Kotevska, Jiahao Xu• 2026

Related benchmarks

TaskDatasetResultRank
Jailbreak DefenseManual (IJP)
ASR0.00e+0
38
Jailbreak DetectionAverage of six attacks
Avg Success Rate7.45
38
Jailbreak DefenseMultiJail
ASR0.00e+0
36
Jailbreak DefenseActorAttack
Attack Success Rate (ASR)0.00e+0
34
Safety GuardrailingHumanEval
False Positive Rate0.00e+0
32
Safety GuardrailingOR-Bench
False Positive Rate0.00e+0
26
Safety GuardrailingAlpacaEval
False Positive Rate0.00e+0
24
Safety GuardrailingGSM8K
False Positive Rate0.00e+0
24
Jailbreak DefenseDrAttack
ASR0.00e+0
22
Jailbreak DefenseTAP
ASR6
18
Showing 10 of 33 rows

Other info

Follow for update