Official PyTorch implementation of Safety-Guided Flow (SGF), as presented in our paper:

Safety-Guided Flow (SGF): A Unified Framework for Negative Guidance in Safe Generation (ICLR2026)
Mingyu Kim, Young-Heon Kim and Mijung Park

• Safety (SD-v1.4): Nudity + COCO-30k.
• Memorization (SD-v2.1): memorization / replication setting.
• Diversity (SD-v3): diversity evaluation setting.

• (Jan. 2026) Our paper has been accepted to ICLR 2026.
• (Feb. 2026) Our paper has been selected for an Oral Presentation at ICLR 2026.

• Safety (SD-v1.4): see nudity_sdv1/README.md
• Memorization (SD-v2.1): see memorized_sdv21/README.md
• Diversity (SD-v3): see diversity_sdv3/README.md

@inproceedings{
  kim2026safetyguided,
  title={Safety-Guided Flow (SGF): A Unified Framework for Negative Guidance in Safe Generation},
  author={Mingyu Kim and Young-Heon Kim and Mijung Park},
  booktitle={The Fourteenth International Conference on Learning Representations},
  year={2026},
  url={https://openreview.net/forum?id=EA80Zib9UI}
}