Towards SFW sampling for diffusion models via external conditioning

TL;DR

This paper introduces an external conditioning method called SFW sampler for diffusion models to effectively reduce unsafe content generation without extensive fine-tuning, maintaining image quality.

Contribution

The paper proposes a novel external conditioning approach using multimodal models and CLIP for safe content generation in diffusion models, avoiding fine-tuning.

Findings

Effective reduction of NSFW content in generated images.

Minimal impact on image quality for non-NSFW samples.

Comparable performance to fine-tuning methods in safety measures.

Abstract

Score-based generative models (SBM), also known as diffusion models, are the de facto state of the art for image synthesis. Despite their unparalleled performance, SBMs have recently been in the spotlight for being tricked into creating not-safe-for-work (NSFW) content, such as violent images and non-consensual nudity. Current approaches that prevent unsafe generation are based on the models' own knowledge, and the majority of them require fine-tuning. This article explores the use of external sources for ensuring safe outputs in SBMs. Our safe-for-work (SFW) sampler implements a Conditional Trajectory Correction step that guides the samples away from undesired regions in the ambient space using multimodal models as the source of conditioning. Furthermore, using Contrastive Language Image Pre-training (CLIP), our method admits user-defined NSFW classes, which can vary in different settings. Our experiments on the text-to-image SBM Stable Diffusion validate that the proposed SFW sampler effectively reduces the generation of explicit content while being competitive with other fine-tuning-based approaches, as assessed via independent NSFW detectors. Moreover, we evaluate the impact of the SFW sampler on image quality and show that the proposed correction scheme comes at a minor cost with negligible effect on samples not needing correction. Our study confirms the suitability of the SFW sampler towards aligned SBM models and the potential of using model-agnostic conditioning for the prevention of unwanted images.