IDU: Unified Unlearning and One-Step Distillation for Flow and Diffusion Models

Aleksei Leonov · hf · 2026-10-06

A Hugging Face paper introduces Inverse Distillation Unlearning (IDU), reportedly the first unified framework combining unlearning and distillation for unconditional flow-matching and score-based models: distill a multi-step teacher into an efficient one-step student while suppressing outputs tied to a forget set.

Approach: formulate distillation as a min-max objective over a data distribution, represent it as a mixture of the forget-set and generated distributions, and compare against the teacher's training distribution so only retained data is recovered at optimum. Requires only a pretrained full-data teacher and forget-set data—no access to retained samples, extra feature extractors, or classifiers.

Results: on MNIST and CIFAR-10 under both settings, IDU substantially cuts generation of forgotten classes while preserving quality on retained classes.

Original post →

More from Multimodal

Multimodal channel →