microsoft / microsoft/PyRIT

FEAT: Native MITRE ATLAS v5.4 technique tagging for orchestrator outputs and scoring results

Open
#2,126 5 comments 0 reactions 0 assignees View on GitHub
praise
Dominant language
Python
Stars
4.5k
Forks
893
Avg merge
3d 50m
Merged PRs (30d)
165

Description

Problem Statement
PyRIT is widely used as the execution engine for AI red teaming, but currently has no native MITRE ATLAS technique mapping built into its orchestrators, scorers, or reporting layer. As noted in the CSA's Agentic AI Red Teaming Guide (2026): "MITRE ATLAS techniques can be encoded into PyRIT orchestrators and datasets; PyRIT doesn't have its own TTP ontology." This means practitioners must manually map every attack run to ATLAS techniques — an error-prone, time-consuming step that breaks the automated red teaming pipeline.
Why This Matters Now
MITRE ATLAS released v5.4.0 in February 2026, adding 14+ new agentic AI techniques (e.g., AML.T0051 Prompt Injection, AML.T0054 Jailbreak, "Publish Poisoned AI Agent Tool", "Escape to Host"). Azure AI security best practices documentation (learn.microsoft.com/azure/security/fundamentals/ai-security-best-practices) explicitly recommends testing against ATLAS tactics — yet PyRIT produces no ATLAS-tagged output that can be fed directly into Azure Monitor, Microsoft Sentinel, or compliance reports.
Proposed Solution
Add an optional atlas_technique_id metadata field to:

AttackStrategy / orchestrator configuration (user declares intent: "this scenario maps to AML.T0051")
Score output objects (auto-populated from declared mapping)
Memory/logging layer (persisted alongside prompt-response pairs)
Summary report output (ATLAS technique coverage table: which techniques were tested, pass/fail rates per technique)

Minimal Implementation Path
A lightweight YAML-based mapping file (e.g., atlas_mappings.yaml) could define which datasets/converters correspond to which ATLAS technique IDs. This wouldn't require hardcoding ATLAS logic into PyRIT's core — just a tagging layer that flows through the existing memory interface.
Azure Security Impact
This would directly enable:

Azure DevOps CI/CD gates that fail on untested ATLAS technique categories
Sentinel workbooks that visualize ATLAS technique coverage over time
Compliance evidence for Azure AI security benchmark (MCSB v2 AI-5, AI-6)

Happy to draft a reference schema for the mapping file.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing how AttackStrategy or orchestrator configuration, score output objects, the memory/logging layer, and summary reports are represented and connected. Review the existing memory interface and reporting entry points before deciding how an optional atlas_technique_id and YAML mapping should flow through them. Done should include persisted technique tags and a report showing technique coverage with pass/fail rates, but the schema and implementation boundaries still need agreement.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
ai, security
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Active
Clarity
Needs clarification
Newbie friendliness
32/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.