Roots Beneath the Cut: Uncovering the Risk of Concept Revival in Pruning-Based Unlearning for Diffusion Models

Ci Zhang, Zhaojun Ding, Chence Yang, Jun Liu, Xiaoming Zhai +5 more
2/28/2026
cs.CVcs.LG

No official code found. Best implementation: Brankozz/Roots-Beneath-the-Cut (8 stars, MIT, last commit 2026-04-14).

Abstract

Pruning-based unlearning has recently emerged as a fast, training-free, and data-independent approach to remove undesired concepts from diffusion models. It promises high efficiency and robustness, offering an attractive alternative to traditional fine-tuning or editing-based unlearning. However, in this paper we uncover a hidden danger behind this promising paradigm. We find that the locations of pruned weights, typically set to zero during unlearning, can act as side-channel signals that leak critical information about the erased concepts. To verify this vulnerability, we design a novel attack framework capable of reviving erased concepts from pruned diffusion models in a fully data-free and training-free manner. Our experiments confirm that pruning-based unlearning is not inherently secure, as erased concepts can be effectively revived without any additional data or retraining. Extensive experiments on diffusion-based unlearning based on concept related weights lead to the conclusion: once the critical concept-related weights in diffusion models are identified, our method can effectively recover the original concept regardless of how the weights are manipulated. Finally, we explore potential defense strategies and advocate safer pruning mechanisms that conceal pruning locations while preserving unlearning effectiveness, providing practical insights for designing more secure pruning-based unlearning frameworks.

View on arXivView PDF

Code Implementations(1)

MIT

Official implementation of CVPR 2026 "Roots Beneath the Cut: Uncovering the Risk of Concept Revival in Pruning-Based Unlearning for Diffusion Models"

80Shell, PythonFeb 22, 20266 months agoMIT

Similar Papers

Cite this paper

@article{zhang2026roots,
  title  = {Roots Beneath the Cut: Uncovering the Risk of Concept Revival in Pruning-Based Unlearning for Diffusion Models},
  author = {Ci Zhang and Zhaojun Ding and Chence Yang and Jun Liu and Xiaoming Zhai and Shaoyi Huang and Beiwen Li and Xiaolong Ma and Jin Lu and Geng Yuan},
  year   = {2026},
  eprint = {2603.06640},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.06640v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.06640.svg)](https://codeofpaper.com/paper/2603.06640)

Discussion