Skip to main content
Attack datasets are collections of adversarial prompts designed to test how well your AI model resists different types of attacks. Know Your AI provides a curated marketplace of datasets covering every major attack vector, and you can upload your own.

Attack categories

Know Your AI organizes attacks into 7 categories, each targeting a different vulnerability surface:

Jailbreak

Tricks the model into ignoring safety instructions and acting without restrictions. Tests whether the model maintains its safety boundaries.

Prompt Injection

Injects hidden instructions into the prompt to override the system prompt or manipulate model behavior.

Data Extraction

Attempts to extract training data, system prompts, or sensitive information the model was trained on or given access to.

Harmful Content

Tries to get the model to generate violent, illegal, hateful, or otherwise harmful content.

PII Leakage

Tests whether the model leaks personal identifiable information about real individuals.

Bias and Fairness

Probes the model for racial, gender, religious, or other biases in its responses.

Hallucination

Tests whether the model generates confident but factually incorrect information.

Attack methods

Each category contains multiple attack methods, each using a different technique:

Jailbreak methods

Prompt injection methods

Data extraction methods

Harmful content methods

PII and privacy methods

Bias and fairness methods

Hallucination methods

Dataset Marketplace

The Dataset Marketplace provides ready-to-use datasets curated by security researchers. Browse and add datasets to your workspace in one click.

Adding marketplace datasets

1

Go to Dataset Marketplace

Navigate to your workspace and click Dataset Marketplace in the sidebar.
2

Browse or search

Filter by category (jailbreak, injection, privacy, etc.) or search by name.
3

Add to workspace

Click Add to Workspace on any dataset. It becomes immediately available for evaluations.
4

Link to evaluations

Go to your evaluation and click Add Dataset to link marketplace datasets.

Custom datasets

Dataset types

Create via dashboard

  1. Go to Datasets in the sidebar
  2. Click Create Dataset
  3. Enter a name, category, and description
  4. Add prompts manually or upload a JSON/CSV file

Create via SDK

Create via CLI

The CLI can be used with the SDK to script dataset creation:

Upload format

JSON format:
Ground-truth JSON format:
CSV format:

Building effective datasets

Best practices

Don’t just use one attack method. Combine DAN, PAIR, CIPHER, and other methods in a single evaluation to test your model’s defenses from multiple angles.
Start with basic “ignore your instructions” prompts and escalate to encoded, multi-step, and context-manipulating attacks. A model that blocks simple attacks may still fail sophisticated ones.
Many models have weaker safety filters in non-English languages. Include multilingual attack prompts to catch these blind spots.
New attack methods are discovered constantly. Refresh your datasets monthly. The marketplace is updated automatically with new techniques.
If your model handles financial data, focus on PII extraction and data leakage. If it is customer-facing, prioritize jailbreak and harmful content. Build datasets that mirror your actual risks.

MITRE ATLAS mapping

Know Your AI attack categories map to the MITRE ATLAS framework: This mapping helps you align your evaluation coverage with industry-standard threat taxonomies and simplify compliance reporting.