PyRIT: A Framework for Security Risk Identification and Red Teaming in Generative AI System
<|reference_start|>PyRIT: A Framework for Security Risk Identification and Red Teaming in Generative AI System: Generative Artificial Intelligence (GenAI) is becoming ubiquitous in our daily lives. The increase in computational power and data availability has led to a proliferation of both single- and multi-modal models. As the GenAI ecosystem matures, the need for extensible and model-agnostic risk identification frameworks is growing. To meet this need, we introduce the Python Risk Identification Toolkit (PyRIT), an open-source framework designed to enhance red teaming efforts in GenAI systems. PyRIT is a model- and platform-agnostic tool that enables red teamers to probe for and identify novel harms, risks, and jailbreaks in multimodal generative AI models. Its composable architecture facilitates the reuse of core building blocks and allows for extensibility to future models and modalities. This paper details the challenges specific to red teaming generative AI systems, the development and features of PyRIT, and its practical applications in real-world scenarios.<|reference_end|>
@article{munoz2024pyrit:,
    title={PyRIT: A Framework for Security Risk Identification and Red Teaming in
  Generative AI System},
    author={Gary D. Lopez Munoz, Amanda J. Minnich, Roman Lutz, Richard Lundeen,
  Raja Sekhar Rao Dheekonda, Nina Chikanov, Bolor-Erdene Jagdagdorj, Martin
  Pouliot, Shiven Chawla, Whitney Maxwell, Blake Bullwinkel, Katherine Pratt,
  Joris de Gruyter, Charlotte Siska, Pete Bryan, Tori Westerhoff, Chang
  Kawaguchi, Christian Seifert, Ram Shankar Siva Kumar, Yonatan Zunger},
    journal={arXiv preprint arXiv:2410.02828},
    year={2024},
    archivePrefix={arXiv},
    eprint={2410.02828},
    primaryClass={cs.CR cs.AI cs.CL}
}