ml-research/localizing_memorization_in_diffusion_models
[NeurIPS 2024] Source code for our paper "Finding NeMo: Localizing Neurons Responsible For Memorization in Diffusion Models".
This project helps AI safety researchers and machine learning engineers identify and address privacy concerns in diffusion models. It takes a trained diffusion model and a set of prompts as input. The output is a list of specific neurons responsible for memorizing individual training images, allowing you to deactivate them to prevent sensitive or copyrighted data from being reproduced. This tool is for those who develop, deploy, or audit large image generation models.
No commits in the last 6 months.
Use this if you need to pinpoint exactly which parts of a diffusion model are causing it to reproduce training data, enhancing privacy and mitigating intellectual property risks.
Not ideal if you are looking for a simple plug-and-play solution to prevent all memorization without understanding the underlying neural network components.
Stars
13
Forks
1
Language
Python
License
MIT
Category
Last pushed
Jul 18, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/ml-research/localizing_memorization_in_diffusion_models"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
bghira/SimpleTuner
A general fine-tuning kit geared toward image/video/audio diffusion models.
mcmonkeyprojects/SwarmUI
SwarmUI (formerly StableSwarmUI), A Modular Stable Diffusion Web-User-Interface, with an...
nateraw/stable-diffusion-videos
Create 🔥 videos with Stable Diffusion by exploring the latent space and morphing between text prompts
TheDesignFounder/DreamLayer
Benchmark diffusion models faster. Automate evals, seeds, and metrics for reproducible results.