LLM-Assisted Red Teaming of Diffusion Models through "Failures Are Fated, But Can Be Faded"

Som Sagar, Aditya Taparia, Ransalu Senanayake

Keywords: Failure Detection, Reinforcement Learning, Trustworthy Machine Learning

TL;DR: Characterizing and mitigating failure modes in diffusion models

Abstract: In large deep neural networks that seem to perform surprisingly well on many tasks, we also observe a few failures related to accuracy, social biases, and alignment with human values, among others. Therefore, before deploying these models, it is crucial to characterize this failure landscape for engineers to debug or audit models. Nevertheless, it is infeasible to exhaustively test for all possible combinations of factors that could lead to a model's failure. In this workshop paper, we improve the "Failures are fated, but can be faded" framework—a post-hoc method to explore and construct the failure landscape in pre-trained generative models—with a variety of *deep reinforcement learning* algorithms, screening tests, and LLM-based rewards. With the aid of limited human feedback, we then demonstrate how to restructure the failure landscape to be more desirable by moving away from the discovered failure modes. We empirically demonstrate the effectiveness of the proposed method on diffusion models. We also highlight the strengths and weaknesses of each algorithm in identifying failure modes.

Serve As Reviewer: [e-mail]

Submission Number: 31