We introduce SAFEMax, a novel method for Machine Unlearning in diffusion models. Grounded in information-theoretic principles, SAFEMax maximizes the entropy in generated images, causing the model to generate Gaussian noise when conditioned on impermissible classes by ultimately halting its denoising process. Also, our method controls the balance between forgetting and retention by selectively focusing on the early diffusion steps, where class-specific information is prominent. Our results demonstrate the effectiveness of SAFEMax and highlight its substantial efficiency gains over state-of-the-art methods.
@article{arxiv.2508.20773,
title = {Unleashing Uncertainty: Efficient Machine Unlearning for Generative AI},
author = {Christoforos N. Spartalis and Theodoros Semertzidis and Petros Daras and Efstratios Gavves},
journal= {arXiv preprint arXiv:2508.20773},
year = {2025}
}
Comments
ICML 2025 workshop on Machine Unlearning for Generative AI