Decoupling forgetting and preservation in federated unlearning via knowledge distillation
摘要
Federated Learning (FL) enables collaborative model training without sharing raw data, but compliance with data privacy regulations such as the “Right to be Forgotten” requires mechanisms to remove specific clients’ contributions from trained models. A key challenge in federated unlearning is selectivity: removing a target client’s influence while preserving performance for other designated clients. This is inherently difficult because deep neural networks learn shared feature representations, and aggressive forgetting operations degrade overall model utility. We propose a Three-Phase Selective Unlearning framework that decouples the conflicting objectives of forgetting and preservation into sequential phases: (1) aggressive forgetting via gradient ascent on the target client’s data, (2) friend restoration through knowledge distillation using the pre-unlearning model as a teacher, and (3) balanced fine-tuning to maintain both objectives simultaneously. By separating these phases, we avoid gradient conflicts that arise from joint optimization. We conduct comprehensive experiments across four datasets—MNIST, CIFAR-10, CIFAR-100 (100 classes), and Tiny-ImageNet (200 classes,