Decoupled Distillation to Erase: A General Unlearning Method for Any Class-centric Tasks
Yu Zhou, Dian Zheng, Qijie Mo, Renjie Lu, Kun-Yu Lin, Wei-Shi Zheng
Abstract
In this work, we present DEcoupLEd Distillation To Erase (DELETE), a general and strong unlearning method for any class-centric tasks. To derive this, we first propose a theoretical framework to analyze the general form of unlearning loss and decompose it into forgetting and retention terms. Through the theoretical framework, we point out that a class of previous methods could be mainly formulated as a loss that implicitly optimizes the forgetting term while lacking supervision for the retention term, disturbing the distribution of pre-trained model and struggling to adequately preserve knowledge of the remaining classes. To address it, we refine the retention term using "dark knowledge" and propose a mask distillation unlearning method. By applying a mask to separate forgetting logits from retention logits, our approach optimizes both the forgetting and refined retention components simultaneously, retaining knowledge of the remaining classes while ensuring thorough forgetting of the target class. Without access to the remaining data or intervention (i.e., used in some works), we achieve state-of-the-art performance across various benchmarks. What's more, DELETE is a general solution that can be applied to various downstream tasks, including face recognition, backdoor defense, and semantic segmentation with great performance.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d58079ae-2e13-4b36-8427-4beb84b496c9Cited by top-tier papers13
- Machine Unlearning under Retain–Forget EntanglementJingpu Cheng, Ping Liu, Qianxiao Li, CHI ZHANGICLR 2026 · 11 citations
- Distilling LLM Prior to Flow Model for Generalizable Agent's Imagination in Object Goal NavigationBadi Li, Renjie Lu, Yu Zhou, Jingke Meng et al.NeurIPS 2025 · 5 citations
- SALMUBench: A Benchmark for Sensitive Association-Level Multimodal UnlearningCai Selvas-Sala, Lei Kang, Lluís GómezCVPR 2026 · 4 citations
- Unlearning’s Blind Spots: Over‑Unlearning and Prototypical Relearning AttackSeungBum Ha, Saerom Park, Sung Whan YoonICML 2026 · 2 citations
- Easy to Learn, Yet Hard to Forget: Towards Robust Unlearning Under BiasJuneHyoung Kwon, MiHyeon Kim, Eunju Lee, Yoonji Lee et al.AAAI 2026 · 1 citation
Builds on27
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Membership Inference Attacks Against Machine Learning ModelsReza Shokri, Marco Stronati, Congzheng Song, Vitaly ShmatikovS&P 2017 · 5,137 citations
- Decoupled Knowledge DistillationBorui Zhao, Quan Cui, Renjie Song, Yiyu Qiu et al.CVPR 2022 · 835 citations
- Erasing Concepts from Diffusion ModelsRohit Gandikota, Joanna Materzynska, Jaden Fiotto-Kaufman, David BauICCV 2023 · 536 citations
Related papers
- Targeted Forgetting of Image Subgroups in CLIP ModelsZeliang Zhang, Gaowen Liu, Charles Fleming, Ramana Rao Kompella et al.CVPR 2025
- Forgetting Whenever You Want: A Decentralized Continual Learning Framework with On-Demand UnlearningXiao Zhang, Zengzhe Chen, Mingyi Li, Jing Qiao et al.ICML 2026
- Learning to Unlearn: Instance-Wise Unlearning for Pre-trained ClassifiersSungmin Cha, Sungjun Cho, Dasol Hwang, Honglak Lee et al.AAAI 2024 · 79 citations
- VL-Eraser: Vacuum Distillation for Machine Unlearning in Vision-Language ModelsYili Wang, Lu Dai, Tairan Huang, Yijie Xu et al.CVPR 2026
- Forget What Has Seen: Selective Concept Unlearning in Segmentation Foundation ModelsMiaozeng Du, Jiaqi Li, Sirui Pan, Yi Zhan et al.AAAI 2026
