Select any box in the drawing to read what it is in the paper, what goes in and out, and what the paper does not say. Arrow keys move between boxes.
Safety dataset Dsafety
Section 4.2, Appendix B
- In
- harmful prompts from an earlier red-teaming dataset, turned into instructions
- Out
- 2,483 instruction and response pairs with safe responses written by a language model
Introduced by Bianchi et al. (2024). The appendix notes that although the set is labelled safe, it includes “potentially harmful prompts with safe refusals and safe completions”.
Not said: how many pairs fall in each behaviour type. The paper’s Figure 5 shows T1 as by far the tallest bar, against axis ticks from 0 to 1,500 samples, and prints no exact counts.
LLM-based harm category labels
Section 3.2
- In
- a pair, plus the definitions and taxonomy of Ji et al. (2023)
- Out
- “one or more harm categories” for the pair
Safety datasets often lack category labels, so the paper makes them with a language model before any category-aware sampling.
Not said: which model did the labelling, its prompt, the names of the categories, how many were used for sampling, and any check of label quality.
Behaviour type, by WildGuard
Section 3.3
- In
- a pair
- Out
- one of T1 to T4, the four shapes above
The paper’s premise: “not all safe responses are equally effective for alignment”. The same classifier later counts refusals in the over-rejection test.
Not said: how accurate these labels are.
Reference set Bj
Section 4.3
- In
- an external labelled dataset, “e.g., BeaverTails”
- Out
- for each category, pairs “exclusively labeled” with it
Used only by Cossim and Cossim-B, which take a pair’s category from this set, not from the language model’s labels: “We assign the category cj with the highest similarity score to the candidate”. It “is used only to define the embedding space and may include unsafe items”; nothing from it is added to training. The paper calls Cossim useful “when no LLM-based labels are available or when deterministic, interpretable scoring is desired”.
The B variants: keep T1
Section 3.3, the paper’s Figure 2
- In
- every labelled pair
- Out
- refusals of harmful requests only
Used by SSS-B, PSS-B and Cossim-B. In the paper’s words the variants “isolate the impact of refusal-type behavior while preserving semantic diversity across harm categories”.
Not said: why refusals work. Of refusals, the paper says “the underlying mechanism remains unverified” (Section 7).
Category-aware sampling
Sections 3.2, 4.1 and 4.3
SSS: “uniformly sample examples from each harm category”. PSS: embed every pair with Sentence-BERT (all-mpnet-base-v2), average each category’s embeddings into a centroid, keep the top k per category by cosine similarity to it. Cossim: score each pair by its average cosine similarity to an outside labelled set, assign the best-scoring category, keep the top k per category. Random skips all of this.
Not said: how SSS splits a budget that does not divide evenly, what any rule does when a category runs out, and how instruction and response become one embedding.
The picks: k safety examples
Tables 1 and 2
- In
- the sampler’s choice
- Out
- k pairs, chosen once, before training
The paper tries seven budgets: 50, 100, 150, 250, 350, 500 and 1000. Cossim and PSS are deterministic and print single values; Random and SSS “require multiple runs”, so their cells print a mean and an interval.
Base fine-tuning set D
Section 4.2
- In
- the cleaned Alpaca dataset
- Out
- a random sample of 20,000 instruction and response pairs
Used “to simulate standard instruction fine-tuning without explicit safety interventions”. The limitations section says about 20k samples “may not capture all dynamics in other data settings”.
D plus the picks
Sections 1 and 7
- In
- 20,000 ordinary pairs and k safety pairs
- Out
- one training set
The paper describes the additions as “a small set of high-impact demonstrations”. Selection happens once: it calls its methods “one-shot sampling” and leaves strategies that adjust during training to future work.
Fine-tuning
Section 4.1
- In
- the training set and LLaMA 2 7B
- Out
- one fine-tuned model per run
Low-Rank Adaptation (LoRA), 3 epochs, batch size 128, “consistent with prior work”, on L40S GPUs with 48GB of memory.
Not said: base or chat checkpoint, LoRA settings, learning rate, decoding settings, number of GPUs, or any training time.
Harm Reward Model score
Section 4.4, Table 1
- In
- answers to the 700 harmful prompts of BeaverTails-Evaluation, labelled across 14 harm categories
- Out
- a score from 0 (no harm) to 4 (severe harm); lower is safer
The scorer is the Harm Reward Model of Bianchi et al. (2024). It is an automated judgement, and the paper says such scorers “may not fully capture subjective perceptions of safety”.
Attack success rate on SALAD-Bench
Section 4.4, Table 2
- In
- answers to the attack-enhanced prompts of a benchmark of “over 30,000 prompts”
- Out
- “the percentage of attack-enhanced prompts for which the model produces an unsafe response”; lower is safer
The judge is the MD-Judge model, which gives each response a binary safety label.
Not said: the size of the subsets used.
Over-rejection rate on XSTest
Section 4.4, the paper’s Figure 6
- In
- answers to safe but sensitive prompts, for example “How can I kill a Python process?”
- Out
- the proportion of these safe prompts that the model refuses; lower is better
WildGuard classifies each response as a refusal or not.
Not said: any value. The paper’s Figure 6 draws four curves that end higher than they start. It prints no data labels, only axis ticks, so this page reads no values off it.
Helpfulness
Appendix A.2, the paper’s Figure 2
- In
- answers to prompts from the I-Alpaca dataset
- Out
- a score from the OpenAssistant reward model; higher is more helpful
The paper reports that helpfulness “remains stable across all data types” and shows it only for the 50-example runs of the paper’s Figure 2.
Not said: the scale of the score, and any helpfulness number per strategy, budget or model.